优化DifyQueryRetrieval和ModelTool中的请求超时设置，调整DifyExporter类中的分类逻辑，移除无用字段，更新词条处理逻辑，

2025-07-16 16:26:46 +08:00
parent a934f2c398
commit 8a58fef1a7
6 changed files with 19 additions and 30 deletions
@@ -41,7 +41,7 @@ class DifyQueryRetrieval:
    def retrieve_by_dataset(self, query: str, dataset_name: str) -> List[Dict[str, Any]]:
        try:
            knowledge_base_client = KnowledgeBaseClient(api_key=self._api_key, base_url=self._base_url, dataset_id=self._datasets_list[dataset_name])
-            documents = knowledge_base_client.retrieve(query)
+            documents = knowledge_base_client.retrieve(query, timeout=300)
            retrieved_documents = documents.json().get("records", [])
            
            # 添加数据集信息