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 = {