Vòng lặp AI agent: để chatbot quán Mộc tự làm nhiều bước

04-vong-lap-agent/
  package.json
  src/
    thong-tin-quan.ts   giờ mở cửa, phí ship, bảng giá (giữ nguyên)
    don-hang.ts         dữ liệu đơn hàng, hai hàm tra cứu và hàm thêm món (mới)
    cong-cu.ts          mô tả 3 công cụ cho mô hình + code chạy công cụ
    prompts.ts          system prompt
    offline.ts          mô hình giả lập, để chạy được khi chưa có API key
    goi-llm.ts          gọi Claude thật (hoặc bản giả lập)
    agent.ts            vòng lặp agent (mới)
    hoi-moc.ts          hỏi 6 câu mẫu
    gioi-han.ts         thử chạm giới hạn số lần gọi
    hai-tin-nhan.ts     thử hai tin nhắn liên tiếp

{
  "name": "moc-ai-agent-04",
  "private": true,
  "type": "module",
  "dependencies": {
    "@anthropic-ai/sdk": "1.5.0"
  }
}

// Thông tin cố định của quán Mộc (minh họa), giống dữ liệu ở tập 1.
export const MENU: Record<string, number> = {
  "cà phê đen": 25000,
  "cà phê sữa": 29000,
  "bạc xỉu": 32000,
  "cà phê muối": 35000,
  "bánh mì trứng": 25000,
};

export const GIO_MO_CUA = "Mộc mở cửa từ 6:30 đến 22:00 mỗi ngày.";
export const PHI_SHIP = "Mộc giao trong bán kính 5 km, phí 15.000đ; đơn từ 150.000đ được miễn phí.";

export function bangGia(): string {
  return Object.entries(MENU)
    .map(([mon, gia]) => `- ${mon}: ${gia.toLocaleString("vi-VN")}đ`)
    .join("\n");
}

import { MENU } from "./thong-tin-quan.ts";

export type Don = {
  maDon: string;
  soDienThoai: string;
  mon: string[];
  trangThai: "đang pha chế" | "đang giao" | "đã giao";
  datLuc: string;
};

const DON_HANG: Don[] = [
  { maDon: "1041", soDienThoai: "0901000001", mon: ["cà phê sữa", "bánh mì trứng"], trangThai: "đang pha chế", datLuc: "hôm nay 8:12" },
  { maDon: "1042", soDienThoai: "0901000002", mon: ["bạc xỉu"], trangThai: "đang giao", datLuc: "hôm nay 9:40" },
];

export function timDonTheoMa(maDon: string): Don | undefined {
  return DON_HANG.find((don) => don.maDon === maDon);
}

export function timDonTheoSdt(soDienThoai: string): Don | undefined {
  return DON_HANG.find((don) => don.soDienThoai === soDienThoai);
}

// Thêm món vào đơn. Chỉ được thêm khi đơn còn đang pha chế: đơn đã lên xe thì không sửa được nữa.
export function themMon(maDon: string, mon: string, soLuong: number): { ok: true; don: Don } | { ok: false; lyDo: string } {
  const don = timDonTheoMa(maDon);
  if (!don) return { ok: false, lyDo: `Không có đơn ${maDon}.` };
  if (!(mon in MENU)) return { ok: false, lyDo: `Quán không bán "${mon}".` };
  if (don.trangThai !== "đang pha chế") return { ok: false, lyDo: `Đơn ${maDon} ${don.trangThai}, không thêm món được nữa.` };
  for (let i = 0; i < soLuong; i++) don.mon.push(mon);
  return { ok: true, don };
}

import type Anthropic from "@anthropic-ai/sdk";

import { themMon, timDonTheoMa, timDonTheoSdt } from "./don-hang.ts";

