AI Agent开发入门与实践

def agent_loop(user_input):
    messages = [system_prompt, user_input]
    
    while True:
        response = llm.chat(messages)
        
        if response.has_tool_call():
            # 执行工具调用，结果塞回消息列表
            result = execute_tool(response.tool_call)
            messages.append(response)
            messages.append(tool_result_to_message(result))
            # 继续下一轮
        else:
            # 没有工具调用，输出最终回复
            return response.text

import json
import requests

# 工具1：搜索网页
def web_search(query: str) -> str:
    """搜索互联网，返回相关结果"""
    # 示例：使用 DuckDuckGo 的免费接口
    url = f"https://api.duckduckgo.com/?q={query}&format=json"
    resp = requests.get(url, timeout=10)
    data = resp.json()
    results = data.get("RelatedTopics", [])[:3]
    return json.dumps([r.get("Text", "") for r in results], ensure_ascii=False)

# 工具2：数学计算
def calculator(expression: str) -> str:
    """安全计算数学表达式"""
    try:
        # 只允许数字和数学运算符
        import re
        if not re.match(r'^[\d\s+\-*/().%]+$', expression):
            return "Error: 包含非法字符"
        return str(eval(expression))
    except Exception as e:
        return f"Error: {e}"

TOOLS = [
    {
        "type": "function",
        "function": {
            "name": "web_search",
            "description": "搜索互联网获取信息，当需要查找实时资料时使用",
            "parameters": {
                "type": "object",
                "properties": {
                    "query": {
                        "type": "string",
                        "description": "搜索关键词"
                    }
                },
                "required": ["query"]
            }
        }
    },
    {
        "type": "function",
        "function": {
            "name": "calculator",
            "description": "计算数学表达式，当需要精确计算时使用",
            "parameters": {
                "type": "object",
                "properties": {
                    "expression": {
                        "type": "string",
                        "description": "数学表达式，如 '(2+3)*4'"
                    }
                },
                "required": ["expression"]
            }
        }
    }
]

# 工具名 → 函数的映射
TOOL_MAP = {
    "web_search": web_search,
    "calculator": calculator
}

from openai import OpenAI

client = OpenAI()  # 也可以用任何兼容 OpenAI API 的服务

SYSTEM_PROMPT = """你是一个智能助手，可以：
1. 使用 web_search 搜索最新信息
2. 使用 calculator 进行精确计算

遇到需要查资料的问题先搜索；遇到计算问题先算再回答。
不要编造没有搜索过的信息。"""

def run_agent(user_input: str, max_turns: int = 5) -> str:
    messages = [
        {"role": "system", "content": SYSTEM_PROMPT},
        {"role": "user", "content": user_input}
    ]
    
    for turn in range(max_turns):
        response = client.chat.completions.create(
            model="claude-sonnet-5-20251001",  # 或其他模型
            messages=messages,
            tools=TOOLS,
            tool_choice="auto"
        )
        
        msg = response.choices[0].message
        
        # 没有工具调用 → 返回最终文本
        if not msg.tool_calls:
            return msg.content
        
        # 有工具调用 → 执行并继续
        messages.append(msg.model_dump())  # assistant 消息
        
        for tool_call in msg.tool_calls:
            func_name = tool_call.function.name
            func_args = json.loads(tool_call.function.arguments)
            
            print(f"[Agent] 调用工具: {func_name}({func_args})")
            
            result = TOOL_MAP[func_name](**func_args)
            
            messages.append({
                "role": "tool",
                "tool_call_id": tool_call.id,
                "content": result
            })
    
    return "达到最大轮次限制"

# 测试
if __name__ == "__main__":
    result = run_agent("昨天BTC涨了多少？特斯拉股价是多少？")
    print(f"[回复] {result}")