Merge pull request #205 from mindsdb/bugfix/add_skills_extra_params_to_agent

Added the skills_extra_parameters Parameter to Agents

Author: MinuraPunchihewa

mindsdb_sdk/__about__.py

@@ -1,6 +1,6 @@
 __title__ = 'mindsdb_sdk'
 __package_name__ = 'mindsdb_sdk'
-__version__ = '3.4.5'
+__version__ = '3.4.6'
 __description__ = "MindsDB Python SDK, provides an SDK to use a remote mindsdb instance"
 __email__ = "[email protected]"
 __author__ = 'MindsDB Inc'

mindsdb_sdk/agents.py

@@ -87,6 +87,7 @@ def __init__(
             data: dict = {},
             prompt_template: str = None,
             params: dict = {},
+            skills_extra_parameters: dict = {},
             collection: CollectionBase = None
     ):
         self.name = name
@@ -98,6 +99,7 @@ def __init__(
         self.data = data
         self.prompt_template = prompt_template
         self.params = params
+        self.skills_extra_parameters = skills_extra_parameters
         self.collection = collection
 
     def completion(self, messages: List[dict]) -> AgentCompletion:
@@ -190,10 +192,16 @@ def __eq__(self, other):
             return False
         if self.provider != other.provider:
             return False
+        if self.data != other.data:
+            return False
+        if self.prompt_template != other.prompt_template:
+            return False
         if self.skills != other.skills:
             return False
         if self.params != other.params:
             return False
+        if self.skills_extra_parameters != other.skills_extra_parameters:
+            return False
         if self.created_at != other.created_at:
             return False
         return self.updated_at == other.updated_at
@@ -216,6 +224,7 @@ def from_json(cls, json: dict, collection: CollectionBase):
             json.get('data', {}),
             json.get('prompt_template'),
             json.get('params', {}),
+            json.get('skills_extra_parameters', {}),
             collection
         )
 

tests/test_sdk.py

@@ -1508,7 +1508,7 @@ def test_update(self, mock_put, mock_get):
             },
             'data': {
                 'tables': ['test_database.test_table'],
-                'knowledge_bases': ['test_kb'],
+                'knowledge_bases': ['test_kb', 'test_kb2'],
             },
             'created_at': created_at,
             'updated_at': updated_at,