export const CONG_CU: Anthropic.Tool[] = [
  {
    name: "xem_don",
    description:
      "Xem món và trạng thái hiện tại của một đơn hàng theo mã đơn (4 chữ số, ví dụ 1041). " +
      "Gọi công cụ này mỗi khi khách hỏi đơn đã giao chưa, đang ở đâu, khi nào tới.",
    strict: true,
    input_schema: {
      type: "object",
      properties: { ma_don: { type: "string", description: "Mã đơn gồm 4 chữ số, không có dấu #" } },
      required: ["ma_don"],
      additionalProperties: false,
    },
  },
  {
    name: "tim_don_theo_sdt",
    description:
      "Tìm mã đơn gần nhất của khách theo số điện thoại (10 chữ số, bắt đầu bằng 0). " +
      "Chỉ trả về mã đơn; muốn biết trạng thái thì gọi tiếp xem_don.",
    strict: true,
    input_schema: {
      type: "object",
      properties: { so_dien_thoai: { type: "string", description: "Số điện thoại 10 chữ số" } },
      required: ["so_dien_thoai"],
      additionalProperties: false,
    },
  },
  {
    name: "them_mon",
    description:
      "Thêm món vào một đơn còn đang pha chế. Chỉ gọi khi khách nói rõ muốn thêm món nào, bao nhiêu phần. " +
      "Đơn đang giao hoặc đã giao thì công cụ sẽ báo lỗi.",
    strict: true,
    input_schema: {
      type: "object",
      properties: {
        ma_don: { type: "string", description: "Mã đơn gồm 4 chữ số" },
        mon: { type: "string", description: "Tên món đúng như trong bảng giá, ví dụ: bạc xỉu" },
        so_luong: { type: "integer", description: "Số phần, từ 1 đến 5" },
      },
      required: ["ma_don", "mon", "so_luong"],
      additionalProperties: false,
    },
  },
];

export type KetQuaCongCu = { noiDung: string; loi: boolean };

// Đầu vào do mô hình viết ra: luôn kiểm tra lại trước khi dùng, như với dữ liệu người dùng gửi lên.
export function chayCongCu(ten: string, dauVao: unknown): KetQuaCongCu {
  const input = (dauVao ?? {}) as Record<string, unknown>;
  if (ten === "xem_don") {
    const ma = String(input.ma_don ?? "").trim();
    if (!/^\d{4}$/.test(ma)) return { noiDung: `Mã đơn "${ma}" không hợp lệ, cần đúng 4 chữ số.`, loi: true };
    const don = timDonTheoMa(ma);
    if (!don) return { noiDung: `Không có đơn ${ma}.`, loi: true };
    return { noiDung: JSON.stringify({ ma_don: don.maDon, mon: don.mon, trang_thai: don.trangThai, dat_luc: don.datLuc }), loi: false };
  }
  if (ten === "tim_don_theo_sdt") {
    const sdt = String(input.so_dien_thoai ?? "").replace(/\s/g, "");
    if (!/^0\d{9}$/.test(sdt)) return { noiDung: `Số điện thoại "${sdt}" không hợp lệ.`, loi: true };
    const don = timDonTheoSdt(sdt);
    if (!don) return { noiDung: "Không tìm thấy đơn nào với số này.", loi: true };
    return { noiDung: JSON.stringify({ ma_don: don.maDon }), loi: false };
  }
  if (ten === "them_mon") {
    const ma = String(input.ma_don ?? "").trim();
    const mon = String(input.mon ?? "").trim().toLowerCase();
    const soLuong = Number(input.so_luong);
    if (!/^\d{4}$/.test(ma)) return { noiDung: `Mã đơn "${ma}" không hợp lệ, cần đúng 4 chữ số.`, loi: true };
    if (!Number.isInteger(soLuong) || soLuong < 1 || soLuong > 5) return { noiDung: `Số lượng ${input.so_luong} không hợp lệ (1 đến 5).`, loi: true };
    const kq = themMon(ma, mon, soLuong);
    if (!kq.ok) return { noiDung: kq.lyDo, loi: true };
    return { noiDung: JSON.stringify({ ma_don: kq.don.maDon, mon: kq.don.mon, trang_thai: kq.don.trangThai }), loi: false };
  }
  return { noiDung: `Không có công cụ tên "${ten}".`, loi: true };
}

import { GIO_MO_CUA, PHI_SHIP, bangGia } from "./thong-tin-quan.ts";

