Merge pull request #939 from airtai/dev

Update packages (#938)
airtai · Sep 19, 2024 · 21b702f · 21b702f
2 parents 5038b8f + 0a1b781
commit 21b702f
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 14 deletions.
diff --git a/captn/captn_agents/backend/teams/_team_with_client.py b/captn/captn_agents/backend/teams/_team_with_client.py
@@ -87,9 +87,9 @@ def _add_client(self) -> None:
             if agent.name in self.roles_with_client:
                 if agent.llm_config["tools"] is None:
                     agent.llm_config.pop("tools")
-                self.client.register_for_llm(agent)
+                self.client._register_for_llm(agent)
 
-        self.client.register_for_execution(self.user_proxy)
+        self.client._register_for_execution(self.user_proxy)
 
     def _add_tools(self) -> None:
         kwargs = {

diff --git a/captn/captn_agents/backend/tools/patch_client.py b/captn/captn_agents/backend/tools/patch_client.py
@@ -29,7 +29,7 @@ def _patch_register_for_execution() -> None:
         global _org_register_for_execution
 
         if _org_register_for_execution is None:
-            _org_register_for_execution = OpenAPI.register_for_execution
+            _org_register_for_execution = OpenAPI._register_for_execution
 
         def register_for_execution(
             self: OpenAPI,
@@ -47,6 +47,6 @@ def register_for_execution(
             _org_register_for_execution(self, agent)
 
         # mock register_for_execution on the instance level
-        client.register_for_execution = MethodType(register_for_execution, client)
+        client._register_for_execution = MethodType(register_for_execution, client)
 
     return _patch_register_for_execution
diff --git a/pyproject.toml b/pyproject.toml
@@ -68,9 +68,9 @@ lint = [
 
 test-core = [
     "coverage[toml]==7.6.0",
-    "pytest==8.3.2",
+    "pytest==8.3.3",
     "pytest-asyncio>=0.23.6",
-    "dirty-equals==0.7.1.post0",
+    "dirty-equals==0.8.0",
     "pyarrow==17.0.0",
     "pytest-retry==1.6.3",
     "pytest-repeat==0.9.3",
@@ -90,7 +90,7 @@ benchmarking = [
 ]
 
 agents = [
-    "fastapi==0.114.0",
+    "fastapi==0.114.2",
     "APScheduler==3.10.4",
     "prisma==0.13.1",
     "google-ads==25.0.0",
@@ -100,7 +100,7 @@ agents = [
     "pyautogen[websurfer,websockets,anthropic,together]==0.2.35",
     "pandas>=2.1",
     "fastcore==1.7.4",
-    "asyncer==0.0.7",
+    "asyncer==0.0.8",
     "pydantic==2.8.2",
     "tenacity==9.0.0",
     "prometheus-client==0.20.0",

diff --git a/tests/ci/captn/captn_agents/backend/tools/test_google_sheets_tools.py b/tests/ci/captn/captn_agents/backend/tools/test_google_sheets_tools.py
@@ -32,8 +32,8 @@ def test_google_sheets_fastapi_openapi(
             "user_id": 2525,
         }
         get_patch_register_for_execution(client, kwargs_to_patch)()
-        client.register_for_llm(assistant)
-        client.register_for_execution(user_proxy)
+        client._register_for_llm(assistant)
+        client._register_for_execution(user_proxy)
 
         user_proxy.initiate_chat(
             assistant,

diff --git a/tests/ci/captn/captn_agents/backend/tools/test_patch_client.py b/tests/ci/captn/captn_agents/backend/tools/test_patch_client.py
@@ -36,8 +36,8 @@ def test_patch_client(self, weather_fastapi_openapi_url: str) -> None:
 
         user_proxy = autogen.UserProxyAgent(name="user_proxy", human_input_mode="NEVER")
 
-        client.register_for_llm(assistant)
-        client.register_for_execution(user_proxy)
+        client._register_for_llm(assistant)
+        client._register_for_execution(user_proxy)
 
         user_proxy.initiate_chat(
             assistant,

diff --git a/tests/ci/captn/captn_agents/backend/tools/test_weather_team_tools.py b/tests/ci/captn/captn_agents/backend/tools/test_weather_team_tools.py
@@ -74,8 +74,8 @@ def test_llm_config(self) -> None:
         agent = AssistantAgent(name="agent", llm_config=self.llm_config)
         user_proxy = UserProxyAgent(name="user_proxy", code_execution_config=False)
 
-        self.client.register_for_execution(user_proxy)
-        self.client.register_for_llm(agent)
+        self.client._register_for_execution(user_proxy)
+        self.client._register_for_llm(agent)
 
         llm_config = agent.llm_config
         name_desc_dict = {