mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-18 18:34:27 -04:00
Yoshi
4e8f11b36a
Solves #8644 This embedding models output identical random embedding vectors, given the input texts are identical. Useful when used in unittest. @baskaryan
17 lines
631 B
Python
17 lines
631 B
Python
from langchain.embeddings import DeterministicFakeEmbedding
|
|
|
|
|
|
def test_deterministic_fake_embeddings() -> None:
|
|
"""
|
|
Test that the deterministic fake embeddings return the same
|
|
embedding vector for the same text.
|
|
"""
|
|
fake = DeterministicFakeEmbedding(size=10)
|
|
text = "Hello world!"
|
|
assert fake.embed_query(text) == fake.embed_query(text)
|
|
assert fake.embed_query(text) != fake.embed_query("Goodbye world!")
|
|
assert fake.embed_documents([text, text]) == fake.embed_documents([text, text])
|
|
assert fake.embed_documents([text, text]) != fake.embed_documents(
|
|
[text, "Goodbye world!"]
|
|
)
|