@opentiny/tiny-robot-kit 0.2.0-alpha.0

package/src/error.ts

@@ -0,0 +1,100 @@
+/**
+ * 错误处理模块
+ * 用于统一处理各种错误情况并提供标准化的错误格式
+ */
+import { AIAdapterError, ErrorType } from './types'
+
+/**
+ * 创建标准化的AI适配器错误
+ * @param error 错误信息
+ * @returns 标准化的AI适配器错误
+ */
+export function createError(error: Partial<AIAdapterError>): AIAdapterError {
+  return {
+    type: error.type || ErrorType.UNKNOWN_ERROR,
+    message: error.message || '未知错误',
+    statusCode: error.statusCode,
+    originalError: error.originalError,
+  }
+}
+
+interface Error {
+  response?: object
+  code?: string
+  message?: string
+}
+
+/**
+ * 处理API请求错误
+ * @param error 原始错误
+ * @returns 标准化的AI适配器错误
+ */
+export function handleRequestError(error: Error): AIAdapterError {
+  // 网络错误
+  if (!error.response) {
+    return createError({
+      type: ErrorType.NETWORK_ERROR,
+      message: '网络连接错误，请检查您的网络连接',
+      originalError: error,
+    })
+  }
+
+  // 服务器返回的错误
+  if (error.response) {
+    const { status, data } = error.response
+
+    // 身份验证错误
+    if (status === 401 || status === 403) {
+      return createError({
+        type: ErrorType.AUTHENTICATION_ERROR,
+        message: '身份验证失败，请检查您的API密钥',
+        statusCode: status,
+        originalError: error,
+      })
+    }
+
+    // 速率限制错误
+    if (status === 429) {
+      return createError({
+        type: ErrorType.RATE_LIMIT_ERROR,
+        message: '超出API调用限制，请稍后再试',
+        statusCode: status,
+        originalError: error,
+      })
+    }
+
+    // 服务器错误
+    if (status >= 500) {
+      return createError({
+        type: ErrorType.SERVER_ERROR,
+        message: '服务器错误，请稍后再试',
+        statusCode: status,
+        originalError: error,
+      })
+    }
+
+    // 其他HTTP错误
+    return createError({
+      type: ErrorType.UNKNOWN_ERROR,
+      message: data?.error?.message || `请求失败，状态码: ${status}`,
+      statusCode: status,
+      originalError: error,
+    })
+  }
+
+  // 超时错误
+  if (error.code === 'ECONNABORTED') {
+    return createError({
+      type: ErrorType.TIMEOUT_ERROR,
+      message: '请求超时，请稍后再试',
+      originalError: error,
+    })
+  }
+
+  // 默认错误
+  return createError({
+    type: ErrorType.UNKNOWN_ERROR,
+    message: error.message || '发生未知错误',
+    originalError: error,
+  })
+}

package/src/index.ts

@@ -0,0 +1,10 @@
+export { AIClient } from './client'
+
+export { BaseModelProvider } from './providers/base'
+export { OpenAIProvider } from './providers/openai'
+
+export { formatMessages, extractTextFromResponse } from './utils'
+
+export * from './vue'
+
+export * from './types'

package/src/providers/base.ts

