Merge pull request #11 from mindsdb/integration-test

Minds integration tests

Author: ZoranPandovski

minds/minds.py

@@ -126,11 +126,14 @@ def completion(self, message: str, stream: bool = False) -> Union[str, Iterable[
             stream=stream
         )
         if stream:
-            for chunk in response:
-                yield chunk.choices[0].delta
+            return self.stream_response(response)
         else:
             return response.choices[0].message.content
 
+    def stream_response(self, response):
+        for chunk in response:
+            yield chunk.choices[0].delta
+
 
 class Minds:
     def __init__(self, client):

requirements_test.txt

@@ -0,0 +1 @@
+pytest

tests/integration/test_base_flow.py

@@ -0,0 +1,131 @@
+import os
+import copy
+
+api_key = os.getenv('API_KEY')
+base_url = 'https://dev.mindsdb.com'
+
+from minds.client import Client
+
+client = Client(api_key, base_url=base_url)
+
+import logging
+logging.basicConfig(level=logging.DEBUG)
+
+from minds.datasources.examples import example_ds
+
+from minds.exceptions import ObjectNotFound
+
+
+def test_datasources():
+
+    # remove previous object
+    try:
+        client.datasources.drop(example_ds.name)
+    except ObjectNotFound:
+        ...
+
+    # create
+    ds = client.datasources.create(example_ds)
+    ds = client.datasources.create(example_ds, replace=True)
+    assert ds.name == example_ds.name
+
+    # get
+    ds = client.datasources.get(example_ds.name)
+
+    # list
+    ds_list = client.datasources.list()
+    assert len(ds_list) > 0
+
+    # drop
+    client.datasources.drop(ds.name)
+
+
+def test_minds():
+    ds_name = 'test_datasource_'
+    ds_name2 = 'test_datasource2_'
+    mind_name = 'int_test_mind_'
+    mind_name2 = 'int_test_mind2_'
+    prompt1 = 'answer in german'
+    prompt2 = 'answer in spanish'
+
+    # remove previous objects
+    for name in (mind_name, mind_name2):
+        try:
+            client.minds.drop(name)
+        except ObjectNotFound:
+            ...
+
+    # prepare datasources
+    ds_cfg = copy.copy(example_ds)
+    ds_cfg.name = ds_name
+    ds = client.datasources.create(example_ds, replace=True)
+
+    # second datasource
+    ds2_cfg = copy.copy(example_ds)
+    ds2_cfg.name = ds_name2
+
+    # create
+    mind = client.minds.create(
+        mind_name,
+        datasources=[ds],
+        provider='openai'
+    )
+    mind = client.minds.create(
+        mind_name,
+        replace=True,
+        datasources=[ds.name, ds2_cfg],
+        parameters={
+            'prompt_template': prompt1
+        }
+    )
+
+    # get
+    mind = client.minds.get(mind_name)
+    assert len(mind.datasources) == 2
+    assert mind.parameters['prompt_template'] == prompt1
+
+    # list
+    mind_list = client.minds.list()
+    assert len(mind_list) > 0
+
+    # rename & update
+    mind.update(
+        name=mind_name2,
+        datasources=[ds.name],
+        parameters={
+            'prompt_template': prompt2
+        }
+    )
+    try:
+        mind = client.minds.get(mind_name)
+    except ObjectNotFound:
+        ...
+    else:
+        raise Exception('mind is not renamed')
+
+    mind = client.minds.get(mind_name2)
+    assert len(mind.datasources) == 1
+    assert mind.parameters['prompt_template'] == prompt2
+
+    # add datasource
+    mind.add_datasource(ds2_cfg)
+    assert len(mind.datasources) == 2
+
+    # del datasource
+    mind.del_datasource(ds2_cfg.name)
+    assert len(mind.datasources) == 1
+
+    # completion
+    answer = mind.completion('say hello')
+    assert 'hola' in answer.lower()
+    # stream completion
+    success = False
+    for chunk in mind.completion('say hello', stream=True):
+        if 'hola' in chunk.content.lower():
+            success = True
+    assert success is True
+
+    # drop
+    client.minds.drop(mind_name2)
+    client.datasources.drop(ds.name)
+    client.datasources.drop(ds2_cfg.name)