Merge branch 'mindsdb:main' into patch-2

Author: Better-Boy

.github/workflows/test_on_deploy.yml

@@ -9,7 +9,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.8', '3.9','3.10', '3.11']
+        python-version: ['3.10']
     steps:
       - name: Checkout code
         uses: actions/checkout@v2
@@ -28,4 +28,4 @@ jobs:
         env:
           PYTHONPATH: ./
           API_KEY: ${{ secrets.API_KEY }}
-          BASE_URL: ${{ secrets.BASE_URL }}
+          BASE_URL: 'https://mdb.ai'

README.md

@@ -150,4 +150,9 @@ client.datasources.drop('my_datasource')
 >Note: The SDK currently does not support automatically removing a data source if it is no longer connected to any mind.
 
 ### Other SDKs
-#### [Command-Line](https://github.com/Better-Boy/minds-cli-sdk)
+- [Dart-SDK](https://github.com/ArnavK-09/mdb_dart)
+
+#### Command Line Tools
+- [Minds CLI](https://github.com/Better-Boy/minds-cli-sdk)
+
+

minds/__about__.py

@@ -1,6 +1,6 @@
 __title__ = 'minds_sdk'
 __package_name__ = 'minds'
-__version__ = '1.0.8'
+__version__ = '1.0.9'
 __description__ = 'An AI-Data Mind is an LLM with the built-in power to answer data questions for Agents'
 __email__ = 'hello@mindsdb.com'
 __author__ = 'MindsDB Inc'

minds/datasources/datasources.py

@@ -40,7 +40,7 @@ def create(self, ds_config: DatabaseConfig, update=False):
         utils.validate_datasource_name(name)
 
         if update:
-            self.api.put('/datasources', data=ds_config.model_dump())
+            self.api.put(f'/datasources/{name}', data=ds_config.model_dump())
         else:
             self.api.post('/datasources', data=ds_config.model_dump())
         return self.get(name)

minds/minds.py

@@ -1,5 +1,4 @@
 from typing import List, Union, Iterable
-import utils
 from openai import OpenAI
 import minds.utils as utils
 import minds.exceptions as exc
@@ -278,11 +277,13 @@ def create(
 
         if update:
             method = self.api.put
+            url = f'/projects/{self.project}/minds/{name}'
         else:
             method = self.api.post
+            url = f'/projects/{self.project}/minds'
 
         method(
-            f'/projects/{self.project}/minds',
+            url,
             data={
                 'name': name,
                 'model_name': model_name,

tests/integration/test_base_flow.py

@@ -82,7 +82,7 @@ def test_minds():
     # prepare datasources
     ds_cfg = copy.copy(example_ds)
     ds_cfg.name = ds_name
-    ds = client.datasources.create(example_ds, replace=True)
+    ds = client.datasources.create(example_ds, update=True)
 
     # second datasource
     ds2_cfg = copy.copy(example_ds)
@@ -91,7 +91,7 @@ def test_minds():
 
     # create
     with pytest.raises(MindNameInvalid):
-        mind = client.minds.create(
+        client.minds.create(
             invalid_mind_name,
             datasources=[ds],
             provider='openai'
@@ -119,9 +119,6 @@ def test_minds():
     mind = client.minds.get(mind_name)
     assert len(mind.datasources) == 2
     assert mind.prompt_template == prompt1
-    
-    with pytest.raises(MindNameInvalid):
-        client.minds.get(invalid_mind_name)
 
     # list
     mind_list = client.minds.list()
@@ -188,6 +185,4 @@ def test_minds():
     client.minds.drop(mind_name2)
     client.datasources.drop(ds.name)
     client.datasources.drop(ds2_cfg.name)
-    
-    with pytest.raises(MindNameInvalid):
-        client.minds.drop(invalid_mind_name)
+

tests/unit/test_unit.py

@@ -44,19 +44,19 @@ def test_create_datasources(self, mock_del, mock_post, mock_put, mock_get):
         response_mock(mock_get, example_ds.model_dump())
 
         ds = client.datasources.create(example_ds)
-        def check_ds_created(ds, mock_post):
+        def check_ds_created(ds, mock_post, url):
             self._compare_ds(ds, example_ds)
             args, kwargs = mock_post.call_args
 
             assert kwargs['headers'] == {'Authorization': 'Bearer ' + API_KEY}
             assert kwargs['json'] == example_ds.model_dump()
-            assert args[0] == 'https://mdb.ai/api/datasources'
+            assert args[0] == url
 
-        check_ds_created(ds, mock_post)
+        check_ds_created(ds, mock_post, 'https://mdb.ai/api/datasources')
 
         # with update
         ds = client.datasources.create(example_ds, update=True)
-        check_ds_created(ds, mock_put)
+        check_ds_created(ds, mock_put, f'https://mdb.ai/api/datasources/{ds.name}')
 
     @patch('requests.get')
     def test_get_datasource(self, mock_get):
@@ -132,17 +132,17 @@ def test_create(self, mock_del, mock_post, mock_put, mock_get):
         }
         mind = client.minds.create(**create_params)
 
-        def check_mind_created(mind, mock_post, create_params):
+        def check_mind_created(mind, mock_post, create_params, url):
             args, kwargs = mock_post.call_args
-            assert args[0].endswith('/api/projects/mindsdb/minds')
+            assert args[0].endswith(url)
             request = kwargs['json']
             for key in ('name', 'datasources', 'provider', 'model_name'),:
                 assert request.get(key) == create_params.get(key)
             assert create_params.get('prompt_template') == request.get('parameters', {}).get('prompt_template')
 
             self.compare_mind(mind, self.mind_json)
 
-        check_mind_created(mind, mock_post, create_params)
+        check_mind_created(mind, mock_post, create_params, '/api/projects/mindsdb/minds')
 
         # -- with replace --
         create_params = {
@@ -156,15 +156,15 @@ def check_mind_created(mind, mock_post, create_params):
         args, _ = mock_del.call_args
         assert args[0].endswith(f'/api/projects/mindsdb/minds/{mind_name}')
 
-        check_mind_created(mind, mock_post, create_params)
+        check_mind_created(mind, mock_post, create_params, '/api/projects/mindsdb/minds')
 
         # -- with update --
         mock_del.reset_mock()
         mind = client.minds.create(update=True, **create_params)
         # is not deleted
         assert not mock_del.called
 
-        check_mind_created(mind, mock_put, create_params)
+        check_mind_created(mind, mock_put, create_params, f'/api/projects/mindsdb/minds/{mind_name}')
 
     @patch('requests.get')
     @patch('requests.patch')