klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Bump up LangChain version and bugfix (#536)

Browse Source 
Co-authored-by: Taranjeet Singh <reachtotj@gmail.com>
This commit is contained in:
Qihang
2023-09-03 12:41:09 -07:00
committed by GitHub
parent c07fbc07b1
commit 2aa25a5169
3 changed files with 7 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/vectordb/test_chroma_db.py
View File

@@ -3,11 +3,9 @@
import unittest
from unittest.mock import patch
from chromadb.config import Settings
from embedchain import App
from embedchain.config import AppConfig
from embedchain.vectordb.chroma_db import ChromaDB, chromadb
from embedchain.vectordb.chroma_db import ChromaDB
class TestChromaDbHosts(unittest.TestCase):
@@ -18,15 +16,10 @@ class TestChromaDbHosts(unittest.TestCase):
host = "test-host"
port = "1234"
with patch.object(chromadb, "HttpClient") as mock_client:
_db = ChromaDB(host=host, port=port, embedding_fn=len)
expected_settings = Settings(
chroma_server_host="test-host",
chroma_server_http_port="1234",
)
mock_client.assert_called_once_with(expected_settings)
db = ChromaDB(host=host, port=port, embedding_fn=len)
settings = db.client.get_settings()
self.assertEqual(settings.chroma_server_host, host)
self.assertEqual(settings.chroma_server_http_port, port)
# Review this test