Skip to content

Commit

Permalink
Add provider field to agent update functionality
Browse files Browse the repository at this point in the history
Extended the update_agent method to include the updated_provider parameter in both the REST API connector and agents class. Updated an example script to demonstrate the use of the new provider field for creating and managing agents.
  • Loading branch information
dusvyat committed Nov 21, 2024
1 parent 9826cc5 commit 31698d3
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 0 deletions.
1 change: 1 addition & 0 deletions mindsdb_sdk/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -514,6 +514,7 @@ def update(self, name: str, updated_agent: Agent):
self.project.name,
name,
updated_agent.name,
updated_agent.provider,
updated_agent.model_name,
list(skills_to_add),
list(skills_to_remove),
Expand Down
2 changes: 2 additions & 0 deletions mindsdb_sdk/connectors/rest_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,6 +309,7 @@ def update_agent(
project: str,
name: str,
updated_name: str,
updated_provider: str,
updated_model: str,
skills_to_add: List[str],
skills_to_remove: List[str],
Expand All @@ -321,6 +322,7 @@ def update_agent(
'agent': {
'name': updated_name,
'model_name': updated_model,
'provider': updated_provider,
'skills_to_add': skills_to_add,
'skills_to_remove': skills_to_remove,
'params': updated_params
Expand Down

0 comments on commit 31698d3

Please sign in to comment.