跳到主要内容

小程序调用

微信小程序通过 wx.cloud.extend.AI 调用 CloudBase AI 大模型，无需额外安装 SDK。

准备工作​

  • 微信基础库版本 ≥ 3.15.1
  • 已创建云开发环境（旧套餐可升级），并已启用模型开关（见 接入大模型）

初始化​

在 app.js 中初始化云开发：

App({
  onLaunch() {
    wx.cloud.init({
      env: "您的环境ID"
    });
  }
});

文本生成​

generateText() - 非流式​

一次性返回完整结果，适用于短文本生成。

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

const res = await model.generateText({
  model: "hy3",
  messages: [{ role: "user", content: "介绍一下李白" }],
});

// 返回值为原始模型响应
console.log(res.choices[0].message.content);
console.log(res.usage); // { prompt_tokens, completion_tokens, total_tokens }

返回值​

interface GenerateTextResponse {
  id: string;
  object: "chat.completion";
  created: number;
  model: string;
  choices: Array<{
    index: number;
    message: {
      role: "assistant";
      content: string;
    };
    finish_reason: string;
  }>;
  usage: {
    prompt_tokens: number;
    completion_tokens: number;
    total_tokens: number;
  };
}

streamText() - 流式​

流式返回文本，适用于实时对话、长文本生成场景。

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

const res = await model.streamText({
  data: {
    model: "hy3",
    messages: [{ role: "user", content: "介绍一下李白" }],
  }
});

// 使用 textStream 获取增量文本
for await (const text of res.textStream) {
  console.log("文本片段:", text);
}

参数​

参数类型必填说明
dataobject是请求参数，包含 model 和 messages
data.modelstring是模型名称
data.messagesarray是消息列表

返回值​

属性类型说明
textStreamAsyncIterable<string>增量文本流
eventStreamAsyncIterable<object>原始事件流（包含元数据）

检测流结束​

使用 eventStream 可获取完整事件信息，当返回 [DONE] 时表示流结束：

for await (const event of res.eventStream) {
  console.log(event);
  if (event.data === "[DONE]") {
    break;
  }
}

完整示例​

聊天页面​

Page({
  data: {
    messages: [],
    inputValue: "",
    isLoading: false
  },

  async sendMessage() {
    const { inputValue, messages } = this.data;
    if (!inputValue.trim() || this.data.isLoading) return;

    // 添加用户消息
    const userMessage = { role: "user", content: inputValue };
    const newMessages = [...messages, userMessage];
    
    this.setData({
      messages: [...newMessages, { role: "assistant", content: "" }],
      inputValue: "",
      isLoading: true
    });

    try {
      const model = wx.cloud.extend.AI.createModel("cloudbase");
      let assistantContent = "";

      const res = await model.streamText({
        data: {
          model: "hy3",
          messages: newMessages,
        }
      });

      // 使用 textStream 逐步更新 UI
      for await (const text of res.textStream) {
        assistantContent += text;
        this.setData({
          messages: [
            ...newMessages,
            { role: "assistant", content: assistantContent }
          ]
        });
      }
    } catch (error) {
      console.error("调用失败:", error);
      wx.showToast({ title: "请求失败", icon: "error" });
    } finally {
      this.setData({ isLoading: false });
    }
  }
});

与 Web SDK 的差异​

特性小程序Web SDK
命名空间wx.cloud.extend.AIapp.ai()
参数格式streamText 包裹在 data 对象中，generateText 直接传递直接传递
返回值原始响应 { choices, usage }封装后 { text, usage, messages }
图片生成不支持支持