export const SYSTEM_V4 = `Bạn là nhân viên chăm sóc khách hàng của quán cà phê Mộc. Trả lời ngắn gọn, thân thiện, bằng tiếng Việt.

Chỉ dùng thông tin trong thẻ <thong_tin_quan> để trả lời câu hỏi chung.
<thong_tin_quan>
${GIO_MO_CUA}
${PHI_SHIP}
Bảng giá:
${bangGia()}
</thong_tin_quan>

Khi khách hỏi về một đơn hàng, hãy dùng công cụ để tra, và chỉ nói trạng thái đúng như kết quả công cụ trả về.
Chỉ thêm món khi khách nói rõ món nào và bao nhiêu phần; sau khi thêm, đọc lại cho khách các món trong đơn.
Nếu công cụ báo lỗi hoặc không tìm thấy đơn, hãy nói thật và xin khách kiểm tra lại mã đơn hoặc số điện thoại.`;

// Gửi cuộc hội thoại kèm danh sách công cụ cho mô hình, nhận về phản hồi.
// Có ANTHROPIC_API_KEY: gọi Claude thật. Không có key: dùng chế độ offline (giả lập).
import Anthropic from "@anthropic-ai/sdk";

import { CONG_CU } from "./cong-cu.ts";
import { moHinhOffline, type PhanHoi } from "./offline.ts";

export const MODEL = "claude-opus-5-5";

export function coKey(): boolean {
  return Boolean(process.env.ANTHROPIC_API_KEY) && process.env.MOC_OFFLINE !== "1";
}

export async function goiMoHinh(system: string, messages: Anthropic.MessageParam[]): Promise<PhanHoi> {
  if (!coKey()) return moHinhOffline(system, messages);
  const client = new Anthropic(); // tự đọc ANTHROPIC_API_KEY từ biến môi trường
  const response = await client.messages.create({
    model: MODEL,
    max_tokens: 16000,
    output_config: { effort: "low" },
    system,
    tools: CONG_CU, // giờ có ba công cụ, gồm cả them_mon
    messages,
  });
  return {
    content: response.content,
    stop_reason: response.stop_reason,
    inputTokens: response.usage.input_tokens,
    outputTokens: response.usage.output_tokens,
  };
}

export function layChu(content: Anthropic.ContentBlock[]): string {
  return content
    .filter((b) => b.type === "text")
    .map((b) => b.text)
    .join("");
}

import type Anthropic from "@anthropic-ai/sdk";

import { chayCongCu } from "./cong-cu.ts";
import { goiMoHinh, layChu } from "./goi-llm.ts";
import { SYSTEM_V4 } from "./prompts.ts";

export type KetQuaAgent = { traLoi: string; soLanGoi: number; nhatKy: string[] };

// Vòng lặp agent: gọi mô hình, chạy công cụ nó xin, gửi kết quả lại, lặp tới khi nó trả lời bằng chữ.
export async function chayAgent(cauHoi: string, maxLanGoi = 6): Promise<KetQuaAgent> {
  const messages: Anthropic.MessageParam[] = [{ role: "user", content: cauHoi }];
  const nhatKy: string[] = [];

  for (let lan = 1; lan <= maxLanGoi; lan++) {
    const phanHoi = await goiMoHinh(SYSTEM_V4, messages);
    messages.push({ role: "assistant", content: phanHoi.content }); // luôn gửi lại nguyên văn

    // Không xin công cụ nữa: đây là câu trả lời cuối.
    if (phanHoi.stop_reason !== "tool_use") return { traLoi: layChu(phanHoi.content), soLanGoi: lan, nhatKy };

    // Chạy MỌI khối tool_use của lượt này, gom kết quả vào MỘT tin nhắn.
    const ketQua: Anthropic.ToolResultBlockParam[] = [];
    for (const block of phanHoi.content) {
      if (block.type !== "tool_use") continue;
      const kq = chayCongCu(block.name, block.input);
      nhatKy.push(`lần ${lan}: ${block.name}(${JSON.stringify(block.input)}) → ${kq.noiDung}${kq.loi ? "  (is_error)" : ""}`);
      ketQua.push({ type: "tool_result", tool_use_id: block.id, content: kq.noiDung, is_error: kq.loi });
    }
    messages.push({ role: "user", content: ketQua });
  }

  // Chạm giới hạn: dừng lại và nói thật, thay vì lặp mãi.
  return {
    traLoi: "Dạ yêu cầu này Mộc chưa xử lý xong, nhân viên sẽ kiểm tra và nhắn lại bạn ngay ạ.",
    soLanGoi: maxLanGoi,
    nhatKy: [...nhatKy, `dừng: đã gọi mô hình ${maxLanGoi} lần mà chưa có câu trả lời cuối`],
  };
}

