fix: update assistants client import (#4150)

This commit is contained in:
Sebastián Estévez 2024-10-16 08:22:38 -04:00 • committed by GitHub
commit 9213ee21ce
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 7 additions and 7 deletions

View file

@ -97,7 +97,7 @@ dependencies = [
"yfinance>=0.2.40", "yfinance>=0.2.40",
"langchain-google-community==1.0.7", "langchain-google-community==1.0.7",
"wolframalpha>=5.1.3", "wolframalpha>=5.1.3",
"astra-assistants>=2.1.4", "astra-assistants>=2.2.2",
"composio-langchain==0.5.9", "composio-langchain==0.5.9",
"spider-client>=0.0.27", "spider-client>=0.0.27",
"nltk>=3.9.1", "nltk>=3.9.1",

View file

@ -7,7 +7,7 @@ import threading
import astra_assistants.tools as astra_assistants_tools import astra_assistants.tools as astra_assistants_tools
import requests import requests
from astra_assistants import OpenAI, patch from astra_assistants import OpenAIWithDefaultKey, patch
from astra_assistants.tools.tool_interface import ToolInterface from astra_assistants.tools.tool_interface import ToolInterface
client_lock = threading.Lock() client_lock = threading.Lock()
@ -18,7 +18,7 @@ def get_patched_openai_client(shared_component_cache):
os.environ["ASTRA_ASSISTANTS_QUIET"] = "true" os.environ["ASTRA_ASSISTANTS_QUIET"] = "true"
client = shared_component_cache.get("client") client = shared_component_cache.get("client")
if client is None: if client is None:
client = patch(OpenAI()) client = patch(OpenAIWithDefaultKey())
shared_component_cache.set("client", client) shared_component_cache.set("client", client)
return client return client

8
uv.lock generated
View file

@ -266,7 +266,7 @@ wheels = [
[[package]] [[package]]
name = "astra-assistants" name = "astra-assistants"
version = "2.1.4" version = "2.2.2"
source = { registry = "https://pypi.org/simple" } source = { registry = "https://pypi.org/simple" }
dependencies = [ dependencies = [
{ name = "aiohttp" }, { name = "aiohttp" },
@ -281,9 +281,9 @@ dependencies = [
{ name = "tree-sitter" }, { name = "tree-sitter" },
{ name = "tree-sitter-python" }, { name = "tree-sitter-python" },
] ]
sdist = { url = "https://files.pythonhosted.org/packages/1f/fe/e747dbe5b4adcd6f02cece3a1a113865267257113d6f5ba7118097245d33/astra_assistants-2.1.4.tar.gz", hash = "sha256:cfd9c2914baa0f03a6b69b417e402a478127fb1f0f539226dabe5bd4578f387f", size = 67581 } sdist = { url = "https://files.pythonhosted.org/packages/b9/a1/2a0baaa257b4de12dccf2c18e1cfad29d3f860321f11ad500d334254f67d/astra_assistants-2.2.2.tar.gz", hash = "sha256:523758c87599ad17d083c99dce77ecd2eda17f967f43c3269e6cd85751ee15a6", size = 67554 }
wheels = [ wheels = [
{ url = "https://files.pythonhosted.org/packages/1b/cc/d05434d4b8e4c31cd58f8c4479503138b93c410c3b5d30e8764f39faa663/astra_assistants-2.1.4-py3-none-any.whl", hash = "sha256:c2dbb32bacb82a674be2b5f1a28918adc4fddec2655934c66c47dbb5c8bc1e6e", size = 78310 }, { url = "https://files.pythonhosted.org/packages/fb/2f/d69cfe69dfe5bf37d17fa5b47e2ecd001195d04b2a25702968cb32a1e8e4/astra_assistants-2.2.2-py3-none-any.whl", hash = "sha256:0a99968ec7b913f30152d3942488a91ffd207dd4f3333d2af18417c3f2b0a800", size = 78332 },
] ]
[[package]] [[package]]
@ -3632,7 +3632,7 @@ dev = [
[package.metadata] [package.metadata]
requires-dist = [ requires-dist = [
{ name = "assemblyai", specifier = ">=0.33.0" }, { name = "assemblyai", specifier = ">=0.33.0" },
{ name = "astra-assistants", specifier = ">=2.1.4" }, { name = "astra-assistants", specifier = ">=2.2.2" },
{ name = "beautifulsoup4", specifier = ">=4.12.2" }, { name = "beautifulsoup4", specifier = ">=4.12.2" },
{ name = "boto3", specifier = "~=1.34.162" }, { name = "boto3", specifier = "~=1.34.162" },
{ name = "cassio", marker = "extra == 'cassio'", specifier = ">=0.1.7" }, { name = "cassio", marker = "extra == 'cassio'", specifier = ">=0.1.7" },