Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update packages (#938) #939

Merged
merged 1 commit into from
Sep 19, 2024
Merged

Update packages (#938) #939

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions captn/captn_agents/backend/teams/_team_with_client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 = {
Expand Down
4 changes: 2 additions & 2 deletions captn/captn_agents/backend/tools/patch_client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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
8 changes: 4 additions & 4 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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",
Expand All @@ -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",
Expand Down
4 changes: 2 additions & 2 deletions tests/ci/captn/captn_agents/backend/tools/test_google_sheets_tools.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
4 changes: 2 additions & 2 deletions tests/ci/captn/captn_agents/backend/tools/test_patch_client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
4 changes: 2 additions & 2 deletions tests/ci/captn/captn_agents/backend/tools/test_weather_team_tools.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 = {
Expand Down
Loading