Fix http of default milvus uri (#86)

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
zilliztech · Oct 24, 2023 · 936c6b2 · 936c6b2
1 parent cbfde4d
commit 936c6b2
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/config.py b/config.py
@@ -72,7 +72,7 @@
 # Vector db configs
 VECTORDB_CONFIG = {
     'connection_args': {
-        'uri': os.getenv('MILVUS_URI', 'https://localhost:19530'),
+        'uri': os.getenv('MILVUS_URI', 'http://localhost:19530'),
         'user': os.getenv('MILVUS_USER', ''),
         'password': os.getenv('MILVUS_PASSWORD', ''),
         'secure': True if os.getenv('MILVUS_SECURE', 'False').lower() == 'true' else False

diff --git a/tests/requirements.txt b/tests/requirements.txt
@@ -7,4 +7,4 @@ transformers
 dashscope
 zhipuai
 sentence-transformers
-erniebot
+erniebot==0.4.0
diff --git a/tests/unit_tests/src_langchain/llm/test_ernie.py b/tests/unit_tests/src_langchain/llm/test_ernie.py
@@ -11,12 +11,12 @@ class TestERNIE(unittest.TestCase):
     def test_generate(self):
         from erniebot.response import EBResponse
         with patch('erniebot.ChatCompletion.create') as mock_post:
-            mock_res = EBResponse(code=200,
-                                  body={'id': 'as-0000000000', 'object': 'chat.completion', 'created': 11111111,
+            mock_res = EBResponse(rcode=200,
+                                  rbody={'id': 'as-0000000000', 'object': 'chat.completion', 'created': 11111111,
                                         'result': 'OK, this is a mock answer.',
                                         'usage': {'prompt_tokens': 1, 'completion_tokens': 13, 'total_tokens': 14},
                                         'need_clear_history': False, 'is_truncated': False},
-                                  headers={'Connection': 'keep-alive',
+                                  rheaders={'Connection': 'keep-alive',
                                            'Content-Security-Policy': 'frame-ancestors https://*.baidu.com/',
                                            'Content-Type': 'application/json', 'Date': 'Mon, 23 Oct 2023 03:30:53 GMT',
                                            'Server': 'nginx', 'Statement': 'AI-generated',

diff --git a/tests/unit_tests/src_towhee/pipelines/test_pipelines.py b/tests/unit_tests/src_towhee/pipelines/test_pipelines.py
@@ -142,12 +142,12 @@ def test_ernie(self):
         from erniebot.response import EBResponse
 
         with patch('erniebot.ChatCompletion.create') as mock_post:
-            mock_res = EBResponse(code=200,
-                                  body={'id': 'as-0000000000', 'object': 'chat.completion', 'created': 11111111,
+            mock_res = EBResponse(rcode=200,
+                                  rbody={'id': 'as-0000000000', 'object': 'chat.completion', 'created': 11111111,
                                         'result': MOCK_ANSWER,
                                         'usage': {'prompt_tokens': 1, 'completion_tokens': 13, 'total_tokens': 14},
                                         'need_clear_history': False, 'is_truncated': False},
-                                  headers={'Connection': 'keep-alive',
+                                  rheaders={'Connection': 'keep-alive',
                                            'Content-Security-Policy': 'frame-ancestors https://*.baidu.com/',
                                            'Content-Type': 'application/json', 'Date': 'Mon, 23 Oct 2023 03:30:53 GMT',
                                            'Server': 'nginx', 'Statement': 'AI-generated',