Nhúng AI vào app của bạn

npx wrangler secret put ANTHROPIC_API_KEY

async function hoiClaude(env, cauHoi, boiCanh) {
  const r = await fetch("https://api.anthropic.com/v1/messages", {
    method: "POST",
    headers: {
      "x-api-key": env.ANTHROPIC_API_KEY,
      "anthropic-version": "2023-06-01",
      "content-type": "application/json",
    },
    body: JSON.stringify({
      model: "claude-sonnet-5-5",
      max_tokens: 1024,
      system: "Bạn trả lời ngắn gọn bằng tiếng Việt, chỉ dựa trên dữ liệu được cung cấp. " +
              "Nếu dữ liệu không có thông tin, nói thẳng là không có. Không đoán.",
      messages: [{ role: "user", content: `Dữ liệu:\n${boiCanh}\n\nCâu hỏi: ${cauHoi}` }],
    }),
    signal: AbortSignal.timeout(30000),          // luôn có hạn giờ
  });

  if (!r.ok) {
    console.log("Claude API loi:", r.status);
    return null;                                  // đường lui: app vẫn chạy
  }
  const j = await r.json();
  return j.content[0].text;
}

[ai]
binding = "AI"

const kq = await env.AI.run("@cf/meta/llama-3.1-8b-instruct", {
  messages: [{ role: "user", content: "Tóm tắt đoạn này trong 2 câu: " + vanBan }],
});

// --- Bước 1: lúc thêm tài liệu, cắt nhỏ rồi lưu vector ---
const doan = catDoan(vanBan, 800);                    // ~800 ký tự mỗi đoạn, có chồng lấn
for (const [i, d] of doan.entries()) {
  const { data } = await env.AI.run("@cf/baai/bge-m3", { text: [d] });
  await env.VEC.insert([{ id: `${docId}:${i}`, values: data[0], metadata: { docId, doan: d } }]);
}

// --- Bước 2: lúc hỏi, tìm đoạn liên quan ---
const { data } = await env.AI.run("@cf/baai/bge-m3", { text: [cauHoi] });
const kq = await env.VEC.query(data[0], { topK: 8, returnMetadata: true });

// --- Bước 3: đưa đoạn tìm được cho mô hình trả lời ---
const boiCanh = kq.matches.map(m => m.metadata.doan).join("\n---\n");
const traLoi = await hoiClaude(env, cauHoi, boiCanh);

// 1. Trần theo người dùng mỗi ngày
const khoa = `ai:${u.id}:${new Date().toISOString().slice(0, 10)}`;
const n = (await env.PHIEN.get(khoa, { type: "json" })) || 0;
if (n >= 50) return json({ loi: "Ban da dung het luot hoi hom nay" }, 429);
await env.PHIEN.put(khoa, JSON.stringify(n + 1), { expirationTtl: 86400 });

// 2. Chặn đầu vào quá dài
if (cauHoi.length > 2000) return json({ loi: "Cau hoi qua dai" }, 400);

// 3. Nhớ lại câu đã hỏi (rất hiệu quả — nhiều người hỏi giống nhau)
const khoaDem = "ai-dem:" + await bam(cauHoi.trim().toLowerCase());
const cu = await env.PHIEN.get(khoaDem);
if (cu) return json({ traLoi: cu, tuBoNho: true });

// 4. Ghi lại từng lượt để biết tiền đi đâu
await env.DB.prepare(
  "INSERT INTO nhat_ky_ai (nguoi_dung, cau_hoi, chu_vao, chu_ra, luc) VALUES (?,?,?,?,datetime('now'))"
).bind(u.id, cauHoi.slice(0, 200), soChuVao, soChuRa).run();