pip install openai

from openai import OpenAI

client = OpenAI(
    api_key="your-key",
    base_url="https://api.ofox.ai/v1"  # 聚合接口，一個 Key 用所有模型
)

# 模擬的業(yè)務函數(shù)
def query_order(order_id: str) -> dict:
    """根據(jù)訂單號查詢訂單信息"""
    # 實際項目里這里查數(shù)據(jù)庫
    fake_db = {
        "ORD001": {"order_id": "ORD001", "product": "機械鍵盤", "status": "已發(fā)貨", "amount": 399},
        "ORD002": {"order_id": "ORD002", "product": "顯示器支架", "status": "待付款", "amount": 89},
    }
    return fake_db.get(order_id, {"error": f"訂單 {order_id} 不存在"})


def query_logistics(order_id: str) -> dict:
    """根據(jù)訂單號查詢物流信息"""
    fake_logistics = {
        "ORD001": {"carrier": "順豐", "tracking_no": "SF1234567890", "status": "在途中，預計明天到"},
    }
    return fake_logistics.get(order_id, {"error": f"訂單 {order_id} 暫無物流信息"})

tools = [
    {
        "type": "function",
        "function": {
            "name": "query_order",
            "description": "根據(jù)訂單號查詢訂單詳情，包括商品名、狀態(tài)、金額",
            "parameters": {
                "type": "object",
                "properties": {
                    "order_id": {
                        "type": "string",
                        "description": "訂單編號，格式如 ORD001"
                    }
                },
                "required": ["order_id"]
            }
        }
    },
    {
        "type": "function",
        "function": {
            "name": "query_logistics",
            "description": "根據(jù)訂單號查詢物流狀態(tài)，包括快遞公司、運單號、當前狀態(tài)",
            "parameters": {
                "type": "object",
                "properties": {
                    "order_id": {
                        "type": "string",
                        "description": "訂單編號，格式如 ORD001"
                    }
                },
                "required": ["order_id"]
            }
        }
    }
]

import json

# 工具名 → 實際函數(shù)的映射
TOOL_MAP = {
    "query_order": query_order,
    "query_logistics": query_logistics,
}

SYSTEM_PROMPT = """你是一個電商客服助手。你可以幫用戶查詢訂單信息和物流狀態(tài)。
請用簡潔友好的語氣回復。如果用戶沒有提供訂單號，請先詢問訂單號。"""


def run_agent(user_input: str, messages: list = None, max_turns: int = 5) -> str:
    """
    運行 Agent 主循環(huán)
    max_turns: 最大工具調(diào)用輪次，防止死循環(huán)
    """
    if messages is None:
        messages = [{"role": "system", "content": SYSTEM_PROMPT}]

    messages.append({"role": "user", "content": user_input})

    for turn in range(max_turns):
        response = client.chat.completions.create(
            model="gpt-4o",  # 換成 claude-3.5-sonnet 等也行
            messages=messages,
            tools=tools,
            tool_choice="auto",  # 讓模型自己決定要不要調(diào)工具
        )

        msg = response.choices[0].message
        messages.append(msg)  # 把 assistant 的回復加入上下文

        # 如果沒有工具調(diào)用，說明模型準備好直接回答了
        if not msg.tool_calls:
            return msg.content

        # 執(zhí)行每個工具調(diào)用
        for tool_call in msg.tool_calls:
            func_name = tool_call.function.name
            func_args = json.loads(tool_call.function.arguments)

            print(f"  [Agent] 調(diào)用工具: {func_name}({func_args})")

            # 執(zhí)行函數(shù)
            if func_name in TOOL_MAP:
                result = TOOL_MAP[func_name](**func_args)
            else:
                result = {"error": f"未知工具: {func_name}"}

            # 把工具執(zhí)行結(jié)果塞回消息列表
            messages.append({
                "role": "tool",
                "tool_call_id": tool_call.id,
                "content": json.dumps(result, ensure_ascii=False)
            })

    return "抱歉，我處理這個問題遇到了困難，請聯(lián)系人工客服。"

if __name__ == "__main__":
    # 測試 1：正常查詢
    print("=" * 50)
    print("用戶：幫我看看 ORD001 到哪了")
    print("Agent：", run_agent("幫我看看 ORD001 到哪了"))

    print()

    # 測試 2：需要先查訂單再查物流
    print("=" * 50)
    print("用戶：ORD001 買的什么？快遞到哪了？")
    print("Agent：", run_agent("ORD001 買的什么？快遞到哪了？"))

    print()

    # 測試 3：缺少訂單號
    print("=" * 50)
    print("用戶：我想查一下我的快遞")
    print("Agent：", run_agent("我想查一下我的快遞"))

try:
    func_args = json.loads(tool_call.function.arguments)
except json.JSONDecodeError:
    # 嘗試用 ast.literal_eval 兜底，再不行就報錯
    import ast
    try:
        func_args = ast.literal_eval(tool_call.function.arguments)
    except:
        result = {"error": "參數(shù)解析失敗"}
        # 繼續(xù)把 error 喂給模型，讓它重試

def trim_messages(messages: list, max_tokens: int = 8000) -> list:
    """保留 system prompt + 最近的消息"""
    system_msg = messages[0]  # system prompt 永遠保留
    recent = messages[1:]

    # 粗略估算：1個中文字符約2個token
    while len(json.dumps(recent, ensure_ascii=False)) > max_tokens * 2 and len(recent) > 2:
        recent.pop(0)

    return [system_msg] + recent