Skip to content
Snippets Groups Projects
Unverified Commit 809dc3b1 authored by Andrei Fajardo's avatar Andrei Fajardo Committed by GitHub
Browse files

[BUG-FIX] retriever_mode param missing when constructing KGTableRetriever (#10725)

add retriever_mode param when constructing KGTableRetriever
parent c6b43b25
Branches
Tags
No related merge requests found
......@@ -138,6 +138,7 @@ class KnowledgeGraphIndex(BaseIndex[KG]):
object_map=self._object_map,
llm=self._llm,
embed_model=embed_model or self._embed_model,
retriever_mode=retriever_mode,
**kwargs,
)
......
"""KG Retrievers."""
import logging
from collections import defaultdict
from enum import Enum
......@@ -120,7 +121,11 @@ class KGTableRetriever(BaseRetriever):
self.query_keyword_extract_template = query_keyword_extract_template or DQKET
self.similarity_top_k = similarity_top_k
self._include_text = include_text
self._retriever_mode = KGRetrieverMode(retriever_mode)
self._retriever_mode = (
KGRetrieverMode(retriever_mode)
if retriever_mode
else KGRetrieverMode.KEYWORD
)
self._llm = llm or llm_from_settings_or_context(Settings, index.service_context)
self._embed_model = embed_model or embed_model_from_settings_or_context(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment