feat: Use rag.

Author: MisluNotFound

api/configs/apo/__init__.py

@@ -30,4 +30,16 @@ class APOConfig(BaseSettings):
     DATAPLANE_URL: str = Field(
         description="dataplane url",
         default="http://localhost:8089"
-    )
+    )
+    APO_KNOWLEDGE_BASE_URL: str = Field(
+        description="apo knowledge base url",
+        default="http://localhost:8080",
+    )
+    APO_KNOWLEDGE_BASE_API_KEY: str = Field(
+        description="apo knowledge base api key",
+        default="",
+    )
+    APO_DEFAULT_KNOWLEDGE_BASE_ID: str = Field(
+        description="apo knowledge base default id",
+        default="syncause_default",
+    )

api/core/tools/builtin_tool/providers/apo_select/tools/knowledge_retrieval.py

@@ -0,0 +1,55 @@
+import json
+from collections.abc import Generator
+from typing import Any, Optional
+
+import requests
+
+from configs import dify_config
+from core.tools.builtin_tool.tool import BuiltinTool
+from core.tools.entities.tool_entities import ToolInvokeMessage
+from models.dataset import Dataset
+
+
+class KnowledgeRetrievalTool(BuiltinTool):
+    def _invoke(
+        self,
+        user_id: str,
+        tool_parameters: dict[str, Any],
+        conversation_id: Optional[str] = None,
+        app_id: Optional[str] = None,
+        message_id: Optional[str] = None,
+    ) -> Generator[ToolInvokeMessage, None, None]:
+        # 获取参数
+        query = tool_parameters.get('query')
+        knowledge_base_id = tool_parameters.get('knowledge_base_id')
+        top_k = tool_parameters.get('top_k', 5)
+        score_threshold = tool_parameters.get('score_threshold', 0.0)
+
+        data = {
+            "query": query,
+            "knowledge_id": knowledge_base_id if knowledge_base_id else dify_config.APO_DEFAULT_KNOWLEDGE_BASE_ID,
+            "retrieval_setting": {
+                "top_k": top_k,
+                "score_threshold": score_threshold
+            }
+        }
+
+        try:
+            url = dify_config.APO_KNOWLEDGE_BASE_URL + '/retrieval'
+            response = requests.post(
+                url, json=data, timeout=30)
+
+            if response.status_code == 200:
+                result = response.json()
+                yield self.create_text_message(json.dumps(result, ensure_ascii=False))
+            else:
+                yield self.create_text_message(json.dumps({
+                    'success': False,
+                    'error': f'检索服务返回错误: {response.status_code}',
+                    'details': response.text
+                }))
+        except Exception as e:
+            yield self.create_text_message(json.dumps({
+                'success': False,
+                'error': f'请求检索服务时发生错误: {str(e)}'
+            }))

api/core/tools/builtin_tool/providers/apo_select/tools/knowledge_retrieval.yaml

@@ -0,0 +1,60 @@
+identity:
+  name: knowledge_retrieval
+  author: APO
+  label:
+    en_US: Retrieve knowledge from APO.
+    zh_Hans: 在APO检索知识
+description:
+  human:
+    en_US: Retrieve knowledge from APO.
+    zh_Hans: 在APO检索知识
+  llm: Retrieve knowledge from APO.
+display:
+  type: metric
+  title: 知识检索
+  unit: s
+parameters:
+  - name: query
+    type: string
+    required: true
+    label:
+      en_US: Knowledge Retrieval Query statement
+      zh_Hans: 知识检索查询语句
+    human_description:
+      en_US: Knowledge retrieval query statement
+      zh_Hans: 知识检索查询语句
+    llm_description: Knowledge retrieval query statement
+    form: llm
+  - name: top_k
+    type: number
+    required: false
+    label:
+      en_US: Top K
+      zh_Hans: 前K个结果
+    human_description:
+      en_US: Top K results
+      zh_Hans: 前K个结果
+    llm_description: Top K results
+    form: llm
+  - name: score_threshold
+    type: number
+    required: false
+    label:
+      en_US: Score Threshold
+      zh_Hans: 分数阈值
+    human_description:
+      en_US: Score threshold for knowledge retrieval
+      zh_Hans: 知识检索分数阈值
+    llm_description: Score threshold for knowledge retrieval
+    form: llm
+  - name: knowledge_base_id
+    type: string
+    required: false
+    label:
+      en_US: Knowledge Base ID
+      zh_Hans: 知识库ID
+    human_description:
+      en_US: Knowledge base ID
+      zh_Hans: 知识库ID
+    llm_description: Knowledge base ID
+    form: llm