// Mô hình giả lập: một kịch bản viết tay trả về đúng hình dạng phản hồi của API, để chạy được khi chưa có key.
// Nó quyết định bằng regex, không hiểu câu như mô hình thật.
import type Anthropic from "@anthropic-ai/sdk";

import { MENU } from "./thong-tin-quan.ts";

export type PhanHoi = Pick<Anthropic.Message, "content" | "stop_reason"> & { inputTokens: number; outputTokens: number };

let dem = 0;
function goiCongCu(name: string, input: Record<string, string | number>): PhanHoi {
  dem += 1;
  const block = { type: "tool_use", id: `toolu_offline_${dem}`, name, input } as Anthropic.ToolUseBlock;
  return { content: [block], stop_reason: "tool_use", inputTokens: 0, outputTokens: 0 };
}

function traLoi(text: string): PhanHoi {
  return { content: [{ type: "text", text } as Anthropic.TextBlock], stop_reason: "end_turn", inputTokens: 0, outputTokens: 0 };
}

const monTrongCau = (cau: string) => Object.keys(MENU).find((mon) => cau.includes(mon));
const muonThem = (cau: string) => /\bthêm\b/.test(cau) && monTrongCau(cau) !== undefined;

// Tên công cụ của lượt gọi gần nhất: tìm khối tool_use có id trùng với tool_use_id của kết quả.
function tenCongCu(messages: Anthropic.MessageParam[], toolUseId: string): string {
  const truoc = messages.at(-2)!.content as Anthropic.ContentBlock[];
  return (truoc.find((b) => b.type === "tool_use" && b.id === toolUseId) as Anthropic.ToolUseBlock).name;
}

export function moHinhOffline(system: string, messages: Anthropic.MessageParam[]): PhanHoi {
  const cauHoi = String(messages[0].content).toLowerCase();
  const cuoi = messages.at(-1)!;

  // Lượt đầu: chỉ có câu hỏi của khách.
  if (typeof cuoi.content === "string") {
    const ma = cauHoi.match(/\b\d{4}\b/)?.[0];
    if (ma && muonThem(cauHoi)) return goiCongCu("them_mon", { ma_don: ma, mon: monTrongCau(cauHoi)!, so_luong: 1 });
    if (ma) return goiCongCu("xem_don", { ma_don: ma });
    const sdt = cauHoi.match(/\b0\d{9}\b/)?.[0];
    if (sdt) return goiCongCu("tim_don_theo_sdt", { so_dien_thoai: sdt });
    if (muonThem(cauHoi)) return traLoi("Dạ bạn cho Mộc xin mã đơn hoặc số điện thoại đặt hàng để Mộc thêm món nhé!");
    if (/giờ|mở cửa/.test(cauHoi)) {
      return traLoi(`Dạ, ${(system.split("\n").find((d) => /mở cửa từ/.test(d)) ?? "").replace("Mộc mở", "quán mở")}`);
    }
    return traLoi("Dạ Mộc chưa có thông tin này ạ.");
  }

  // Lượt sau: code của mình gửi lại kết quả công cụ (trong một tin nhắn vai "user").
  const ketQua = (cuoi.content as Anthropic.ToolResultBlockParam[]).find((b) => b.type === "tool_result")!;
  const noiDung = String(ketQua.content);
  if (ketQua.is_error) return traLoi(`Dạ Mộc rất tiếc, ${noiDung.charAt(0).toLowerCase()}${noiDung.slice(1)} Bạn cần Mộc hỗ trợ gì thêm không ạ?`);

  const don = JSON.parse(noiDung) as { ma_don: string; mon?: string[]; trang_thai?: string; dat_luc?: string };
  const congCu = tenCongCu(messages, ketQua.tool_use_id);
  if (congCu === "tim_don_theo_sdt") return goiCongCu("xem_don", { ma_don: don.ma_don });
  if (congCu === "xem_don" && muonThem(cauHoi) && don.trang_thai === "đang pha chế") {
    return goiCongCu("them_mon", { ma_don: don.ma_don, mon: monTrongCau(cauHoi)!, so_luong: 1 });
  }
  if (congCu === "them_mon") return traLoi(`Dạ Mộc đã thêm vào đơn ${don.ma_don}. Đơn giờ gồm: ${don.mon!.join(", ")}, vẫn đang pha chế ạ.`);
  return traLoi(`Dạ đơn ${don.ma_don} (${don.mon!.join(", ")}) bạn đặt ${don.dat_luc} hiện ${don.trang_thai} ạ.`);
}

