ForcePilot/src/agents/common/subagents/calc_agent.py
Wenjie Zhang 67ddbe6f40 feat: 添加 CLAUDE.md 文件以提供项目开发指导
- 新增 CLAUDE.md 文件,包含项目概述、开发环境管理、项目架构、开发规范、特色功能模块及调试和故障排除指南。
- 详细描述了使用 Docker Compose 管理开发环境的步骤和常用命令。
- 介绍了项目的技术栈、核心目录结构及主要服务,确保开发者能够快速上手。
2025-11-12 11:00:39 +08:00

20 lines
773 B
Python

from langchain.agents import create_agent
from langchain.tools import tool
from src import config
from src.agents.common import load_chat_model
from src.agents.common.tools import calculator
calc_agent = create_agent(
model=load_chat_model(config.default_model),
tools=[calculator],
system_prompt="你可以使用计算器工具,处理各种数学计算任务。",
)
@tool(name_or_callable="calc_agent_tool", description="进行计算任务,输入是数学表达式或描述,输出计算结果。")
async def calc_agent_tool(description: str) -> str:
"""CalcAgent 工具 - 使用子智能体 CalcAgent 进行计算任务"""
response = await calc_agent.ainvoke({"messages": [("user", description)]})
return response["messages"][-1].content