@cloudbase/ai 2.12.0 → 2.13.0

package/src/bot/index.ts

@@ -1,6 +1,9 @@
 import { type ParsedEvent } from '../eventsource_parser'
 import { BotReq, ReqOptions } from '../type'
 import {
+  IBotCreateConversation,
+  IBotDeleteConversation,
+  IBotGetConversation,
   IBotPreview,
   IBotSendMessage,
   IBotUploadFiles,
@@ -111,6 +114,36 @@ export class Bot {
     })
   }
 
+  async createConversation(props: IBotCreateConversation, options?: ReqOptions) {
+    return this.req({
+      method: 'post',
+      url: this.join('conversation'),
+      data: props,
+      timeout: options?.timeout,
+    })
+  }
+
+  async getConversation({ pageSize = 10, pageNumber = 1, ...rest }: IBotGetConversation, options?: ReqOptions) {
+    if (pageNumber < 1) throw new Error('pageNumber must be greater than 0')
+    const offset = pageSize * (pageNumber - 1)
+    const limit = pageSize
+    return this.req({
+      method: 'get',
+      url: this.join('conversation'),
+      data: { ...rest, offset, limit },
+      timeout: options?.timeout,
+    })
+  }
+
+  async deleteConversation(props: IBotDeleteConversation, options?: ReqOptions) {
+    return this.req({
+      method: 'delete',
+      url: this.join(`conversation/${props.conversationId}`),
+      data: props,
+      timeout: options?.timeout,
+    })
+  }
+
   async getRecommendQuestions(props: IGetBotRecommendQuestions, options?: ReqOptions) {
     const res = await this.req({
       method: 'post',

package/src/bot/types.ts

@@ -99,6 +99,23 @@ export interface IBotUploadFiles {
   }>
 }
 
+export interface IBotCreateConversation {
+  botId: string
+  title?: string
+}
+
+export interface IBotGetConversation {
+  botId: string
+  pageSize?: number
+  pageNumber?: number
+  isDefault?: boolean
+}
+
+export interface IBotDeleteConversation {
+  botId: string
+  conversationId: string
+}
+
 export interface IUserFeedback {
   recordId: string
   type: string

package/src/models/model.ts

@@ -18,6 +18,7 @@ import {
   ToolCallAssistantMessage,
   ModelTool,
   FunctionTool,
+  ReqOptions,
 } from '../type'
 
 type ReactModelInput = ReactProps &
@@ -69,7 +70,7 @@ function processInput(obj: ReactModelInput): [ReactProps, BaseChatModelInput] {
 export class ReactModel {
   constructor(private model: SimpleChatModel) {}
 
-  public async generateText(_input: ReactModelInput): Promise<{
+  public async generateText(_input: ReactModelInput, options?: ReqOptions): Promise<{
     text: string
     messages: Array<ChatModelMessage>
     usage: Usage
@@ -81,7 +82,7 @@ export class ReactModel {
 
     const [{ onStepFinish, maxSteps = 10 }, input] = processInput(_input)
 
-    const doGenerate = () => this.model.doGenerate(input) // 后续代码会直接对 input.messages 原地修改，这里一直用同一个对象就行
+    const doGenerate = () => this.model.doGenerate(input, options) // 后续代码会直接对 input.messages 原地修改，这里一直用同一个对象就行
     let currentRes = await doGenerate()
     let currentStep = 1
     currentRes.rawResponse && rawResponses.push(currentRes.rawResponse)
@@ -152,7 +153,7 @@ export class ReactModel {
     }
   }
 
-  public async streamText(_input: ReactModelInput): Promise<{
+  public async streamText(_input: ReactModelInput, options?: ReqOptions): Promise<{
     dataStream: DoStreamOutput
     textStream: AsyncIterableReadableStream<string>
     messages: Promise<Array<ChatModelMessage>>
@@ -162,7 +163,7 @@ export class ReactModel {
     const totalUsage: Usage = { completion_tokens: 0, prompt_tokens: 0, total_tokens: 0 }
 
     const [{ onStepFinish, maxSteps = 10 }, input] = processInput(_input)
-    const doStream = () => this.model.doStream(input) // 后续代码会直接对 input.messages 原地修改，这里一直用同一个对象就行
+    const doStream = () => this.model.doStream(input, options) // 后续代码会直接对 input.messages 原地修改，这里一直用同一个对象就行
     let currentRes = await doStream()
     const currentStep = 1
     let readResult: { message: ToolCallAssistantMessage; usage: Usage } | null = null

package/src/type.ts

@@ -1,7 +1,7 @@
 export abstract class SimpleChatModel {
-  public abstract doGenerate(data: BaseChatModelInput): Promise<DoGenerateOutput>
+  public abstract doGenerate(data: BaseChatModelInput, options?: ReqOptions): Promise<DoGenerateOutput>
 
-  public abstract doStream(data: BaseChatModelInput): Promise<DoStreamOutput>
+  public abstract doStream(data: BaseChatModelInput, options?: ReqOptions): Promise<DoStreamOutput>
 }
 
 type RawResponse = { rawResponse?: any }