import { chayAgent } from "./agent.ts";
import { coKey, MODEL } from "./goi-llm.ts";

const CAU_HOI = [
  "Quán mở cửa mấy giờ vậy ạ?",
  "Đơn 1041 của em giao chưa ạ?",
  "Đơn #1099 sao chưa thấy ạ",
  "Sáng nay em đặt mà chưa thấy ai gọi, sđt 0901000001",
  "Em đặt sáng nay, sđt 0901000001. Nếu đơn chưa giao thì thêm giúp em một bạc xỉu nhé",
  "Thêm giúp em một bánh mì trứng vào đơn 1042 nhé",
];

console.log(coKey() ? `Gọi Claude thật (${MODEL})\n` : "Chế độ OFFLINE (giả lập, không phải mô hình thật)\n");

for (const cauHoi of CAU_HOI) {
  const kq = await chayAgent(cauHoi);
  console.log(`Khách: ${cauHoi}`);
  for (const dong of kq.nhatKy) console.log(`  ${dong}`);
  console.log(`Mộc:   ${kq.traLoi}`);
  console.log(`       (${kq.soLanGoi} lần gọi mô hình)\n`);
}

Chế độ OFFLINE (giả lập, không phải mô hình thật)

Khách: Quán mở cửa mấy giờ vậy ạ?
Mộc:   Dạ, quán mở cửa từ 6:30 đến 22:00 mỗi ngày.
       (1 lần gọi mô hình)

Khách: Đơn 1041 của em giao chưa ạ?
  lần 1: xem_don({"ma_don":"1041"}) → {"ma_don":"1041","mon":["cà phê sữa","bánh mì trứng"],"trang_thai":"đang pha chế","dat_luc":"hôm nay 8:12"}
Mộc:   Dạ đơn 1041 (cà phê sữa, bánh mì trứng) bạn đặt hôm nay 8:12 hiện đang pha chế ạ.
       (2 lần gọi mô hình)

Khách: Đơn #1099 sao chưa thấy ạ
  lần 1: xem_don({"ma_don":"1099"}) → Không có đơn 1099.  (is_error)
Mộc:   Dạ Mộc rất tiếc, không có đơn 1099. Bạn cần Mộc hỗ trợ gì thêm không ạ?
       (2 lần gọi mô hình)

Khách: Sáng nay em đặt mà chưa thấy ai gọi, sđt 0901000001
  lần 1: tim_don_theo_sdt({"so_dien_thoai":"0901000001"}) → {"ma_don":"1041"}
  lần 2: xem_don({"ma_don":"1041"}) → {"ma_don":"1041","mon":["cà phê sữa","bánh mì trứng"],"trang_thai":"đang pha chế","dat_luc":"hôm nay 8:12"}
Mộc:   Dạ đơn 1041 (cà phê sữa, bánh mì trứng) bạn đặt hôm nay 8:12 hiện đang pha chế ạ.
       (3 lần gọi mô hình)

