fix(backend): fix RetrieveService

pingcap · Aug 21, 2024 · 62baf71 · 62baf71
1 parent 2a5964c
commit 62baf71
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/backend/app/rag/retrieve.py b/backend/app/rag/retrieve.py
@@ -67,7 +67,8 @@ def _retrieve(self, question: str, top_k: int) -> List[Document]:
             )
 
             if kg_config.using_intent_search:
-                result = graph_index.intent_based_search(question, include_meta=True)
+                intent_relationships = graph_index.intent_analyze(question)
+                result = graph_index.intent_based_search(intent_relationships, include_meta=True)
 
                 graph_knowledges = get_prompt_by_jinja2_template(
                     self.chat_engine_config.llm.intent_graph_knowledge,