使用ddgs代替TAVILY
This commit is contained in:
@@ -1,24 +0,0 @@
|
||||
from langchain.agents.middleware import wrap_tool_call
|
||||
|
||||
from src.server.deep_agent.init_llm import llm
|
||||
from src.server.deep_agent.init_prompt import build_painter_prompt
|
||||
from src.server.deep_agent.tools.generate_furniture_sketch import generate_furniture, edit_furniture
|
||||
|
||||
|
||||
@wrap_tool_call
|
||||
async def log_tool_calls(request, handler):
|
||||
"""Intercept and log every tool call - demonstrates cross-cutting concern."""
|
||||
print(request)
|
||||
return handler(request)
|
||||
|
||||
|
||||
def build_painter_subagent(workspace_dir):
|
||||
painter_subagent = {
|
||||
"name": "painter_subagent",
|
||||
"description": "理解用户意图,利用prompt编辑或生成家具sketch图像",
|
||||
"system_prompt": build_painter_prompt(),
|
||||
"tools": [generate_furniture, edit_furniture],
|
||||
"model": llm,
|
||||
# "middleware": [log_tool_calls],
|
||||
}
|
||||
return painter_subagent
|
||||
Reference in New Issue
Block a user