@@ -0,0 +1,62 @@
+import { AIModelConfig, ChatCompletionRequest, ChatCompletionResponse, StreamHandler } from '../types'
+
+/**
+ * 模型Provider基类
+ */
+export abstract class BaseModelProvider {
+  protected config: AIModelConfig
+
+  /**
+   * @param config AI模型配置
+   */
+  constructor(config: AIModelConfig) {
+    this.config = config
+  }
+
+  /**
+   * 发送聊天请求并获取响应
+   * @param request 聊天请求参数
+   * @returns 聊天响应
+   */
+  abstract chat(request: ChatCompletionRequest): Promise<ChatCompletionResponse>
+
+  /**
+   * 发送流式聊天请求并通过处理器处理响应
+   * @param request 聊天请求参数
+   * @param handler 流式响应处理器
+   */
+  abstract chatStream(request: ChatCompletionRequest, handler: StreamHandler): Promise<void>
+
+  /**
+   * 更新配置
+   * @param config 新的AI模型配置
+   */
+  updateConfig(config: AIModelConfig): void {
+    this.config = { ...this.config, ...config }
+  }
+
+  /**
+   * 获取当前配置
+   * @returns AI模型配置
+   */
+  getConfig(): AIModelConfig {
+    return { ...this.config }
+  }
+
+  /**
+   * 验证请求参数
+   * @param request 聊天请求参数
+   */
+  protected validateRequest(request: ChatCompletionRequest): void {
+    if (!request.messages || !Array.isArray(request.messages) || request.messages.length === 0) {
+      throw new Error('请求必须包含至少一条消息')
+    }
+
+    // 验证每条消息的格式
+    for (const message of request.messages) {
+      if (!message.role || !message.content) {
+        throw new Error('每条消息必须包含角色和内容')
+      }
+    }
+  }
+}

package/src/providers/openai.ts

@@ -0,0 +1,134 @@
+/**
+ * OpenAI提供商
+ * 用于与OpenAI API进行交互
+ */
+
+import { BaseModelProvider } from './base'
+import type { AIModelConfig, ChatCompletionRequest, ChatCompletionResponse, StreamHandler } from '../types'
+import { handleRequestError } from '../error'
+import { handleSSEStream } from '../utils'
+
+export class OpenAIProvider extends BaseModelProvider {
+  private baseURL: string
+  private apiKey: string
+  private defaultModel: string = 'gpt-3.5-turbo'
+
+  /**
+   * @param config AI模型配置
+   */
+  constructor(config: AIModelConfig) {
+    super(config)
+
+    this.baseURL = config.apiUrl || 'https://api.openai.com/v1'
+    this.apiKey = config.apiKey || ''
+
+    // 设置默认模型
+    if (config.defaultModel) {
+      this.defaultModel = config.defaultModel
+    }
+
+    if (!this.apiKey) {
+      console.warn('API key is not provided. Authentication will likely fail.')
+    }
+  }
+
+  /**
+   * 发送聊天请求并获取响应
+   * @param request 聊天请求参数
+   * @returns 聊天响应
+   */
+  async chat(request: ChatCompletionRequest): Promise<ChatCompletionResponse> {
+    try {
+      this.validateRequest(request)
+
+      const requestData = {
+        model: request.options?.model || this.config.defaultModel || this.defaultModel,
+        messages: request.messages,
+        ...request.options,
+        stream: false,
+      }
+
+      const response = await fetch(`${this.baseURL}/chat/completions`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          Authorization: `Bearer ${this.apiKey}`,
+        },
+        body: JSON.stringify(requestData),
+      })
+
+      if (!response.ok) {
+        const errorText = await response.text()
+        throw new Error(`HTTP error! status: ${response.status}, details: ${errorText}`)
+      }
+
+      return await response.json()
+    } catch (error: unknown) {
+      // 处理错误
+      throw handleRequestError(error as { response: object })
+    }
+  }
+
+  /**
+   * 发送流式聊天请求并通过处理器处理响应
+   * @param request 聊天请求参数
+   * @param handler 流式响应处理器
+   */
+  async chatStream(request: ChatCompletionRequest, handler: StreamHandler): Promise<void> {
+    const { signal, ...options } = request.options || {}
+
+    try {
+      // 验证请求参数
+      this.validateRequest(request)
+
+      const requestData = {
+        model: request.options?.model || this.config.defaultModel || this.defaultModel,
+        messages: request.messages,
+        ...options,
+        stream: true,
+      }
+
+      const response = await fetch(`${this.baseURL}/chat/completions`, {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          Authorization: `Bearer ${this.apiKey}`,
+          Accept: 'text/event-stream',
+        },
+        body: JSON.stringify(requestData),
+        signal,
+      })
+
+      if (!response.ok) {
+        const errorText = await response.text()
+        throw new Error(`HTTP error! status: ${response.status}, details: ${errorText}`)
+      }
+
+      await handleSSEStream(response, handler, signal)
+    } catch (error: unknown) {
+      if (signal?.aborted) return
+      handler.onError(handleRequestError(error as { response: object }))
+    }
+  }
+
+  /**
+   * 更新配置
+   * @param config 新的AI模型配置
+   */
+  updateConfig(config: AIModelConfig): void {
+    super.updateConfig(config)
+
+    // 更新配置
+    if (config.apiUrl) {
+      this.baseURL = config.apiUrl
+    }
+
+    if (config.apiKey) {
+      this.apiKey = config.apiKey
+    }
+
+    if (config.defaultModel) {
+      this.defaultModel = config.defaultModel
+    }
+  }
+}

