From 2276d15dca4988910fe23c01251804f50268e476 Mon Sep 17 00:00:00 2001 From: dusvyat Date: Fri, 22 Nov 2024 18:28:39 +0200 Subject: [PATCH] Add 'provider' field with 'mindsdb' to test_sdk.py This commit adds the 'provider' field with the value 'mindsdb' to the test data in multiple test cases within test_sdk.py. The update ensures that the 'provider' information is consistently included in the JSON structures verified by the tests. This change aims to improve the accuracy and completeness of the test coverage. --- tests/test_sdk.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_sdk.py b/tests/test_sdk.py index 0630155..33d7f59 100644 --- a/tests/test_sdk.py +++ b/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