Merge pull request #170 from mindsdb/fix-update-not-using-provider

Add provider field to agent update functionality

Author: dusvyat

mindsdb_sdk/agents.py

@@ -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),

mindsdb_sdk/connectors/rest_api.py

@@ -313,6 +313,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],
@@ -325,6 +326,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

tests/test_sdk.py

@@ -1445,7 +1445,8 @@ def test_update(self, mock_get, mock_put, _):
                 'model_name': 'updated_model',
                 'skills_to_add': ['updated_skill'],
                 'skills_to_remove': [],
-                'params': {'k2': 'v2'}
+                'params': {'k2': 'v2'},
+                'provider': 'mindsdb'
             }
         }
 
@@ -1546,6 +1547,7 @@ def test_add_file(self, mock_post, mock_put, mock_get):
                 'skills_to_add': [agent_update_json['agent']['skills_to_add'][0]],
                 'skills_to_remove': [],
                 'params': {},
+                'provider': 'mindsdb'
             }
         }
         assert agent_update_json == expected_agent_json
@@ -1613,6 +1615,7 @@ def test_add_webpage(self, mock_post, mock_put, mock_get):
                 'skills_to_add':[agent_update_json['agent']['skills_to_add'][0]],
                 'skills_to_remove':[],
                 'params':{},
+                'provider': 'mindsdb'
             }
         }
         assert agent_update_json == expected_agent_json
@@ -1682,6 +1685,7 @@ def test_add_database(self, mock_post, mock_put, mock_get):
                 'skills_to_add': [agent_update_json['agent']['skills_to_add'][0]],
                 'skills_to_remove': [],
                 'params': {'prompt_template': 'using mindsdb sqltoolbox'},
+                'provider': 'mindsdb'
             }
         }
         assert agent_update_json == expected_agent_json