done

Author: parthiv11

mindsdb_sdk/chatbot.py

@@ -20,26 +20,11 @@ def __init__(self, api, project, data: dict):
         self.database_name = data.get('database')
         self.agent_name = data.get('agent')
         self.model_name = data.get('model_name')
+        self.is_running = data.get('is_running')
 
     def __repr__(self):
         return f"{self.__class__.__name__}({self.project.name}.{self.name})"
 
-    def ask(self, query: str, **options):
-        """
-        Ask the chatbot a question or send a message.
-
-        >>> response = chatbot.ask('What is the weather today?')
-
-        :param query: The input query or message for the chatbot.
-        :param options: Additional options to customize the query.
-        :return: Chatbot response.
-        """
-        payload = {
-            'query': query,
-            **options
-        }
-        return self.api.chatbot_interaction(self.project.name, self.name, payload)
-
     def update(self, name: str = None, agent_name: str = None, model_name: str = None, database_name: str = None, inplace: bool = False):
         """
         Update chatbot properties.
@@ -144,14 +129,15 @@ def get(self, name: str) -> Chatbot:
         data = self.api.get_chatbot(self.project.name, name)
         return Chatbot(self.api, self.project, data)
 
-    def create(self, name: str, agent_name: str = None, model_name: str = None, database_name: str = None) -> Chatbot:
+    def create(self, name: str, agent_name: str = None, model_name: str = None, database_name: str = None, is_running: bool = False) -> Chatbot:
         """
         Create a new chatbot.
 
         >>> chatbot = project.chatbots.create(
         ...     'my_chatbot',
         ...     model_name='gpt-4',
         ...     database_name='slack_db'
+        
         ... )
 
         :param name: Name of the chatbot.
@@ -161,7 +147,8 @@ def create(self, name: str, agent_name: str = None, model_name: str = None, data
         """
         payload = {
             'name': name,
-            'database_name': database_name
+            'database_name': database_name,
+            'is_running':is_running
         }
 
         if agent_name:

mindsdb_sdk/connectors/rest_api.py

@@ -520,47 +520,4 @@ def delete_chatbot(self, project: str, chatbot_name: str):
         """
         url = f'{self.url}/api/projects/{project}/chatbots/{chatbot_name}'
         r = self.session.delete(url)
-        _raise_for_status(r)
-
-    @_try_relogin
-    def send_message(self, project: str, chatbot_name: str, message: str, context: dict = None):
-        """
-        Send a message to a chatbot and retrieve the response.
-        :param project: Name of the project.
-        :param chatbot_name: Name of the chatbot.
-        :param message: User message to send.
-        :param context: Optional context for the chatbot.
-        :return: Chatbot's response as a JSON response.
-        """
-        url = f'{self.url}/api/projects/{project}/chatbots/{chatbot_name}/messages'
-        payload = {
-            'message': message
-        }
-        if context:
-            payload['context'] = context
-
-        r = self.session.post(url, json=payload)
-        _raise_for_status(r)
-        return r.json()
-
-    @_try_relogin
-    def stream_chatbot_response(self, project: str, chatbot_name: str, message: str, context: dict = None):
-        """
-        Send a message to a chatbot and stream its response.
-        :param project: Name of the project.
-        :param chatbot_name: Name of the chatbot.
-        :param message: User message to send.
-        :param context: Optional context for the chatbot.
-        :return: A generator yielding streamed chatbot responses.
-        """
-        url = f'{self.url}/api/projects/{project}/chatbots/{chatbot_name}/messages/stream'
-        payload = {
-            'message': message
-        }
-        if context:
-            payload['context'] = context
-
-        stream = self.session.post(url, json=payload, stream=True)
-        client = SSEClient(stream)
-        for chunk in client.events():
-            yield json.loads(chunk.data)
+        _raise_for_status(r)