import json
import os
from openai import OpenAI

client = OpenAI(
    api_key=os.environ["EXAIV_API_KEY"],
    base_url="https://exaiv.com/api/ai/v1",
)

tools = [
    {
        "type": "function",
        "function": {
            "name": "search_orders",
            "description": "按订单号或客户名查询订单状态",
            "parameters": {
                "type": "object",
                "properties": {
                    "query": {"type": "string", "description": "订单号或客户名"},
                },
                "required": ["query"],
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "send_refund",
            "description": "对指定订单发起退款",
            "parameters": {
                "type": "object",
                "properties": {
                    "order_id": {"type": "string"},
                    "amount": {"type": "number"},
                },
                "required": ["order_id", "amount"],
            },
        },
    },
]

def execute_tool(name, args):
    if name == "search_orders":
        return {"order_id": "A1024", "status": "delivered_damaged", "amount": 129.0}
    if name == "send_refund":
        return {"ok": True, "refund_id": "R7788"}
    raise ValueError(f"unknown tool {name}")

messages = [{"role": "user", "content": "客户说订单 A1024 送达时破损，按流程处理"}]

for _ in range(10):  # 最多 10 步，防止无限循环
    resp = client.chat.completions.create(
        model="kimi-k3",
        messages=messages,
        tools=tools,
    )
    msg = resp.choices[0].message
    if not msg.tool_calls:
        print(msg.content)  # 最终回答
        break
    messages.append(msg)
    for call in msg.tool_calls:
        args = json.loads(call.function.arguments)
        result = execute_tool(call.function.name, args)
        messages.append({
            "role": "tool",
            "tool_call_id": call.id,
            "content": json.dumps(result, ensure_ascii=False),
        })