fix: disable refine with kg by default

pingcap · Jan 10, 2025 · 9f3859b · 9f3859b
1 parent 39319ed
commit 9f3859b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backend/app/api/admin_routes/models.py b/backend/app/api/admin_routes/models.py
@@ -44,4 +44,4 @@ class ChatEngineBasedRetrieveRequest(BaseModel):
     top_k: Optional[int] = 5
     similarity_top_k: Optional[int] = None
     oversampling_factor: Optional[int] = 5
-    enable_kg_enchance_query_refine: Optional[bool] = True
+    enable_kg_enchance_query_refine: Optional[bool] = False
diff --git a/backend/app/api/admin_routes/retrieve.py b/backend/app/api/admin_routes/retrieve.py
@@ -51,7 +51,7 @@ async def embedding_retrieve(
     top_k: Optional[int] = 5,
     similarity_top_k: Optional[int] = None,
     oversampling_factor: Optional[int] = 5,
-    enable_kg_enchance_query_refine: Optional[bool] = True,
+    enable_kg_enchance_query_refine: Optional[bool] = False,
 ) -> List[NodeWithScore]:
     try:
         return retrieve_service.chat_engine_retrieve_chunks(