[Improvements] Upgrade langchain-openai package and other improvements (#1372)

This commit is contained in:
Deshraj Yadav
2024-05-21 23:42:50 -07:00
committed by GitHub
parent 9ba445e623
commit 9544c69c55
9 changed files with 86 additions and 232 deletions

View File

@@ -1,5 +1,3 @@
from unittest.mock import MagicMock
import pytest
from chromadb.api.types import Documents, Embeddings
@@ -46,14 +44,6 @@ def test_set_vector_dimension_type_error(base_embedder):
base_embedder.set_vector_dimension(None)
def test_langchain_default_concept():
embeddings = MagicMock()
embeddings.embed_documents.return_value = ["Embedding1", "Embedding2"]
embed_function = BaseEmbedder._langchain_default_concept(embeddings)
result = embed_function(["text1", "text2"])
assert result == ["Embedding1", "Embedding2"]
def test_embedder_with_config():
embedder = BaseEmbedder(BaseEmbedderConfig())
assert isinstance(embedder.config, BaseEmbedderConfig)