@cloudbase/ai 3.0.0 → 3.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cloudbase/ai",
-  "version": "3.0.0",
+  "version": "3.0.1",
   "description": "cloudbase js sdk ai module",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",
@@ -30,8 +30,8 @@
   "author": "",
   "license": "Apache-2.0",
   "dependencies": {
-    "@cloudbase/types": "3.0.0",
-    "@cloudbase/utilities": "3.0.0",
+    "@cloudbase/types": "3.0.1",
+    "@cloudbase/utilities": "3.0.1",
     "@mattiasbuelens/web-streams-adapter": "^0.1.0",
     "text-encoding-shim": "^1.0.5",
     "web-streams-polyfill": "^4.0.0"
@@ -39,5 +39,5 @@
   "devDependencies": {
     "@cloudbase/adapter-interface": "^0.7.1"
   },
-  "gitHead": "a70e7266db7be901d909b1fa63c1651e2a7943e4"
+  "gitHead": "a29767afc9a2b16cd632074c7c751abcf81746ed"
 }

package/src/AI.ts

@@ -1,3 +1,5 @@
+/* eslint-disable eqeqeq */
+/* eslint-disable @typescript-eslint/naming-convention */
 import type { SDKRequestInterface } from '@cloudbase/adapter-interface'
 import { Bot } from './bot'
 import * as models from './models'
