@cloudbase/ai 2.11.0 → 2.12.0

package/src/bot/index.ts

@@ -1,5 +1,5 @@
 import { type ParsedEvent } from '../eventsource_parser'
-import { BotReq } from '../type'
+import { BotReq, ReqOptions } from '../type'
 import {
   IBotPreview,
   IBotSendMessage,
@@ -36,91 +36,125 @@ export class Bot {
     }
   }
 
-  list(props: IGetBotList) {
+  list(props: IGetBotList, options?: ReqOptions) {
     return this.req({
       method: 'get',
       url: this.join('bots'),
       data: props,
+      timeout: options?.timeout,
     })
   }
 
-  create({ botInfo }: ICreateBot) {
+  create({ botInfo }: ICreateBot, options?: ReqOptions) {
     return this.req({
       method: 'post',
       url: this.join('bots'),
       data: botInfo,
+      timeout: options?.timeout,
     })
   }
 
-  get({ botId }: IGetBot) {
+  get({ botId }: IGetBot, options?: ReqOptions) {
     return this.req({
       method: 'get',
       url: this.join(`bots/${botId}`),
+      timeout: options?.timeout,
     })
   }
 
-  update({ botId, botInfo }: IUpdateBot) {
+  update({ botId, botInfo }: IUpdateBot, options?: ReqOptions) {
     return this.req({
       method: 'PATCH',
       url: this.join(`bots/${botId}`),
       data: botInfo,
+      timeout: options?.timeout,
     })
   }
 
-  delete({ botId }: IDeleteBot) {
-    return this.req({ method: 'delete', url: this.join(`bots/${botId}`) })
+  delete({ botId }: IDeleteBot, options?: ReqOptions) {
+    return this.req({ method: 'delete', url: this.join(`bots/${botId}`), timeout: options?.timeout })
   }
 
-  getChatRecords(props: IGetBotChatRecords) {
-    return this.req({ method: 'get', url: this.join(`bots/${props.botId}/records`), data: props })
+  getChatRecords(props: IGetBotChatRecords, options?: ReqOptions) {
+    return this.req({
+      method: 'get',
+      url: this.join(`bots/${props.botId}/records`),
+      data: props,
+      timeout: options?.timeout,
+    })
   }
 
-  sendFeedback({ userFeedback }: ISendBotFeedback) {
-    return this.req({ method: 'post', url: this.join(`bots/${userFeedback.botId}/feedback`), data: userFeedback })
+  sendFeedback({ userFeedback }: ISendBotFeedback, options?: ReqOptions) {
+    return this.req({
+      method: 'post',
+      url: this.join(`bots/${userFeedback.botId}/feedback`),
+      data: userFeedback,
+      timeout: options?.timeout,
+    })
   }
 
-  getFeedback(props: IGetBotFeedback) {
-    return this.req({ method: 'get', url: this.join(`bots/${props.botId}/feedback`), data: props })
+  getFeedback(props: IGetBotFeedback, options?: ReqOptions) {
+    return this.req({
+      method: 'get',
+      url: this.join(`bots/${props.botId}/feedback`),
+      data: props,
+      timeout: options?.timeout,
+    })
   }
 
-  async uploadFiles(props: IBotUploadFiles) {
+  async uploadFiles(props: IBotUploadFiles, options?: ReqOptions) {
     return this.req({
       method: 'post',
       url: this.join(`bots/${props.botId}/files`),
       data: props,
+      timeout: options?.timeout,
     })
   }
 
-  async getRecommendQuestions(props: IGetBotRecommendQuestions) {
+  async getRecommendQuestions(props: IGetBotRecommendQuestions, options?: ReqOptions) {
     const res = await this.req({
       method: 'post',
       url: this.join(`bots/${props.botId}/recommend-questions`),
       data: props,
       stream: true,
+      timeout: options?.timeout,
     })
     return new StreamResult(res)
   }
 
-  async generateBot(props: IGenerateBot) {
-    const res = await this.req({ method: 'post', url: this.join('generate-bot'), data: props, stream: true })
+  async generateBot(props: IGenerateBot, options?: ReqOptions) {
+    const res = await this.req({
+      method: 'post',
+      url: this.join('generate-bot'),
+      data: props,
+      stream: true,
+      timeout: options?.timeout,
+    })
     return new StreamResult(res)
   }
 
-  async getPreview(props: IBotPreview) {
-    const res = await this.req({ method: 'post', url: this.join('preview'), data: props, stream: true })
+  async getPreview(props: IBotPreview, options?: ReqOptions) {
+    const res = await this.req({
+      method: 'post',
+      url: this.join('preview'),
+      data: props,
+      stream: true,
+      timeout: options?.timeout,
+    })
     return new StreamResult(res)
   }
 
-  generateImage(props: IGenerateImage) {
-    return this.req({ method: 'post', url: this.join('generate-image'), data: props })
+  generateImage(props: IGenerateImage, options?: ReqOptions) {
+    return this.req({ method: 'post', url: this.join('generate-image'), data: props, timeout: options?.timeout })
   }
 
-  async sendMessage(props: IBotSendMessage) {
+  async sendMessage(props: IBotSendMessage, options?: ReqOptions) {
     const res = await this.req({
       method: 'post',
       url: this.join(`bots/${props.botId}/send-message`),
       data: props,
       stream: true,
+      timeout: options?.timeout,
     })
     return new StreamResult(res)
   }

package/src/models/Ark/index.ts

@@ -13,6 +13,7 @@ import type {
   DoGenerateOutput,
   DoStreamOutput,
   BaseDoStreamOutputChunk,
+  ReqOptions,
 } from '../../type'
 
 function processInput(input: BaseChatModelInput): ArkInputData {
@@ -40,7 +41,7 @@ export class ArkSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       data: {
@@ -48,11 +49,12 @@ export class ArkSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as ArkGenerateTextOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
     const _stream = await this.req({
       url: this.url,
@@ -61,6 +63,7 @@ export class ArkSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/DashScope/index.ts

@@ -13,6 +13,7 @@ import type {
   DoStreamOutput,
   BaseDoStreamOutputChunk,
   SimpleChatModel,
+  ReqOptions,
 } from '../../type'
 
 function processInput(input: BaseChatModelInput): DSInput {
@@ -41,7 +42,7 @@ export class DSSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       data: {
@@ -49,11 +50,12 @@ export class DSSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as DSGenerateOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
     const _stream = await this.req({
       url: this.url,
@@ -62,6 +64,7 @@ export class DSSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/DeepSeek/index.ts

@@ -12,6 +12,7 @@ import type {
   DoStreamOutput,
   BaseDoStreamOutputChunk,
   DoGenerateOutput,
+  ReqOptions,
 } from '../../type'
 
 export class DeepSeekSimpleModel implements SimpleChatModel {
@@ -26,7 +27,7 @@ export class DeepSeekSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       data: {
@@ -34,11 +35,12 @@ export class DeepSeekSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as DoGenerateOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
     const _stream = await this.req({
       url: this.url,
@@ -47,6 +49,7 @@ export class DeepSeekSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/HunYuan/index.ts

@@ -12,6 +12,7 @@ import type {
   DoStreamOutput,
   SimpleChatModel,
   DoGenerateOutput,
+  ReqOptions,
 } from '../../type'
 import { processInput, titleCaseToSnakeCase } from './util'
 import { HunYuanGenerateTextOutput, HunYuanStreamTextOutput } from './type'
@@ -28,7 +29,7 @@ export class HunYuanSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       headers: {
@@ -39,6 +40,7 @@ export class HunYuanSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as { Response: any }
     const output = titleCaseToSnakeCase(res.Response) as HunYuanGenerateTextOutput
     return {
@@ -47,7 +49,7 @@ export class HunYuanSimpleModel implements SimpleChatModel {
     }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
 
     const _stream = await this.req({
@@ -60,6 +62,7 @@ export class HunYuanSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/HunYuanBeta/index.ts

@@ -12,6 +12,7 @@ import type {
   DoStreamOutput,
   BaseDoStreamOutputChunk,
   DoGenerateOutput,
+  ReqOptions,
 } from '../../type'
 import { processInput } from '../HunYuan/util'
 import { HunYuanGenerateTextOutput, HunYuanStreamTextOutput } from '../HunYuan/type'
@@ -28,7 +29,7 @@ export class HunYuanBetaSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       data: {
@@ -36,11 +37,12 @@ export class HunYuanBetaSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as HunYuanGenerateTextOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
 
     const _stream = await this.req({
@@ -50,6 +52,7 @@ export class HunYuanBetaSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/HunYuanOpen/index.ts

@@ -12,6 +12,7 @@ import type {
   DoStreamOutput,
   BaseDoStreamOutputChunk,
   DoGenerateOutput,
+  ReqOptions,
 } from '../../type'
 import { processInput } from '../HunYuan/util'
 import { HunYuanGenerateTextOutput, HunYuanStreamTextOutput } from '../HunYuan/type'
@@ -28,7 +29,7 @@ export class HunYuanOpenSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       data: {
@@ -36,11 +37,12 @@ export class HunYuanOpenSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as HunYuanGenerateTextOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
 
     const _stream = await this.req({
@@ -50,6 +52,7 @@ export class HunYuanOpenSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/Moonshot/index.ts

@@ -13,6 +13,7 @@ import type {
   DoStreamOutput,
   SimpleChatModel,
   BaseChatModelInput,
+  ReqOptions,
 } from '../../type'
 
 function processInput(input: BaseChatModelInput): MoonshotInputData {
@@ -40,7 +41,7 @@ export class MoonshotSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       data: {
@@ -48,11 +49,12 @@ export class MoonshotSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as MoonshotGenerateTextOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
     const _stream = await this.req({
       url: this.url,
@@ -61,6 +63,7 @@ export class MoonshotSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/Yi/index.ts

@@ -13,6 +13,7 @@ import type {
   DoStreamOutput,
   BaseDoStreamOutputChunk,
   SimpleChatModel,
+  ReqOptions,
 } from '../../type'
 
 function processInput(input: BaseChatModelInput): YiInput {
@@ -41,7 +42,7 @@ export class YiSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const res = (await this.req({
       url: this.url,
       data: {
@@ -49,11 +50,12 @@ export class YiSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as YiGenerateOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     let isToolCall: null | boolean = null
     const _stream = await this.req({
       url: this.url,
@@ -62,6 +64,7 @@ export class YiSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/models/ZhiPu/index.ts

@@ -13,6 +13,7 @@ import type {
   DoStreamOutput,
   BaseDoStreamOutputChunk,
   DoGenerateOutput,
+  ReqOptions,
 } from '../../type'
 
 function processInput(input: BaseChatModelInput): ZhiPuInputData {
@@ -48,7 +49,7 @@ export class ZhiPuSimpleModel implements SimpleChatModel {
     return `${this.baseUrl}/${this.subUrl}`
   }
 
-  public async doGenerate(_data: BaseChatModelInput): Promise<DoGenerateOutput> {
+  public async doGenerate(_data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput> {
     const data = processInput(_data)
     const res = (await this.req({
       url: this.url,
@@ -57,11 +58,12 @@ export class ZhiPuSimpleModel implements SimpleChatModel {
         stream: false,
       },
       stream: false,
+      timeout: options?.timeout,
     })) as ZhiPuGenerateTextOutput
     return { ...res, rawResponse: res }
   }
 
-  public async doStream(_data: BaseChatModelInput): Promise<DoStreamOutput> {
+  public async doStream(_data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput> {
     const data = processInput(_data)
     let isToolCall: null | boolean = null
     const _stream = await this.req({
@@ -71,6 +73,7 @@ export class ZhiPuSimpleModel implements SimpleChatModel {
         stream: true,
       },
       stream: true,
+      timeout: options?.timeout,
     })
     const stream = toPolyfillReadable(_stream) as typeof _stream
 

package/src/type.ts

@@ -20,7 +20,8 @@ export interface IModelReqInput {
   url: string
   headers?: Record<string, string>
   data?: Object
-  stream?: boolean
+  stream?: boolean,
+  timeout?: number
 }
 
 export type ModelReq = <T extends IModelReqInput>(
@@ -33,6 +34,11 @@ export interface IBotReqInput {
   headers?: Record<string, string>
   data?: Object
   stream?: boolean
+  timeout?: number
+}
+
+export interface ReqOptions {
+  timeout?: number
 }
 
 export type BotReq = <T extends IBotReqInput>(