Skip to content

Commit 2276d15

Browse files
committed
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.
1 parent 31698d3 commit 2276d15

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/test_sdk.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1445,7 +1445,8 @@ def test_update(self, mock_get, mock_put, _):
14451445
'model_name': 'updated_model',
14461446
'skills_to_add': ['updated_skill'],
14471447
'skills_to_remove': [],
1448-
'params': {'k2': 'v2'}
1448+
'params': {'k2': 'v2'},
1449+
'provider': 'mindsdb'
14491450
}
14501451
}
14511452

@@ -1546,6 +1547,7 @@ def test_add_file(self, mock_post, mock_put, mock_get):
15461547
'skills_to_add': [agent_update_json['agent']['skills_to_add'][0]],
15471548
'skills_to_remove': [],
15481549
'params': {},
1550+
'provider': 'mindsdb'
15491551
}
15501552
}
15511553
assert agent_update_json == expected_agent_json
@@ -1613,6 +1615,7 @@ def test_add_webpage(self, mock_post, mock_put, mock_get):
16131615
'skills_to_add':[agent_update_json['agent']['skills_to_add'][0]],
16141616
'skills_to_remove':[],
16151617
'params':{},
1618+
'provider': 'mindsdb'
16161619
}
16171620
}
16181621
assert agent_update_json == expected_agent_json
@@ -1682,6 +1685,7 @@ def test_add_database(self, mock_post, mock_put, mock_get):
16821685
'skills_to_add': [agent_update_json['agent']['skills_to_add'][0]],
16831686
'skills_to_remove': [],
16841687
'params': {'prompt_template': 'using mindsdb sqltoolbox'},
1688+
'provider': 'mindsdb'
16851689
}
16861690
}
16871691
assert agent_update_json == expected_agent_json

0 commit comments

Comments
 (0)