fix chroma init bug (#7639)

This commit is contained in:
Bagatur
2023-07-13 03:00:33 -04:00
committed by GitHub
parent f307ca094b
commit b08f903755
2 changed files with 28 additions and 11 deletions

View File

@@ -267,3 +267,17 @@ def test_chroma_with_relevance_score_custom_normalization_fn() -> None:
(Document(page_content="bar", metadata={"page": "1"}), -0.0),
(Document(page_content="baz", metadata={"page": "2"}), -0.0),
]
def test_init_from_client() -> None:
import chromadb
client = chromadb.Client(chromadb.config.Settings())
Chroma(client=client)
def test_init_from_client_settings() -> None:
import chromadb
client_settings = chromadb.config.Settings()
Chroma(client_settings=client_settings)