package/src/types.ts

@@ -0,0 +1,163 @@
+/**
+ * 消息角色类型
+ */
+export type MessageRole = 'system' | 'user' | 'assistant'
+
+/**
+ * 聊天消息接口
+ */
+export interface ChatMessage {
+  role: MessageRole
+  content: string
+  name?: string
+}
+
+/**
+ * 聊天历史记录
+ */
+export type ChatHistory = ChatMessage[]
+
+/**
+ * 聊天完成请求选项
+ */
+export interface ChatCompletionOptions {
+  model?: string
+  temperature?: number
+  top_p?: number
+  n?: number
+  stream?: boolean
+  max_tokens?: number
+  signal?: AbortSignal
+}
+
+/**
+ * 聊天完成请求参数
+ */
+export interface ChatCompletionRequest {
+  messages: ChatMessage[]
+  options?: ChatCompletionOptions
+}
+
+/**
+ * 聊天完成响应消息
+ */
+export interface ChatCompletionResponseMessage {
+  role: MessageRole
+  content: string
+}
+
+/**
+ * 聊天完成响应选择
+ */
+export interface ChatCompletionResponseChoice {
+  index: number
+  message: ChatCompletionResponseMessage
+  finish_reason: string
+}
+
+/**
+ * 聊天完成响应使用情况
+ */
+export interface ChatCompletionResponseUsage {
+  prompt_tokens: number
+  completion_tokens: number
+  total_tokens: number
+}
+
+/**
+ * 聊天完成响应
+ */
+export interface ChatCompletionResponse {
+  id: string
+  object: string
+  created: number
+  model: string
+  choices: ChatCompletionResponseChoice[]
+  usage: ChatCompletionResponseUsage
+}
+
+/**
+ * 流式聊天完成响应增量
+ */
+export interface ChatCompletionStreamResponseDelta {
+  content?: string
+  role?: MessageRole
+}
+
+/**
+ * 流式聊天完成响应选择
+ */
+export interface ChatCompletionStreamResponseChoice {
+  index: number
+  delta: ChatCompletionStreamResponseDelta
+  finish_reason: string | null
+}
+
+/**
+ * 流式聊天完成响应
+ */
+export interface ChatCompletionStreamResponse {
+  id: string
+  object: string
+  created: number
+  model: string
+  choices: ChatCompletionStreamResponseChoice[]
+}
+
+/**
+ * AI模型提供商类型
+ */
+export type AIProvider = 'openai' | 'deepseek' | 'custom'
+
+/**
+ * AI模型配置接口
+ */
+export interface AIModelConfig {
+  provider: AIProvider
+  apiKey?: string
+  apiUrl?: string
+  apiVersion?: string
+  defaultModel?: string
+  defaultOptions?: ChatCompletionOptions
+}
+
+/**
+ * 错误类型
+ */
+export enum ErrorType {
+  NETWORK_ERROR = 'network_error',
+  AUTHENTICATION_ERROR = 'authentication_error',
+  RATE_LIMIT_ERROR = 'rate_limit_error',
+  SERVER_ERROR = 'server_error',
+  MODEL_ERROR = 'model_error',
+  TIMEOUT_ERROR = 'timeout_error',
+  UNKNOWN_ERROR = 'unknown_error',
+}
+
+/**
+ * AI适配器错误
+ */
+export interface AIAdapterError {
+  type: ErrorType
+  message: string
+  statusCode?: number
+  originalError?: object
+}
+
+/**
+ * 流式响应事件类型
+ */
+export enum StreamEventType {
+  DATA = 'data',
+  ERROR = 'error',
+  DONE = 'done',
+}
+
+/**
+ * 流式响应处理器
+ */
+export interface StreamHandler {
+  onData: (data: ChatCompletionStreamResponse) => void
+  onError: (error: AIAdapterError) => void
+  onDone: () => void
+}

