MockEmbeddings Class — langchain Architecture

Architecture documentation for the MockEmbeddings class in test_caching.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  1edabc9d_e73c_8bf1_13cc_acd640cb321f["MockEmbeddings"]
  b1e4f760_c634_d3bf_ca9a_db7ab899cc4a["Embeddings"]
  1edabc9d_e73c_8bf1_13cc_acd640cb321f -->|extends| b1e4f760_c634_d3bf_ca9a_db7ab899cc4a
  6b1a1e58_d1bc_2756_cc7d_9c01f64a73a2["test_caching.py"]
  1edabc9d_e73c_8bf1_13cc_acd640cb321f -->|defined in| 6b1a1e58_d1bc_2756_cc7d_9c01f64a73a2
  612e991b_9f93_9ffa_30df_6f17578ea52c["embed_documents()"]
  1edabc9d_e73c_8bf1_13cc_acd640cb321f -->|method| 612e991b_9f93_9ffa_30df_6f17578ea52c
  8cf0588a_034d_73d9_0bd2_30dbc746184a["embed_query()"]
  1edabc9d_e73c_8bf1_13cc_acd640cb321f -->|method| 8cf0588a_034d_73d9_0bd2_30dbc746184a

Relationship Graph

Source Code

libs/langchain/tests/unit_tests/embeddings/test_caching.py lines 16–31

class MockEmbeddings(Embeddings):
    @override
    def embed_documents(self, texts: list[str]) -> list[list[float]]:
        # Simulate embedding documents
        embeddings: list[list[float]] = []
        for text in texts:
            if text == "RAISE_EXCEPTION":
                msg = "Simulated embedding failure"
                raise ValueError(msg)
            embeddings.append([len(text), len(text) + 1])
        return embeddings

    @override
    def embed_query(self, text: str) -> list[float]:
        # Simulate embedding a query
        return [5.0, 6.0]