mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-06 23:24:48 +00:00
27 lines
770 B
Python
27 lines
770 B
Python
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}
|