Khách: Em đặt sáng nay, sđt 0901000001. Nếu đơn chưa giao thì thêm giúp em một bạc xỉu nhé
  lần 1: tim_don_theo_sdt({"so_dien_thoai":"0901000001"}) → {"ma_don":"1041"}
  lần 2: xem_don({"ma_don":"1041"}) → {"ma_don":"1041","mon":["cà phê sữa","bánh mì trứng"],"trang_thai":"đang pha chế","dat_luc":"hôm nay 8:12"}
  lần 3: them_mon({"ma_don":"1041","mon":"bạc xỉu","so_luong":1}) → {"ma_don":"1041","mon":["cà phê sữa","bánh mì trứng","bạc xỉu"],"trang_thai":"đang pha chế"}
Mộc:   Dạ Mộc đã thêm vào đơn 1041. Đơn giờ gồm: cà phê sữa, bánh mì trứng, bạc xỉu, vẫn đang pha chế ạ.
       (4 lần gọi mô hình)

Khách: Thêm giúp em một bánh mì trứng vào đơn 1042 nhé
  lần 1: them_mon({"ma_don":"1042","mon":"bánh mì trứng","so_luong":1}) → Đơn 1042 đang giao, không thêm món được nữa.  (is_error)
Mộc:   Dạ Mộc rất tiếc, đơn 1042 đang giao, không thêm món được nữa. Bạn cần Mộc hỗ trợ gì thêm không ạ?
       (2 lần gọi mô hình)

// Chạy câu hỏi cần 4 lần gọi mô hình, nhưng chỉ cho phép 2 lần, để xem agent dừng thế nào.
import { chayAgent } from "./agent.ts";

const cauHoi = "Em đặt sáng nay, sđt 0901000001. Nếu đơn chưa giao thì thêm giúp em một bạc xỉu nhé";
const kq = await chayAgent(cauHoi, 2);
console.log(`Khách: ${cauHoi}`);
for (const dong of kq.nhatKy) console.log(`  ${dong}`);
console.log(`Mộc:   ${kq.traLoi}`);

Khách: Em đặt sáng nay, sđt 0901000001. Nếu đơn chưa giao thì thêm giúp em một bạc xỉu nhé
  lần 1: tim_don_theo_sdt({"so_dien_thoai":"0901000001"}) → {"ma_don":"1041"}
  lần 2: xem_don({"ma_don":"1041"}) → {"ma_don":"1041","mon":["cà phê sữa","bánh mì trứng"],"trang_thai":"đang pha chế","dat_luc":"hôm nay 8:12"}
  dừng: đã gọi mô hình 2 lần mà chưa có câu trả lời cuối
Mộc:   Dạ yêu cầu này Mộc chưa xử lý xong, nhân viên sẽ kiểm tra và nhắn lại bạn ngay ạ.

// Khách nhắn hai tin liên tiếp. Mỗi tin là một lần chayAgent mới, với mảng messages mới.
import { chayAgent } from "./agent.ts";

for (const tinNhan of ["Đơn 1041 của em giao chưa ạ?", "Vậy thêm cho em một bạc xỉu nữa nhé"]) {
  const kq = await chayAgent(tinNhan);
  console.log(`Khách: ${tinNhan}`);
  for (const dong of kq.nhatKy) console.log(`  ${dong}`);
  console.log(`Mộc:   ${kq.traLoi}\n`);
}

Khách: Đơn 1041 của em giao chưa ạ?
  lần 1: xem_don({"ma_don":"1041"}) → {"ma_don":"1041","mon":["cà phê sữa","bánh mì trứng"],"trang_thai":"đang pha chế","dat_luc":"hôm nay 8:12"}
Mộc:   Dạ đơn 1041 (cà phê sữa, bánh mì trứng) bạn đặt hôm nay 8:12 hiện đang pha chế ạ.

Khách: Vậy thêm cho em một bạc xỉu nữa nhé
Mộc:   Dạ bạn cho Mộc xin mã đơn hoặc số điện thoại đặt hàng để Mộc thêm món nhé!