mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-16 21:53:45 +00:00
21 lines
822 B
Python
21 lines
822 B
Python
from langchain.agents import create_agent
|
|
|
|
from src.agents.lead_agent.prompt import apply_prompt_template
|
|
from src.agents.middlewares.thread_data_middleware import ThreadDataMiddleware
|
|
from src.agents.middlewares.title_middleware import TitleMiddleware
|
|
from src.agents.thread_state import ThreadState
|
|
from src.models import create_chat_model
|
|
from src.sandbox.middleware import SandboxMiddleware
|
|
from src.tools import get_available_tools
|
|
|
|
# ThreadDataMiddleware must be before SandboxMiddleware to ensure thread_id is available
|
|
middlewares = [ThreadDataMiddleware(), SandboxMiddleware(), TitleMiddleware()]
|
|
|
|
lead_agent = create_agent(
|
|
model=create_chat_model(thinking_enabled=True),
|
|
tools=get_available_tools(),
|
|
middleware=middlewares,
|
|
system_prompt=apply_prompt_template(),
|
|
state_schema=ThreadState,
|
|
)
|