Skip to content

Commit 31698d3

Browse files
committed
Add provider field to agent update functionality
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.
1 parent 9826cc5 commit 31698d3

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

mindsdb_sdk/agents.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -514,6 +514,7 @@ def update(self, name: str, updated_agent: Agent):
514514
self.project.name,
515515
name,
516516
updated_agent.name,
517+
updated_agent.provider,
517518
updated_agent.model_name,
518519
list(skills_to_add),
519520
list(skills_to_remove),

mindsdb_sdk/connectors/rest_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -309,6 +309,7 @@ def update_agent(
309309
project: str,
310310
name: str,
311311
updated_name: str,
312+
updated_provider: str,
312313
updated_model: str,
313314
skills_to_add: List[str],
314315
skills_to_remove: List[str],
@@ -321,6 +322,7 @@ def update_agent(
321322
'agent': {
322323
'name': updated_name,
323324
'model_name': updated_model,
325+
'provider': updated_provider,
324326
'skills_to_add': skills_to_add,
325327
'skills_to_remove': skills_to_remove,
326328
'params': updated_params

0 commit comments

Comments
 (0)