手写一个最小的 Agent 循环

import { z } from 'zod'

// ===== 通用的模型接口，各家 SDK 通过适配器转换成这个形状 =====
type ToolCall = { id: string; name: string; arguments: string } // arguments 是模型生成的 JSON 字符串
type AssistantMessage = { role: 'assistant'; content: string; toolCalls?: ToolCall[] }
type Message =
  | { role: 'system' | 'user'; content: string }
  | AssistantMessage
  | { role: 'tool'; toolCallId: string; content: string; isError?: boolean }
type ToolDef = { name: string; description: string; parameters: object }
declare const llm: { chat(req: { messages: Message[]; tools: ToolDef[] }): Promise<AssistantMessage> }

// ===== 工具注册表 =====
interface Tool<S extends z.ZodType = z.ZodType> {
  name: string
  description: string
  schema: S
  execute(args: z.infer<S>): Promise<unknown>
}
const defineTool = <S extends z.ZodType>(tool: Tool<S>) => tool

const tools: Tool[] = [
  defineTool({
    name: 'get_weather',
    description: '查询指定城市今天的天气。用户问天气、穿衣、是否带伞时使用。',
    schema: z.object({ city: z.string().describe('城市名，如：北京') }),
    async execute({ city }) {
      return { city, weather: '小雨', temp: 18 } // 示例数据，实际应请求天气接口
    },
  }),
]
const toolDefs: ToolDef[] = tools.map((t) => ({
  name: t.name,
  description: t.description,
  parameters: z.toJSONSchema(t.schema, { io: 'input' }), // 按输入类型生成，带默认值的字段不会被标成必填
}))

// ===== 执行单个工具：任何错误都转成结果，交给模型处理 =====
async function runTool(call: ToolCall): Promise<{ content: string; isError: boolean }> {
  const tool = tools.find((t) => t.name === call.name)
  if (!tool) return { content: `不存在工具 ${call.name}`, isError: true }
  try {
    const parsed = tool.schema.safeParse(JSON.parse(call.arguments || '{}'))
    if (!parsed.success) {
      const detail = parsed.error.issues.map((i) => `${i.path.join('.') || '参数'}：${i.message}`).join('；')
      return { content: `参数校验失败：${detail}`, isError: true }
    }
    const result = await tool.execute(parsed.data)
    const text = typeof result === 'string' ? result : (JSON.stringify(result) ?? '执行成功')
    return { content: text.length > 8000 ? `${text.slice(0, 8000)}\n（结果过长，已截断）` : text, isError: false }
  } catch (err) { // JSON 解析失败和执行时抛出的异常都会到这里
    return { content: `出错：${err instanceof Error ? err.message : String(err)}`, isError: true }
  }
}

// ===== Agent 循环：没有工具调用时结束，最多 MAX_STEPS 步 =====
const MAX_STEPS = 10

export async function runAgent(input: string): Promise<string> {
  const messages: Message[] = [
    { role: 'system', content: '你是一个能使用工具的助手。工具出错时根据错误信息调整；无法完成时如实说明，不要编造结果。' },
    { role: 'user', content: input },
  ]
  for (let step = 0; step < MAX_STEPS; step++) {
    const reply = await llm.chat({ messages, tools: toolDefs })
    messages.push(reply) // 先追加模型的回复，再追加工具结果
    if (!reply.toolCalls?.length) return reply.content // 没有工具调用，这就是最终回答
    for (const call of reply.toolCalls) {
      const { content, isError } = await runTool(call)
      messages.push({ role: 'tool', toolCallId: call.id, content, isError })
    }
  }
  return `已执行 ${MAX_STEPS} 步仍未完成，请缩小问题范围后重试。`
}