AI agent 大模型实战之 使用langchain 创建和调用智能体

langchain 为1.27版本

from langchain.agents import create_agent
from langchain.tools import tool, ToolRuntime
from app.bailian.common import llm
from dataclasses import dataclass
from langchain.agents.structured_output import ToolStrategy


SYSTEM_PROMPT="""
你是数学计算专家

你有一个工具函数
- add ：根据这个函数去把用户输入的2个值相加 并且返回
"""

@tool
def add(a:int,b:int) -> int:
    """把传入的值相加"""
    return a+b

# Define response format
@dataclass
class ResponseFormat:
    """Response schema for the agent."""
    # 计算返回的结果
    response: int
    # 调用工具的参数
    args: object


agent = create_agent(
    model=llm,
    tools=[add],
    system_prompt=SYSTEM_PROMPT,
    response_format=ToolStrategy(ResponseFormat),
)

# Run the agent
resp = agent.invoke(
    {"messages": [{"role": "user", "content": "10000+100=?"}]}
)
print(resp['structured_response'])

返回