Trying to make use of another class with minimal examples, this time
CypherTemplateRetriever
and I'm getting some errors as it tries to parse my query string into parameters. Here's how I'm setting it up:
class AssessmentTemplateParams(BaseModel):
"""Template params for a cypher query."""
keywords: list[str] = Field(
description="A list of keywords to use for lookup in a knowledge graph."
)
self.cypher_retriever = CypherTemplateRetriever(
self.graph_store,
llm=llm,
output_cls=AssessmentTemplateParams,
cypher_query=TEMPLATED_CYPHER_QUERY
)