2026-01-14 07:20:00 +08:00
|
|
|
from langchain.agents import create_agent
|
2026-01-14 09:08:20 +08:00
|
|
|
|
2026-01-14 07:20:00 +08:00
|
|
|
from src.agents.lead_agent.prompt import apply_prompt_template
|
2026-01-14 23:29:18 +08:00
|
|
|
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
|
|
|
|
|
|
2026-01-14 23:29:18 +08:00
|
|
|
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(),
|
2026-01-14 23:29:18 +08:00
|
|
|
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
|
|
|
)
|