Menu

Google GenAI (Gemini)

Google Vertex AI

Overview Vector Stores Chat History Key-Value Stores

Language

Theme

Method●Since v0.3

similarity_search

Retrieve list of connected Entities found via traversal of KnowledgeGraph.

Use LLM & Prompt to find entities within the input_document itself.
Find Entity Nodes that match those found in the input_document.
Traverse the graph using these as starting points.

similarity_search(
    self,
    input_document: str,
) -> List[Entity]

Returns: List of connected Entity dictionaries.

Parameters

Name	Type	Description
`input_document`*	`str`	String to find relevant documents for.

View source on GitHub