@@ -15,7 +17,11 @@ class AI {
   public i18n: ICloudbaseConfig['i18n']
   defaultHeaders: { [x: string]: any }
 
-  constructor(private req: SDKRequestInterface, public baseUrl: string, i18n: ICloudbaseConfig['i18n']) {
+  constructor(
+    private req: SDKRequestInterface,
+    public baseUrl: string,
+    i18n: ICloudbaseConfig['i18n'],
+  ) {
     this.aiBaseUrl = `${baseUrl}/ai`
     this.aiBotBaseUrl = `${baseUrl}/aibot`
     this.bot = new Bot(this.botRequest, this.aiBotBaseUrl)
@@ -81,6 +87,10 @@ class AI {
     MODELS[name] = model
   }
 
+  createImageModel<T extends 'hunyuan-image' | (string & {})>(provider: T) {
+    return new models.DefaultImageModel<T>(this.modelRequest, this.aiBaseUrl, provider)
+  }
+
   modelRequest: types.ModelReq = async ({ url, data, headers, stream, timeout }) => {
     const fetchHeaders = {
       'Content-Type': 'application/json',

package/src/bot/index.ts

@@ -34,6 +34,7 @@ export class Bot {
   req: BotReq
 
   constructor(req: BotReq, public baseUrl: string) {
+    // eslint-disable-next-line prefer-rest-params
     const token = arguments[2]
     if (typeof token === 'string') {
       this.req = ({ headers = {}, ...rest }) => req({ ...rest, headers: { ...headers, Authorization: `Bearer ${token}` } })
@@ -230,6 +231,7 @@ export class Bot {
 type BotEventStreamData = { content: string }
 
 class StreamResult {
+  // eslint-disable-next-line @typescript-eslint/naming-convention
   private _eventSourceStream: ReadableStream<ParsedEvent>
 
   constructor(_stream: ReadableStream<Uint8Array>) {

package/src/index.ts

@@ -1,3 +1,4 @@
+/* eslint-disable eqeqeq */
 import type { IFetchOptions, SDKRequestInterface } from '@cloudbase/adapter-interface'
 import type { ICloudbase, ICloudbaseConfig } from '@cloudbase/types'
 import type { ICloudbaseComponent } from '@cloudbase/types/component'
@@ -54,7 +55,7 @@ interface AIInitOption {
   baseUrl?: string
 }
 
-interface ICreateAi {
+interface ICreateAI {
   req: SDKRequestInterface
   baseUrl?: string
   env?: string
@@ -73,7 +74,7 @@ interface ICreateAi {
  *
  * 其他参数可按需传入覆盖函数逻辑。
  */
-function createAi({ env, baseUrl, req, getAccessToken, handleReqInstance, i18n }: ICreateAi) {
+function createAI({ env, baseUrl, req, getAccessToken, handleReqInstance, i18n }: ICreateAI) {
   const getBaseUrl = () => {
     if (baseUrl != null) {
       return baseUrl
@@ -97,10 +98,16 @@ function createAi({ env, baseUrl, req, getAccessToken, handleReqInstance, i18n }
   return new AI(getReq(), getBaseUrl(), i18n)
 }
 
+// 兼容
+/**
+ * @deprecated use `createAI`
+ */
+const createAi = createAI
+
 /**
  * 创建 `AI` 实例。
  * 该函数挂载在 js sdk 的 Cloudbase 实例上。
- * 该函数调用了本模块暴露的 `createAi` 函数，但调用时通过传入参数做了逻辑覆盖：
+ * 该函数调用了本模块暴露的 `createAI` 函数，但调用时通过传入参数做了逻辑覆盖：
  *
  * - 传入从 Cloudbase 实例上的中心的 `req`
  * - 传入从 Cloudbase 实例上拿到的 `baseUrl`
@@ -125,7 +132,7 @@ function ai(
 
   const baseUrl = options?.baseUrl ?? getUrlFromCloud()
 
-  return createAi({ req, baseUrl, handleReqInstance: ({ req }) => req, i18n: this.config.i18n })
+  return createAI({ req, baseUrl, handleReqInstance: ({ req }) => req, i18n: this.config.i18n })
 }
 
 const component: ICloudbaseComponent = {
@@ -141,5 +148,9 @@ function registerAi(app: ICloudbase) {
   }
 }
 
-export { AI, Bot, createAi, restModels as models, registerAi, utils }
+export { AI, Bot, createAi, createAI, restModels as models, registerAi, utils }
 export * from './type'
+
+try {
+  (window as any).registerAi = registerAi
+} catch (e) {}

package/src/models/Ark/index.ts

@@ -1,3 +1,5 @@
+/* eslint-disable eqeqeq */
+/* eslint-disable @typescript-eslint/naming-convention */
 import {
   createAsyncIterable,
   TransformStream,

package/src/models/DashScope/index.ts

@@ -1,3 +1,5 @@
+/* eslint-disable eqeqeq */
+/* eslint-disable @typescript-eslint/naming-convention */
 import {
   createAsyncIterable,
   TransformStream,

package/src/models/DeepSeek/index.ts

@@ -1,3 +1,4 @@
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,
@@ -42,6 +43,7 @@ export class DeepSeekSimpleModel implements SimpleChatModel {
 
   public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
+    // eslint-disable-next-line @typescript-eslint/naming-convention
     const _stream = await this.req({
       url: this.url,
       data: {

package/src/models/Default/index.ts

@@ -33,6 +33,7 @@ export class DefaultSimpleModel implements SimpleChatModel {
   public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
 
+    // eslint-disable-next-line @typescript-eslint/naming-convention
     const _stream = await this.req({
       url: this.url,
       data: {
@@ -48,6 +49,7 @@ export class DefaultSimpleModel implements SimpleChatModel {
       transform(chunk, controller) {
         const newChoices = chunk.choices.map((choice) => {
           const message = choice.delta
+          // eslint-disable-next-line eqeqeq
           if (isToolCall == null) isToolCall = isToolCallAssistantMessage(message)
           if (isToolCall) {
             return {

package/src/models/HunYuan/index.ts

@@ -1,3 +1,4 @@
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,
@@ -52,6 +53,7 @@ export class HunYuanSimpleModel implements SimpleChatModel {
   public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
 
+    // eslint-disable-next-line @typescript-eslint/naming-convention
     const _stream = await this.req({
       url: this.url,
       headers: {

package/src/models/HunYuan/util.ts

@@ -1,3 +1,5 @@
+/* eslint-disable eqeqeq */
+/* eslint-disable @typescript-eslint/naming-convention */
 import { HunYuanInputData } from './type'
 import { BaseChatModelInput } from '../../type'
 

package/src/models/HunYuanBeta/index.ts

@@ -1,3 +1,5 @@
+/* eslint-disable @typescript-eslint/naming-convention */
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,

package/src/models/HunYuanExp/index.ts

@@ -1,3 +1,5 @@
+/* eslint-disable @typescript-eslint/naming-convention */
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,

package/src/models/HunYuanOpen/index.ts

@@ -1,3 +1,4 @@
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,
@@ -45,6 +46,7 @@ export class HunYuanOpenSimpleModel implements SimpleChatModel {
   public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
 
+    // eslint-disable-next-line @typescript-eslint/naming-convention
     const _stream = await this.req({
       url: this.url,
       data: {

package/src/models/Moonshot/index.ts

@@ -1,3 +1,4 @@
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,
@@ -17,6 +18,7 @@ import type {
 } from '../../type'
 
 function processInput(input: BaseChatModelInput): MoonshotInputData {
+  // eslint-disable-next-line @typescript-eslint/naming-convention
   const { messages, model, temperature, tools, top_p } = input
 
   return {
@@ -56,6 +58,7 @@ export class MoonshotSimpleModel implements SimpleChatModel {
 
   public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
+    // eslint-disable-next-line @typescript-eslint/naming-convention
     const _stream = await this.req({
       url: this.url,
       data: {

package/src/models/Yi/index.ts

@@ -1,3 +1,5 @@
+/* eslint-disable @typescript-eslint/naming-convention */
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,

package/src/models/ZhiPu/index.ts

@@ -1,3 +1,5 @@
+/* eslint-disable eqeqeq */
+/* eslint-disable @typescript-eslint/naming-convention */
 import {
   createAsyncIterable,
   TransformStream,

package/src/models/image-model.ts

@@ -0,0 +1,59 @@
+import type {
+  ModelReq,
+  // ImageEditInput,
+  HunyuanGenerateImageInput,
+  HunyuanGenerateImageOutput,
+  HunyuanARGenerateImageInput,
+  HunyuanARGenerateImageOutput,
+  // HunyuanEditImageInput,
+  // HunyuanEditImageOutput,
+} from '../type'
+
+interface HunyuanImageGenerateFn {
+  (input: HunyuanARGenerateImageInput): Promise<HunyuanARGenerateImageOutput>
+  (input: HunyuanGenerateImageInput): Promise<HunyuanGenerateImageOutput>
+}
+
+type GenerateImageFn<TProvider> = TProvider extends 'hunyuan-image'
+  ? HunyuanImageGenerateFn
+  : (input: unknown) => Promise<unknown>
+
+export class DefaultImageModel<TProvider extends string> {
+  public defaultGenerateImageSubUrl = 'images/generations'
+
+  public generateImageSubUrlConfig: Record<string, Record<string, string>> = {
+    'hunyuan-image': {
+      'hunyuan-image-v3.0-v1.0.4': 'images/ar/generations',
+      'hunyuan-image-v3.0-v1.0.1': 'images/ar/generations',
+    },
+  }
+
+  /** 生成图片 */
+  public generateImage: GenerateImageFn<TProvider>
+
+  constructor(private req: ModelReq, public baseUrl: string, public provider: TProvider) {
+    this.generateImage = this.generateImageImpl.bind(this) as GenerateImageFn<TProvider>
+  }
+
+  /** generateImage 的实际实现 */
+  private async generateImageImpl(input: HunyuanARGenerateImageInput | HunyuanGenerateImageInput,): Promise<HunyuanARGenerateImageOutput | HunyuanGenerateImageOutput> {
+    let subUrl = this.defaultGenerateImageSubUrl
+
+    if (typeof input === 'object' && input && 'model' in input && typeof input.model === 'string') {
+      const newSubUrl = this.generateImageSubUrlConfig[this.provider]?.[input.model]
+      if (newSubUrl) subUrl = newSubUrl
+    }
+
+    const res = (await this.req({
+      url: this.getGenerateUrl(subUrl),
+      data: input,
+      stream: false,
+    })) as HunyuanARGenerateImageOutput | HunyuanGenerateImageOutput
+
+    return res
+  }
+
+  private getGenerateUrl(subUrl: string) {
+    return `${this.baseUrl}/${this.provider}/${subUrl}`
+  }
+}

package/src/models/index.ts

@@ -9,6 +9,7 @@ import { HunYuanExpSimpleModel } from './HunYuanExp/index'
 import { HunYuanOpenSimpleModel } from './HunYuanOpen/index'
 import { DeepSeekSimpleModel } from './DeepSeek'
 import { DefaultSimpleModel } from './Default'
+import { DefaultImageModel } from './image-model'
 
 export const MODELS = {
   hunyuan: HunYuanSimpleModel,
@@ -34,6 +35,7 @@ export {
   HunYuanExpSimpleModel,
   HunYuanOpenSimpleModel,
   DefaultSimpleModel,
+  DefaultImageModel,
 }
 
 export { ReactModel, toolMap } from './model'

package/src/models/model.ts

@@ -1,3 +1,5 @@
+/* eslint-disable @typescript-eslint/naming-convention */
+/* eslint-disable eqeqeq */
 import {
   createAsyncIterable,
   TransformStream,

package/src/type.ts

@@ -20,7 +20,7 @@ export interface IModelReqInput {
   url: string
   headers?: Record<string, string>
   data?: Object
-  stream?: boolean,
+  stream?: boolean
   timeout?: number
 }
 
@@ -145,3 +145,205 @@ export interface BaseDoStreamOutputChunk {
   usage?: Usage
 }
 // #endregion
+
+// #region Image Model Types
+/**
+ * 图片生成输入参数
+ */
+export interface ImageGenerateInput {
+  model: string // 模型名称，例如 'hunyuan-image'、'dall-e-3'
+  prompt: string // 用于描述要生成的图片内容
+  size?: string // 尺寸
+  n?: number // 张数
+  [key: string]: unknown // 允许额外的厂商特定参数
+}
+
+/**
+ * 图片编辑输入参数
+ */
+export interface ImageEditInput {
+  model: string
+  prompt: string
+  size?: string // 尺寸
+  n?: number // 张数
+  [key: string]: unknown // 允许额外的厂商特定参数
+}
+
+export interface HunyuanGenerateImageInput {
+  model: 'hunyuan-image'
+  /** 用来生成图像的文本描述 */
+  prompt: string
+  /**
+   * 模型版本，支持 v1.8.1 和 v1.9，默认版本 v1.8.1
+   * - v1.8.1: 采用二阶段模型支持文生图任务，针对图像细节、稳定性等要求高的任务较为合适
+   * - v1.9: 采用模型矩阵能力支持文生图任务，后端由人像、游戏、通用模型，以及各类功能插件组成，灵活适配多种文生图任务
+   */
+  version?: 'v1.8.1' | 'v1.9' | (string & {})
+  /**
+   * 图片尺寸，默认 "1024x1024"
+   * v1.9 仅支持以下几个尺寸：
+   * - 1024x1024
+   * - 1024x768, 1152x864
+   * - 768x1024, 864x1152
+   * - 1280x768 (16:9)
+   * - 768x1280 (9:16)
+   * - 1280x720 (16:9)
+   * - 720x1280 (9:16)
+   *
+   * 支持用户自定义尺寸，但输入尺寸只支持 768-1280 之间且为 64 的倍数的数字
+   */
+  size?: string
+  /**
+   * 仅 v1.9 支持，负向词
+   * 如果不传则使用各个模型内部的默认负向词
+   */
+  negative_prompt?: string
+  /**
+   * 仅 v1.9 支持，可指定风格：
+   * - 古风二次元风格
+   * - 都市二次元风格
+   * - 悬疑风格
+   * - 校园风格
+   * - 都市异能风格
+   */
+  style?: '古风二次元风格' | '都市二次元风格' | '悬疑风格' | '校园风格' | '都市异能风格' | (string & {})
+  /**
+   * 为 true 时对 prompt 进行改写，实际生成的图片会使用改写后的 prompt 进行生成
+   * 多数场景可提升生成的图片效果，默认为 true
+   */
+  revise?: boolean
+  /**
+   * 生成图片个数，默认为 1
+   * 目前只支持生成一张，固定为 1，传其他的值会报错
+   */
+  n?: number
+  /**
+   * 业务自定义水印内容
+   * 限制 16 个字符长度（不区分中英文），生成在图片右下角
+   */
+  footnote?: string
+  /**
+   * 生成种子，仅当生成图片数为 1 时生效
+   * 范围 [1, 4294967295]，不传时默认随机
+   */
+  seed?: number
+}
+
+export interface HunyuanARGenerateImageInput {
+  /** 模型名称：hunyuan-image-v3.0-v1.0.4（推荐）或 hunyuan-image-v3.0-v1.0.1 */
+  model: 'hunyuan-image-v3.0-v1.0.4' | 'hunyuan-image-v3.0-v1.0.1'
+  /** 生成图片使用的文本，不超过 8192 字符 */
+  prompt: string
+  /**
+   * 图片尺寸，格式 "${宽}x${高}"，默认 "1024x1024"
+   * hunyuan-image-v3.0-v1.0.4：宽高 [512, 2048]，面积不超过 1024x1024
+   * hunyuan-image-v3.0-v1.0.1：支持固定尺寸列表
+   */
+  size?: string
+  /** 生成种子，仅当生成图片数为 1 时生效，范围 [1, 4294967295] */
+  seed?: number
+  /** 业务自定义水印内容，限制 16 字符，生成在图片右下角 */
+  footnote?: string
+  /** 是否对 prompt 改写，默认开启。改写会增加约 30s 耗时 */
+  revise?: { value: boolean }
+  /** 改写是否开启 thinking 模式，默认开启。开启后效果提升但耗时增加（最大 60s） */
+  enable_thinking?: { value: boolean }
+}
+
+export interface HunyuanARGenerateImageOutput {
+  /** 此次请求的 id */
+  id: string
+  /** unix 时间戳 */
+  created: number
+  /** 返回的图片生成内容 */
+  data: Array<{
+    /** 生成的图片 url，有效期 24 小时 */
+    url: string
+    /** 改写后的 prompt */
+    revised_prompt?: string
+  }>
+}
+
+export type HunyuanEditImageInput = {
+  /** 模型名称，本期为 hunyuan-image，即为商品图编辑场景 */
+  model: 'hunyuan-image' | (string & {})
+  /** 编辑的文本描述 */
+  prompt: string
+  /** 模型版本 */
+  version?: string
+  /**
+   * 图片尺寸，默认 "800x800"
+   * 支持用户自定义尺寸，但输入尺寸只支持 512-1280 之间
+   * 目前只支持 1:1、16:9、9:16 三个尺寸
+   */
+  size?: string
+  /**
+   * 生成图片个数，默认为 1
+   * 目前只支持生成一张，固定为 1，传其他的值会报错
+   */
+  n?: number
+  /**
+   * beta 参数，随机种子
+   * 范围 [1, 4294967295]，不传时默认随机
+   */
+  seed?: number
+  /**
+   * 业务自定义角标内容
+   * 限制 16 个字符长度（不区分中英文）
+   */
+  footnote?: string
+  /**
+   * 图 base64 编码，大小不超过 6M
+   * 支持 jpg/jpeg 格式，和 image_url 二选一
+   * 长短边比例要求 2:1 内
+   */
+  image?: string
+  /**
+   * 图片 url
+   * 长短边比例要求 2:1 内，和 image 二选一
+   */
+  image_url?: string
+} & (
+  | {
+    mask: string
+    mask_url?: never
+  }
+  | {
+    mask?: never
+    mask_url: string
+  }
+)
+
+/**
+ * 混元图片生成输出
+ */
+export interface HunyuanGenerateImageOutput {
+  /** 此次请求的 id */
+  id: string
+  /** unix 时间戳 */
+  created: number
+  /** 返回的图片生成内容 */
+  data: Array<{
+    /** 生成的图片 url，有效期为 24 小时 */
+    url: string
+    /** 原 prompt 改写后的文本。若 revise 为 false，则为原 prompt */
+    revised_prompt?: string
+  }>
+}
+
+/**
+ * 混元图片编辑输出
+ */
+export interface HunyuanEditImageOutput {
+  /** 此次请求的 id */
+  id: string
+  /** unix 时间戳 */
+  created: number
+  /** 返回的图片生成内容 */
+  data: Array<{
+    /** 生成的图片 url，有效期为 24 小时 */
+    url: string
+  }>
+}
+
+// #endregion

package/src/utils.ts

@@ -103,8 +103,9 @@ export class TextDecoderStream {
 }
 
 export function createAsyncIterable<T>(stream: ReadableStream<T>) {
+  // eslint-disable-next-line @typescript-eslint/naming-convention
   const _stream = stream as AsyncIterableReadableStream<T>
-  _stream[Symbol.asyncIterator] = () => {
+  (_stream as any)[Symbol.asyncIterator] = () => {
     const reader = stream.getReader()
     return {
       async next(): Promise<IteratorResult<T>> {
@@ -166,6 +167,7 @@ export function createPromise<T = unknown>() {
 }
 
 export function isToolCallAssistantMessage(message: ChatModelMessage): message is ToolCallAssistantMessage {
+  // eslint-disable-next-line eqeqeq
   return message.role === 'assistant' && 'tool_calls' in message && message.tool_calls?.[0] != null
 }
 

package/tsconfig.json

@@ -1,7 +1,7 @@
 {
   "extends": "../../tsconfig.base.json",
   "compilerOptions": {
-    "module": "commonjs",
+    "module": "node16",
     "outDir": "dist/cjs",
     "rootDir": "src",
   },