开发 SDK

npm install @xpert-ai/xpert-sdk

import { Client, type ChatSendRequest } from "@xpert-ai/xpert-sdk";

async function main() {
  const apiKey = process.env.XPERTAI_API_KEY;
  const xpertId = process.env.XPERTAI_XPERT_ID;
  if (!apiKey || !xpertId) {
    throw new Error("Set XPERTAI_API_KEY and XPERTAI_XPERT_ID");
  }

  const client = new Client({
    apiUrl: process.env.XPERTAI_API_URL ?? "https://api.xpertai.cn/api/ai",
    apiKey,
  });
  const thread = await client.threads.create({ assistantId: xpertId });
  const input: ChatSendRequest = {
    action: "send",
    message: { input: { input: "你好，请介绍一下你能做什么。" } },
  };
  const result = await client.runs.wait(thread.thread_id, xpertId, { input });

  console.log("thread_id:", thread.thread_id);
  console.log(result);
}

main().catch((error) => {
  console.error(error);
  process.exitCode = 1;
});

for await (const chunk of client.runs.stream(thread.thread_id, xpertId, { input })) {
  console.log(chunk.event, chunk.data);
}

for await (const chunk of client.runs.stream(thread.thread_id, xpertId, { input })) {
  const event = chunk.data;
  if (!event || typeof event !== "object" ||
      !("type" in event) || event.type !== "message" || !("data" in event)) continue;

  const data = event.data;
  if (typeof data === "string") {
    process.stdout.write(data);
  } else if (data && typeof data === "object" &&
             "type" in data && data.type === "text" &&
             "text" in data && typeof data.text === "string") {
    process.stdout.write(data.text);
  }
}

const xperts = await client.assistants.search({ limit: 10, offset: 0 });
console.table(xperts.map(({ assistant_id, name }) => ({ assistant_id, name })));
const xpert = await client.assistants.get(xpertId);
console.log(xpert.assistant_id, xpert.name);

const { randomUUID } = await import("node:crypto");
const customThread = await client.threads.create({
  assistantId: xpertId,
  threadId: randomUUID(),
  ifExists: "raise",
});
console.log(customThread.thread_id);

const threads = await client.threads.search({ limit: 10, offset: 0 });
const currentThread = await client.threads.get(thread.thread_id);
const state = await client.threads.getState(thread.thread_id);
console.log(threads, currentThread, state);

const followUp: ChatSendRequest = {
  action: "send",
  message: { input: { input: "请继续刚才的话题。" } },
};
const nextResult = await client.runs.wait(thread.thread_id, xpertId, {
  input: followUp,
});
console.log(nextResult);

const run = await client.runs.create(thread.thread_id, xpertId, { input });
console.log(run.run_id);

const currentRun = await client.runs.get(thread.thread_id, run.run_id);
const runs = await client.runs.list(thread.thread_id, { limit: 10, offset: 0 });
console.log(currentRun.status, runs);

await client.runs.cancel(thread.thread_id, run.run_id);

for await (const chunk of client.runs.joinStream(thread.thread_id, run.run_id)) {
  console.log(chunk.event, chunk.data);
}

const { readFile } = await import("node:fs/promises");
const buffer = await readFile("./manual.pdf");
const file = await client.contexts.uploadFile<{ id: string; url?: string }>(
  new Blob([new Uint8Array(buffer)]),
  { filename: "manual.pdf" },
);
const attachmentInput: ChatSendRequest = {
  action: "send",
  message: {
    input: { input: "请总结附件重点。", files: [file] },
  },
};
const attachmentResult = await client.runs.wait(thread.thread_id, xpertId, {
  input: attachmentInput,
});
console.log(attachmentResult);

const namespace = [xpertId, "sdk-example"];
await client.store.putItem(namespace, "key1", { value: 42 });
const item = await client.store.getItem(namespace, "key1");
console.log(item);

const namespace = [xpertId, "sdk-example"];
const namespaces = await client.store.listNamespaces({
  prefix: [xpertId],
  maxDepth: 2,
  limit: 10,
});
const found = await client.store.searchItems(namespace, { limit: 10 });
console.log(namespaces, found.items);