do not override openai client outside of assistants

langflow-ai · Oct 15, 2024 · f9d7586 · f9d7586
1 parent db2c121
commit f9d7586
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/backend/base/langflow/base/astra_assistants/util.py b/src/backend/base/langflow/base/astra_assistants/util.py
@@ -7,7 +7,8 @@
 
 import astra_assistants.tools as astra_assistants_tools
 import requests
-from astra_assistants import OpenAI, patch
+from astra_assistants import OpenAI as AstraOpenAI
+from astra_assistants import patch
 from astra_assistants.tools.tool_interface import ToolInterface
 
 client_lock = threading.Lock()
@@ -18,7 +19,7 @@ def get_patched_openai_client(shared_component_cache):
     os.environ["ASTRA_ASSISTANTS_QUIET"] = "true"
     client = shared_component_cache.get("client")
     if client is None:
-        client = patch(OpenAI())
+        client = patch(AstraOpenAI())
         shared_component_cache.set("client", client)
     return client