aembed_query() — langchain Function Reference

Architecture documentation for the aembed_query() function in cache.py from the langchain codebase.

Function python LangChainCore ApiManagement

Entity Profile

LangChainCore→ ApiManagement→ aembed_query() — langchain Function Reference

Dependency Diagram

graph TD
  5cbca7c4_941a_8657_0b3f_feaa039eb7e9["aembed_query()"]
  b3be4e54_ae5f_c527_4e99_0843e3d30f72["CacheBackedEmbeddings"]
  5cbca7c4_941a_8657_0b3f_feaa039eb7e9 -->|defined in| b3be4e54_ae5f_c527_4e99_0843e3d30f72
  style 5cbca7c4_941a_8657_0b3f_feaa039eb7e9 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain/langchain_classic/embeddings/cache.py lines 264–285

    async def aembed_query(self, text: str) -> list[float]:
        """Embed query text.

        By default, this method does not cache queries. To enable caching, set the
        `cache_query` parameter to `True` when initializing the embedder.

        Args:
            text: The text to embed.

        Returns:
            The embedding for the given text.
        """
        if not self.query_embedding_store:
            return await self.underlying_embeddings.aembed_query(text)

        (cached,) = await self.query_embedding_store.amget([text])
        if cached is not None:
            return cached

        vector = await self.underlying_embeddings.aembed_query(text)
        await self.query_embedding_store.amset([(text, vector)])
        return vector