From 9a21acd771b773b45e4de144bc278022481ed3f4 Mon Sep 17 00:00:00 2001
From: Robert Jambrecic <robert@airt.ai>
Date: Tue, 17 Sep 2024 14:39:48 +0200
Subject: [PATCH] Update packages

---
 captn/captn_agents/backend/teams/_team_with_client.py     | 4 ++--
 captn/captn_agents/backend/tools/patch_client.py          | 4 ++--
 pyproject.toml                                            | 8 ++++----
 .../backend/tools/test_google_sheets_tools.py             | 4 ++--
 .../captn/captn_agents/backend/tools/test_patch_client.py | 4 ++--
 .../captn_agents/backend/tools/test_weather_team_tools.py | 4 ++--
 6 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/captn/captn_agents/backend/teams/_team_with_client.py b/captn/captn_agents/backend/teams/_team_with_client.py
index 97ed1524..e70d8e65 100644
--- 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
index 0e59e221..593937c6 100644
--- 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
index 841ece5b..8cc194ea 100644
--- 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
index 2cc15d08..dd41d59c 100644
--- 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
index d4472f64..e108c1ea 100644
--- 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
index 96a363b7..96f80036 100644
--- 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 = {