跳到主要内容

微信小程序 AI SDK

微信小程序基础库从 3.7.1 版本开始内置了云开发 AI+ 能力，开发者可以直接通过小程序中的 wx.cloud.extend.AI 调用。

初始化​

在使用基础库 AI+ 能力前，需要传入云开发环境进行初始化。

wx.cloud.init({
  env: "your-env-id"
});

初始化完毕后，即可通过 wx.cloud.extend.AI 使用云开发 AI+ 能力。

大模型​

AI.createModel()​

创建指定的 AI 模型。

使用示例​

const model = wx.cloud.extend.AI.createModel("hunyuan");

类型声明​

function createModel(model: string): ChatModel;

返回一个提供 AI 生成文本能力的对象。

ChatModel.streamText()​

以流式调用大模型生成文本。流式调用时，生成的文本及其他响应数据会通过 SSE 返回，该接口的返回值对 SEE 做了不同程度的封装，开发者能根据实际需求获取到文本流和完整数据流。

使用示例​

const hy = wx.cloud.extend.AI.createModel("hunyuan"); // 创建模型
const res = await hy.streamText({
  data: {
    model: "hunyuan-lite",
    messages: [
      {
        role: "user",
        content: "hi"
      }
    ]
  }
});

for await (let str of res.textStream) {
  console.log(str); // 打印生成的文本
}
for await (let event of res.eventStream) {
  console.log(event); // 打印每次返回的完整数据
}

类型声明​

function streamText(props: StreamTextInput): Promise<StreamTextResult>;

interface StreamTextResult {
  eventStream: EventStream;
  textStream: TextStream;
}

interface StreamTextInput {
  data: unknown;
  onEvent?: OnEvent;
  onText?: OnText;
  onFinish?: OnFinish;
}

interface OnEvent {
  (prop: { data: string }): unknown;
}

interface OnText {
  (text: string): unknown;
}

interface OnFinish {
  (text: string): unknown;
}

interface EventStream {
  [Symbol.asyncIterator](): AsyncIterator<{
    event?: unknown;
    id?: unknown;
    data: string;
  }>;
}

interface TextStream {
  [Symbol.asyncIterator](): AsyncIterator<string>;
}

参数​

参数名必填类型示例说明
props.data是unknown{model: "hunyuan-lite", messages: [{ role: "user", content: "你好，请你介绍一下李白" }]}各家大模型的调用参数不同，请根据实际调用模型传入正确的参数。
props.onText否(text: string) => unknown;(text) => console.log(text)接收到新文本返回时触发的回调函数，参数为增量的文本
props.onEvent否(prop: { data: string }) => unknown;({data}) => console.log(data)接收到新事件返回时触发的回调函数，参数为事件，prop.data 为此次事件包含的数据
props.onFinish否(text: string) => unknown;(text) => console.log(text)当本次调用完全结束时触发的回调函数，参数为本次调用返回的完整文本

返回值​

StreamTextResult 属性名类型说明
textStreamAsyncIterable<string>以流式返回的大模型生成文本，可参考使用示例获取到生成的增量文本。
eventStreamAsyncIterable<{data: string, event?: unknown, id?: unknown}>以流式返回的大模型响应数据，可参考使用示例获取到生成的增量数据。由于各家大模型响应值互有出入，请根据实际情况合理使用。

ChatModel.generateText()​

调用大模型生成文本。

使用示例​

const hy = wx.cloud.extend.AI.createModel("hunyuan"); // 创建模型
const res = await hy.generateText({
  model: "hunyuan-lite",
  messages: [{ role: "user", content: "你好" }],
});
console.log(res);
// {
//   "Response": {
//     "Created": 1735024715,
//     "Usage": {
//       "PromptTokens": 3,
//       "CompletionTokens": 30,
//       "TotalTokens": 33
//     },
//     "Note": "以上内容为AI生成，不代表开发者立场，请勿删除或修改本标记",
//     "Id": "2346914b-87ea-4618-b7b4-df760e6d2b52",
//     "Choices": [
//       {
//         "FinishReason": "stop",
//         "Message": {
//           "Role": "assistant",
//           "Content": "Hello! How can I help you today? If you have any questions, need information, or want to chat about something, feel free to ask!"
//         },
//         "Index": 0
//       }
//     ],
//     "RequestId": "2346914b-87ea-4618-b7b4-df760e6d2b52"
//   }
// }
console.log(res.Response.Choices[0].Message.Content);
// Hello! How can I help you today? If you have any questions, need information, or want to chat about something, feel free to ask!

类型声明​

function generateText(data: unknown): Promise<unknown>;

参数​

参数名必填类型示例说明
data是unknown{model: "hunyuan-lite", messages: [{ role: "user", content: "你好，请你介绍一下李白" }]}各家大模型的调用参数不同，请根据实际调用模型传入正确的参数。

返回值​

该接口直接返回实际调用大模型的响应值，需要根据实际响应内容解析出需要的数据。参见上文使用示例。

Agent​

AI.bot.get()​

获取某个 Agent 的信息。

使用示例​

await wx.cloud.extend.AI.bot.get({ botId: "botId-xxx" });

类型声明​

function get(props: { botId: string });

参数​

参数名必填类型说明
props.botId是string要获取信息的 Agent 的 id

AI.bot.list()​

批量获取多个 Agent 的信息。

使用示例​

await wx.cloud.extend.AI.bot.list({
  pageNumber: 1,
  pageSize: 10,
  name: "",
  enable: true,
  information: "",
  introduction: "",
});

类型声明​