package/src/utils.ts

@@ -0,0 +1,125 @@
+/**
+ * 工具函数模块
+ * 提供一些实用的辅助函数
+ */
+
+import type { ChatMessage, ChatCompletionResponse, ChatCompletionStreamResponse, StreamHandler } from './types'
+
+/**
+ * 处理SSE流式响应
+ * @param response fetch响应对象
+ * @param handler 流处理器
+ */
+export async function handleSSEStream(response: Response, handler: StreamHandler, signal?: AbortSignal): Promise<void> {
+  // 获取ReadableStream
+  const reader = response.body?.getReader()
+  if (!reader) {
+    throw new Error('Response body is null')
+  }
+
+  // 处理流式数据
+  const decoder = new TextDecoder()
+  let buffer = ''
+
+  if (signal) {
+    signal.addEventListener(
+      'abort',
+      () => {
+        reader.cancel().catch((err) => console.error('Error cancelling reader:', err))
+      },
+      { once: true },
+    )
+  }
+
+  try {
+    while (true) {
+      if (signal?.aborted) {
+        await reader.cancel()
+        break
+      }
+
+      const { done, value } = await reader.read()
+      if (done) break
+
+      // 解码二进制数据
+      const chunk = decoder.decode(value, { stream: true })
+      buffer += chunk
+
+      // 处理完整的SSE消息
+      const lines = buffer.split('\n\n')
+      buffer = lines.pop() || ''
+
+      for (const line of lines) {
+        if (line.trim() === '') continue
+        if (line.trim() === 'data: [DONE]') {
+          handler.onDone()
+          continue
+        }
+
+        try {
+          // 解析SSE消息
+          const dataMatch = line.match(/^data: (.+)$/m)
+          if (!dataMatch) continue
+
+          const data = JSON.parse(dataMatch[1]) as ChatCompletionStreamResponse
+          handler.onData(data)
+        } catch (error) {
+          console.error('Error parsing SSE message:', error)
+        }
+      }
+    }
+
+    if (buffer.trim() === 'data: [DONE]' || signal?.aborted) {
+      handler.onDone()
+    }
+  } catch (error) {
+    if (signal?.aborted) return
+    throw error
+  }
+}
+
+/**
+ * 格式化消息
+ * 将各种格式的消息转换为标准的ChatMessage格式
+ * @param messages 消息数组
+ * @returns 标准格式的消息数组
+ */
+export function formatMessages(messages: Array<ChatMessage | string>): ChatMessage[] {
+  return messages.map((msg) => {
+    // 如果已经是标准格式，直接返回
+    if (typeof msg === 'object' && 'role' in msg && 'content' in msg) {
+      return {
+        role: msg.role,
+        content: String(msg.content),
+        ...(msg.name ? { name: msg.name } : {}),
+      }
+    }
+
+    // 如果是字符串，默认为用户消息
+    if (typeof msg === 'string') {
+      return {
+        role: 'user',
+        content: msg,
+      }
+    }
+
+    // 其他情况，尝试转换为字符串
+    return {
+      role: 'user',
+      content: String(msg),
+    }
+  })
+}
+
+/**
+ * 从响应中提取文本内容
+ * @param response 聊天完成响应
+ * @returns 文本内容
+ */
+export function extractTextFromResponse(response: ChatCompletionResponse): string {
+  if (!response.choices || !response.choices.length) {
+    return ''
+  }
+
+  return response.choices[0].message?.content || ''
+}

package/src/vue/index.ts

@@ -0,0 +1 @@
+export * from './useMessage'