updated the operation for creating an Agent

Author: MinuraPunchihewa

mindsdb_sdk/agents.py

@@ -486,19 +486,26 @@ def add_database(self, name: str, database: str, tables: List[str], description:
         self.update(agent.name, agent)
 
     def create(
-            self,
-            name: str,
-            model: Union[Model, dict, str] = None,
-            provider: str = None,
-            skills: List[Union[Skill, str]] = None,
-            params: dict = None,
-            **kwargs) -> Agent:
+        self,
+        name: str,
+        model_name: Union[Model, str] = None,
+        provider: str = None,
+        skills: List[Union[Skill, str]] = None,
+        data: dict = None,
+        model: dict = None,
+        prompt_template: str = None,
+        params: dict = None,
+        **kwargs
+    ) -> Agent:
         """
         Create new agent and return it
 
         :param name: Name of the agent to be created
-        :param model: Model to be used by the agent
+        :param model_name: MindsDB model to be used by the agent
         :param skills: List of skills to be used by the agent. Currently only 'sql' is supported.
+        :param provider: Provider of the model, e.g. 'mindsdb', 'openai', etc.
+        :param data: Data to be used by the agent. This is usually a dictionary with 'tables' and/or 'knowledge_base' keys.
+        :param model: Model parameters to be used by the agent. This is usually a dictionary
         :param params: Parameters for the agent
 
         :return: created agent object
@@ -519,18 +526,23 @@ def create(
         if params is None:
             params = {}
         params.update(kwargs)
-
-        if 'prompt_template' not in params:
-            params['prompt_template'] = _DEFAULT_LLM_PROMPT
-
-        if model is None:
-            model = _DEFAULT_LLM_MODEL
-        elif isinstance(model, Model):
-            model = model.name
+        
+        if isinstance(model_name, Model):
+            model_name = model_name.name
             provider = 'mindsdb'
 
-        data = self.api.create_agent(self.project.name, name, model, provider, skill_names, params)
-        return Agent.from_json(data, self)
+        agent = self.api.create_agent(
+            self.project.name,
+            name,
+            model_name,
+            provider,
+            skill_names,
+            data,
+            model,
+            prompt_template,
+            params
+        )
+        return Agent.from_json(agent, self)
 
     def update(self, name: str, updated_agent: Agent):
         """

mindsdb_sdk/connectors/rest_api.py

@@ -306,16 +306,30 @@ def agent_completion_stream_v2(self, project: str, name: str, messages: List[dic
             yield e
 
     @_try_relogin
-    def create_agent(self, project: str, name: str, model: str = None, provider: str = None, skills: List[str] = None, params: dict = None):
+    def create_agent(
+        self,
+        project: str,
+        name: str,
+        model_name: str = None,
+        provider: str = None,
+        skills: List[str] = None,
+        data: dict = None,
+        model: dict = None,
+        prompt_template: str = None,
+        params: dict = None
+    ):
         url = self.url + f'/api/projects/{project}/agents'
         r = self.session.post(
             url,
             json={
                 'agent': {
                     'name': name,
-                    'model_name': model,
+                    'model_name': model_name,
                     'provider': provider,
                     'skills': skills,
+                    'data': data,
+                    'model': model,
+                    'prompt_template': prompt_template,
                     'params': params
                 }
             }