Files
deer-flow/backend/src/agents/lead_agent/agent.py

19 lines
630 B
Python
Raw Normal View History

2026-01-14 07:20:00 +08:00
from langchain.agents import create_agent
2026-01-14 07:20:00 +08:00
from src.agents.lead_agent.prompt import apply_prompt_template
from src.agents.middlewares.title_middleware import TitleMiddleware
2026-01-14 12:32:34 +08:00
from src.agents.thread_state import ThreadState
2026-01-14 07:20:00 +08:00
from src.models import create_chat_model
2026-01-14 12:32:34 +08:00
from src.sandbox.middleware import SandboxMiddleware
2026-01-14 07:20:00 +08:00
from src.tools import get_available_tools
middlewares = [SandboxMiddleware(), TitleMiddleware()]
2026-01-14 07:20:00 +08:00
lead_agent = create_agent(
model=create_chat_model(thinking_enabled=True),
tools=get_available_tools(),
middleware=middlewares,
2026-01-14 07:20:00 +08:00
system_prompt=apply_prompt_template(),
2026-01-14 12:32:34 +08:00
state_schema=ThreadState,
2026-01-14 07:20:00 +08:00
)