添加rag置信度判断

2026-05-06 01:15:52 +08:00
parent 3ae9daa01a
commit 1260bef5cb
35 changed files with 335 additions and 221 deletions
--- a/backend/app/main_graph/state.py
+++ b/backend/app/main_graph/state.py
@@ -75,6 +75,8 @@ class MainGraphState:
    rag_context: str = ""
    rag_retrieved: bool = False
    rag_docs: List[Dict[str, Any]] = field(default_factory=list)
+    rag_confidence: float = 0.0  # RAG 检索置信度 (0.0-1.0)
+    rag_attempts: int = 0  # RAG 检索次数统计
    
    # ========== 联网搜索相关字段 ==========
    web_search_results: List[str] = field(default_factory=list)