Skip to content

Commit

Permalink
Add 'provider' field with 'mindsdb' to test_sdk.py
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
dusvyat committed Nov 22, 2024
1 parent 31698d3 commit 2276d15
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/test_sdk.py
Original file line number Diff line number Diff line change
Expand Up @@ -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'
}
}

Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 2276d15

Please sign in to comment.