function list(props: {
  name: string;
  introduction: string;
  information: string;
  enable: boolean;
  pageSize: number;
  pageNumber: number;
});

参数​

参数名必填类型说明
props.pageNumber是number分页大小
props.pageSize是number分页下标
props.enable是booleanAgent 是否启用
props.name是stringAgent 名字，用于模糊查询
props.information是stringAgent 信息，用于模糊查询
props.introduction是stringAgent 简介，用于模糊查询

AI.bot.sendMessage()​

与 Agent 进行对话。

使用示例​

const res = await wx.cloud.extend.AI.bot.sendMessage({
  data: {
    botId: 'xxx-bot-id',
    msg: "你是谁"
  }
})
for await (let x of res.textStream) {
  console.log(x)
}

类型声明​

function sendMessage(props: {
  data: {
    botId: string;
    msg: string;
    history: Array<{
      role: string;
      content: string;
    }>;
  },
  onText: OnText,
  onEvent: OnEvent,
  onFinish: OnFinish
});

参数​

参数名必填类型说明
props.data.botId是numberAgent id
props.data.msg是number此次对话要发送的消息
props.data.history是boolean在此次对话前发生的聊天记录
props.data.history[n].role是string本聊天信息的发送角色
props.data.history[n].content是string本聊天信息的内容
props.onText否(text: string) => unknown;接收到新文本返回时触发的回调函数，参数为增量的文本
props.onEvent否(prop: { data: string }) => unknown;接收到新事件返回时触发的回调函数，参数为事件，prop.data 为此次事件包含的数据的字符串
props.onFinish否(text: string) => unknown;当本次调用完全结束时触发的回调函数，参数为本次调用返回的完整文本

AI.bot.getChatRecords()​

获取聊天记录。

使用示例​

await wx.cloud.extend.AI.bot.getChatRecords({
  botId: "botId-xxx",
  pageNumber: 1,
  pageSize: 10,
  sort: "asc",
});

类型声明​

function getChatRecords(props: {
  botId: string;
  sort: string;
  pageSize: number;
  pageNumber: number;
});

参数​

参数名必填类型说明
props.botId是stringAgent id
props.sort是string排序方式
props.pageSize是number分页大小
props.pageNumber是number分页下标

AI.bot.sendFeedback()​

发送对某条聊天记录的反馈信息。

使用示例​

const res = await wx.cloud.extend.AI.bot.sendFeedback({
  userFeedback: {
    botId: "botId-xxx",
    recordId: "recordId-xxx",
    comment: "非常棒",
    rating: 5,
    tags: ["优美"],
    aiAnswer: "落英缤纷",
    input: "来个成语",
    type: "upvote",
  },
});

类型声明​

function sendFeedback(props: { userFeedback: IUserFeedback });

参数​

参数名必填类型说明
props.userFeedback是IUserFeedback用户反馈，详见 IUserFeedback 类型定义
props.botId是string将要获取聊天记录的 Agent 的 id
props.sort是string排序方式
props.pageSize是number分页大小
props.pageNumber是number分页下标

AI.bot.getFeedBack()​

获取已存在的反馈信息。

使用示例​

const res = await wx.cloud.extend.AI.bot.getFeedBack({
  botId: "botId-xxx",
  from: 0,
  to: 0,
  maxRating: 4,
  minRating: 3,
  pageNumber: 1,
  pageSize: 10,
  sender: "user-a",
  senderFilter: "include",
  type: "upvote",
});

类型声明​

function sendFeedback(props: {
  botId: string;
  type: string;
  sender: string;
  senderFilter: string;
  minRating: number;
  maxRating: number;
  from: number;
  to: number;
  pageSize: number;
  pageNumber: number;
});

参数​

参数名必填类型说明
props.botId是stringAgent id
props.type是string用户反馈类型，点赞 upvote 点踩 downvote
props.sender是string评论创建用户
props.senderFilter是string评论创建用户过滤关系 include:包含 exclude:不包含 equal:等于 unequal:不等于 prefix:前缀
props.minRating是number最低评分
props.maxRating是number最高评分
props.from是number开始时间戳
props.to是number结束时间戳
props.pageSize是number分页大小
props.pageNumber是number分页下标

AI.bot.getRecommendQuestions()​

获取推荐的问题。

使用示例​

const res = await wx.cloud.extend.AI.bot.getRecommendQuestions({
  data: {
    botId: "xxx-bot-id",
    msg: "你是谁"
  }
})
for await (let x of res.textStream) {
  console.log(x)
}

类型声明​

function getRecommendQuestions(props: {
  data: {
    botId: string;
    name: string;
    introduction: string;
    agentSetting: string;
    msg: string;
    history: Array<{
      role: string;
      content: string;
    }>;
  },
  onText: OnText,
  onEvent: OnEvent,
  onFinish: OnFinish
});

参数​

参数名必填类型说明
props.data.botId是stringAgent id
props.data.name是stringAgent 名称
props.data.introduction是stringAgent 简介
props.data.agentSetting是stringAgent 设定
props.data.msg是string用户发送信息
props.data.history是Array历史对话信息
props.data.history[n].role是string历史信息角色
props.data.history[n].content是string历史信息内容
props.onText否(text: string) => unknown;接收到新文本返回时触发的回调函数，参数为增量的文本
props.onEvent否(prop: { data: string }) => unknown;接收到新事件返回时触发的回调函数，参数为事件，prop.data 为此次事件包含的数据的字符串
props.onFinish否(text: string) => unknown;当本次调用完全结束时触发的回调函数，参数为本次调用返回的完整文本