tests[patch]: run standard tests for embeddings and populate embeddings API ref (#28545)

plus minor updates to chat models and vector store API refs
This commit is contained in:
ccurme
2024-12-05 14:39:03 -05:00
committed by GitHub
parent 1581857e3d
commit ecdfc98ef6
6 changed files with 214 additions and 33 deletions

View File

@@ -0,0 +1,26 @@
from typing import Type
from langchain_core.embeddings import DeterministicFakeEmbedding, Embeddings
from langchain_tests.integration_tests import EmbeddingsIntegrationTests
from langchain_tests.unit_tests import EmbeddingsUnitTests
class TestFakeEmbeddingsUnit(EmbeddingsUnitTests):
@property
def embeddings_class(self) -> Type[Embeddings]:
return DeterministicFakeEmbedding
@property
def embedding_model_params(self) -> dict:
return {"size": 6} # embedding dimension
class TestFakeEmbeddingsIntegration(EmbeddingsIntegrationTests):
@property
def embeddings_class(self) -> Type[Embeddings]:
return DeterministicFakeEmbedding
@property
def embedding_model_params(self) -> dict:
return {"size": 6}