AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 14:38:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: summary of duckduckgo_search (#5488)

Browse Source
This commit is contained in:
非法操作 2024-06-22 13:58:30 +08:00 committed by GitHub
parent 48757e581e
commit 57063095c1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
api/core/tools/provider/builtin/duckduckgo/tools/ddgo_search.py
View File

@ -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