瀏覽代碼

fix: summary of duckduckgo_search (#5488)

tags/0.6.12
非法操作 1 年之前
父節點
當前提交
57063095c1
No account linked to committer's email address
共有 1 個文件被更改,包括 19 次插入1 次删除
  1. 19
    1
      api/core/tools/provider/builtin/duckduckgo/tools/ddgo_search.py

+ 19
- 1
api/core/tools/provider/builtin/duckduckgo/tools/ddgo_search.py 查看文件

@@ -2,9 +2,20 @@ from typing import Any

from duckduckgo_search import DDGS

from core.model_runtime.entities.message_entities import SystemPromptMessage
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool

SUMMARY_PROMPT = """
User's query:
{query}

Here is the search engine result:
{content}

Please summarize the result in a few sentences.
"""


class DuckDuckGoSearchTool(BuiltinTool):
"""
@@ -25,5 +36,12 @@ class DuckDuckGoSearchTool(BuiltinTool):
results = [res.get("body") for res in response]
results = "\n".join(results)
if require_summary:
results = self.summary(user_id=user_id, content=results)
results = self.summary_results(user_id=user_id, content=results, query=query)
return self.create_text_message(text=results)

def summary_results(self, user_id: str, content: str, query: str) -> str:
prompt = SUMMARY_PROMPT.format(query=query, content=content)
summary = self.invoke_model(user_id=user_id, prompt_messages=[
SystemPromptMessage(content=prompt),
], stop=[])
return summary.message.content

Loading…
取消
儲存