kimi-vercel-ai-sdk-provider 0.2.0

package/src/files/file-utils.ts

@@ -0,0 +1,257 @@
+/**
+ * File utility functions for Kimi API.
+ * @module
+ */
+
+// ============================================================================
+// Constants
+// ============================================================================
+
+/**
+ * File extensions supported by Kimi's file upload API.
+ */
+export const SUPPORTED_FILE_EXTENSIONS = [
+  // Documents
+  '.pdf',
+  '.txt',
+  '.csv',
+  '.doc',
+  '.docx',
+  '.xls',
+  '.xlsx',
+  '.ppt',
+  '.pptx',
+  '.md',
+  '.epub',
+  '.mobi',
+  '.html',
+  '.json',
+  '.log',
+  '.dot',
+  '.ini',
+  '.conf',
+  '.yaml',
+  '.yml',
+  // Images
+  '.jpeg',
+  '.jpg',
+  '.png',
+  '.bmp',
+  '.gif',
+  '.svg',
+  '.svgz',
+  '.webp',
+  '.ico',
+  '.xbm',
+  '.dib',
+  '.pjp',
+  '.tif',
+  '.tiff',
+  '.pjpeg',
+  '.avif',
+  '.apng',
+  '.jfif',
+  // Code files
+  '.go',
+  '.h',
+  '.c',
+  '.cpp',
+  '.cxx',
+  '.cc',
+  '.cs',
+  '.java',
+  '.js',
+  '.css',
+  '.jsp',
+  '.php',
+  '.py',
+  '.py3',
+  '.asp',
+  '.ts',
+  '.tsx'
+] as const;
+
+/**
+ * MIME types supported by Kimi's file upload API.
+ */
+export const SUPPORTED_MIME_TYPES = {
+  // Documents
+  'application/pdf': 'file-extract',
+  'text/plain': 'file-extract',
+  'text/csv': 'file-extract',
+  'application/msword': 'file-extract',
+  'application/vnd.openxmlformats-officedocument.wordprocessingml.document': 'file-extract',
+  'application/vnd.ms-excel': 'file-extract',
+  'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet': 'file-extract',
+  'application/vnd.ms-powerpoint': 'file-extract',
+  'application/vnd.openxmlformats-officedocument.presentationml.presentation': 'file-extract',
+  'text/markdown': 'file-extract',
+  'text/html': 'file-extract',
+  'application/json': 'file-extract',
+  'application/epub+zip': 'file-extract',
+  'text/yaml': 'file-extract',
+  'application/x-yaml': 'file-extract',
+  // Code files (treated as text)
+  'text/javascript': 'file-extract',
+  'text/typescript': 'file-extract',
+  'text/x-python': 'file-extract',
+  'text/x-java': 'file-extract',
+  'text/x-c': 'file-extract',
+  'text/x-c++': 'file-extract',
+  'text/css': 'file-extract',
+  // Images
+  'image/jpeg': 'image',
+  'image/png': 'image',
+  'image/gif': 'image',
+  'image/webp': 'image',
+  'image/svg+xml': 'image',
+  'image/bmp': 'image',
+  'image/tiff': 'image',
+  'image/avif': 'image',
+  'image/apng': 'image',
+  'image/x-icon': 'image',
+  // Videos
+  'video/mp4': 'video',
+  'video/webm': 'video',
+  'video/ogg': 'video',
+  'video/quicktime': 'video'
+} as const;
+
+export type SupportedMimeType = keyof typeof SUPPORTED_MIME_TYPES;
+export type FilePurpose = 'file-extract' | 'image' | 'video';
+
+// ============================================================================
+// Type Guards
+// ============================================================================
+
+/**
+ * Check if a media type is for image files.
+ */
+export function isImageMediaType(mediaType: string): boolean {
+  return mediaType.startsWith('image/');
+}
+
+/**
+ * Check if a media type is for video files.
+ */
+export function isVideoMediaType(mediaType: string): boolean {
+  return mediaType.startsWith('video/');
+}
+
+/**
+ * Check if a media type should use file-extract purpose.
+ */
+export function isFileExtractMediaType(mediaType: string): boolean {
+  // Check explicit types
+  if (mediaType in SUPPORTED_MIME_TYPES) {
+    return SUPPORTED_MIME_TYPES[mediaType as SupportedMimeType] === 'file-extract';
+  }
+  // Check text/* and application/* prefixes
+  if (mediaType.startsWith('text/') || mediaType === 'application/pdf') {
+    return true;
+  }
+  return false;
+}
+
+/**
+ * Check if a media type is for document files (PDFs, Word, etc.).
+ */
+export function isDocumentMediaType(mediaType: string): boolean {
+  const documentTypes = [
+    'application/pdf',
+    'application/msword',
+    'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+    'application/vnd.ms-excel',
+    'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+    'application/vnd.ms-powerpoint',
+    'application/vnd.openxmlformats-officedocument.presentationml.presentation',
+    'application/epub+zip'
+  ];
+  return documentTypes.includes(mediaType);
+}
+
+// ============================================================================
+// Utility Functions
+// ============================================================================
+
+/**
+ * Get the file purpose based on media type.
+ */
+export function getPurposeFromMediaType(mediaType: string): FilePurpose {
+  if (isImageMediaType(mediaType)) {
+    return 'image';
+  }
+  if (isVideoMediaType(mediaType)) {
+    return 'video';
+  }
+  return 'file-extract';
+}
+
+/**
+ * Get MIME type from file extension.
+ */
+export function getMediaTypeFromExtension(extension: string): string {
+  const ext = extension.toLowerCase().startsWith('.') ? extension.toLowerCase() : `.${extension.toLowerCase()}`;
+
+  const extensionToMime: Record<string, string> = {
+    // Documents
+    '.pdf': 'application/pdf',
+    '.txt': 'text/plain',
+    '.csv': 'text/csv',
+    '.doc': 'application/msword',
+    '.docx': 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+    '.xls': 'application/vnd.ms-excel',
+    '.xlsx': 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+    '.ppt': 'application/vnd.ms-powerpoint',
+    '.pptx': 'application/vnd.openxmlformats-officedocument.presentationml.presentation',
+    '.md': 'text/markdown',
+    '.html': 'text/html',
+    '.json': 'application/json',
+    '.epub': 'application/epub+zip',
+    '.yaml': 'text/yaml',
+    '.yml': 'text/yaml',
+    '.log': 'text/plain',
+    '.ini': 'text/plain',
+    '.conf': 'text/plain',
+    // Code
+    '.js': 'text/javascript',
+    '.ts': 'text/typescript',
+    '.tsx': 'text/typescript',
+    '.py': 'text/x-python',
+    '.java': 'text/x-java',
+    '.c': 'text/x-c',
+    '.cpp': 'text/x-c++',
+    '.h': 'text/x-c',
+    '.css': 'text/css',
+    '.go': 'text/plain',
+    '.php': 'text/plain',
+    // Images
+    '.jpeg': 'image/jpeg',
+    '.jpg': 'image/jpeg',
+    '.png': 'image/png',
+    '.gif': 'image/gif',
+    '.webp': 'image/webp',
+    '.svg': 'image/svg+xml',
+    '.bmp': 'image/bmp',
+    '.tif': 'image/tiff',
+    '.tiff': 'image/tiff',
+    '.avif': 'image/avif',
+    '.apng': 'image/apng',
+    '.ico': 'image/x-icon',
+    // Videos
+    '.mp4': 'video/mp4',
+    '.webm': 'video/webm',
+    '.ogg': 'video/ogg',
+    '.mov': 'video/quicktime'
+  };
+
+  return extensionToMime[ext] ?? 'application/octet-stream';
+}
+
+/**
+ * Extract file extension from URL or filename.
+ */
+export function getExtensionFromPath(path: string): string | null {
+  const match = path.match(/\.([^./?#]+)(?:[?#]|$)/);
+  return match ? `.${match[1].toLowerCase()}` : null;
+}

package/src/files/index.ts

@@ -0,0 +1,24 @@
+/**
+ * File handling module for Kimi API.
+ * Provides utilities for uploading and extracting content from files.
+ * @module
+ */
+
+export { type Attachment, type ProcessedAttachment, processAttachments } from './attachment-processor';
+export {
+  SUPPORTED_FILE_EXTENSIONS,
+  SUPPORTED_MIME_TYPES,
+  getMediaTypeFromExtension,
+  getPurposeFromMediaType,
+  isDocumentMediaType,
+  isFileExtractMediaType,
+  isImageMediaType,
+  isVideoMediaType
+} from './file-utils';
+export {
+  type FileUploadOptions,
+  type FileUploadResult,
+  type KimiFile,
+  KimiFileClient,
+  type KimiFileClientConfig
+} from './kimi-file-client';

package/src/files/kimi-file-client.ts

@@ -0,0 +1,292 @@
+/**
+ * Kimi File API client for uploading and managing files.
+ * @module
+ */
+
+import type { FetchFunction } from '@ai-sdk/provider-utils';
+import { getExtensionFromPath, getMediaTypeFromExtension, getPurposeFromMediaType } from './file-utils';
+
+// ============================================================================
+// Types
+// ============================================================================
+
+/**
+ * A file object returned by the Kimi File API.
+ */
+export interface KimiFile {
+  /** Unique file identifier */
+  id: string;
+  /** File size in bytes */
+  bytes: number;
+  /** Unix timestamp of creation */
+  created_at: number;
+  /** Original filename */
+  filename: string;
+  /** Object type (always 'file') */
+  object: 'file';
+  /** File purpose (file-extract, image, video) */
+  purpose: 'file-extract' | 'image' | 'video';
+  /** Processing status */
+  status: 'ok' | 'error' | 'processing';
+  /** Status details if error */
+  status_details?: string;
+}
+
+/**
+ * Options for uploading a file.
+ */
+export interface FileUploadOptions {
+  /** The file data as a Buffer, Uint8Array, or string (base64) */
+  data: Uint8Array | string;
+  /** The filename */
+  filename: string;
+  /** MIME type of the file */
+  mediaType?: string;
+  /** Purpose of the file (defaults based on mediaType) */
+  purpose?: 'file-extract' | 'image' | 'video';
+}
+
+/**
+ * Result of a file upload operation.
+ */
+export interface FileUploadResult {
+  /** The uploaded file object */
+  file: KimiFile;
+  /** The extracted content (for file-extract purpose) */
+  content?: string;
+}
+
+/**
+ * Configuration for the file client.
+ */
+export interface KimiFileClientConfig {
+  /** Base URL for the API */
+  baseURL: string;
+  /** Function to get authorization headers */
+  headers: () => Record<string, string | undefined>;
+  /** Custom fetch implementation */
+  fetch?: FetchFunction;
+}
+
+// ============================================================================
+// File Client
+// ============================================================================
+
+/**
+ * Client for interacting with Kimi's File API.
+ *
+ * Supports uploading files for content extraction, image understanding,
+ * and video understanding.
+ *
+ * @example
+ * ```ts
+ * const client = new KimiFileClient({
+ *   baseURL: 'https://api.moonshot.ai/v1',
+ *   headers: () => ({
+ *     Authorization: `Bearer ${apiKey}`,
+ *   }),
+ * });
+ *
+ * // Upload a PDF and extract content
+ * const result = await client.uploadAndExtract({
+ *   data: pdfBuffer,
+ *   filename: 'document.pdf',
+ *   mediaType: 'application/pdf',
+ * });
+ *
+ * console.log(result.content); // Extracted text content
+ * ```
+ */
+export class KimiFileClient {
+  private readonly config: KimiFileClientConfig;
+
+  constructor(config: KimiFileClientConfig) {
+    this.config = config;
+  }
+
+  /**
+   * Upload a file to the Kimi API.
+   */
+  async upload(options: FileUploadOptions): Promise<KimiFile> {
+    const { data, filename, mediaType, purpose } = options;
+
+    // Determine MIME type
+    const resolvedMediaType =
+      mediaType ?? getMediaTypeFromExtension(getExtensionFromPath(filename) ?? '') ?? 'application/octet-stream';
+
+    // Determine purpose
+    const resolvedPurpose = purpose ?? getPurposeFromMediaType(resolvedMediaType);
+
+    // Create form data
+    const formData = new FormData();
+
+    // Convert data to Blob
+    const fileData = typeof data === 'string' ? base64ToUint8Array(data) : data;
+    const blob = new Blob([new Uint8Array(fileData).buffer as ArrayBuffer], { type: resolvedMediaType });
+
+    formData.append('file', blob, filename);
+    formData.append('purpose', resolvedPurpose);
+
+    const fetchFn = this.config.fetch ?? fetch;
+    const headers = this.config.headers();
+
+    const response = await fetchFn(`${this.config.baseURL}/files`, {
+      method: 'POST',
+      headers: {
+        ...Object.fromEntries(
+          Object.entries(headers).filter((entry): entry is [string, string] => entry[1] !== undefined)
+        )
+        // Don't set Content-Type - let the browser set it with boundary for FormData
+      },
+      body: formData
+    });
+
+    if (!response.ok) {
+      const errorBody = await response.text();
+      throw new Error(`Failed to upload file: ${response.status} ${response.statusText} - ${errorBody}`);
+    }
+
+    return (await response.json()) as KimiFile;
+  }
+
+  /**
+   * Get the content of an uploaded file (for file-extract purpose).
+   */
+  async getContent(fileId: string): Promise<string> {
+    const fetchFn = this.config.fetch ?? fetch;
+    const headers = this.config.headers();
+
+    const response = await fetchFn(`${this.config.baseURL}/files/${fileId}/content`, {
+      method: 'GET',
+      headers: Object.fromEntries(
+        Object.entries(headers).filter((entry): entry is [string, string] => entry[1] !== undefined)
+      )
+    });
+
+    if (!response.ok) {
+      const errorBody = await response.text();
+      throw new Error(`Failed to get file content: ${response.status} ${response.statusText} - ${errorBody}`);
+    }
+
+    return response.text();
+  }
+
+  /**
+   * Upload a file and extract its content in one operation.
+   * Only works for files with purpose="file-extract".
+   */
+  async uploadAndExtract(options: FileUploadOptions): Promise<FileUploadResult> {
+    const file = await this.upload({
+      ...options,
+      purpose: options.purpose ?? 'file-extract'
+    });
+
+    // Wait for processing if needed
+    let currentFile = file;
+    let attempts = 0;
+    const maxAttempts = 30; // 30 seconds max wait
+    const pollInterval = 1000; // 1 second
+
+    while (currentFile.status === 'processing' && attempts < maxAttempts) {
+      await new Promise((resolve) => setTimeout(resolve, pollInterval));
+      currentFile = await this.getFile(file.id);
+      attempts++;
+    }
+
+    if (currentFile.status === 'error') {
+      throw new Error(`File processing failed: ${currentFile.status_details ?? 'Unknown error'}`);
+    }
+
+    if (currentFile.status === 'processing') {
+      throw new Error('File processing timed out');
+    }
+
+    // Get content for file-extract purpose
+    let content: string | undefined;
+    if (currentFile.purpose === 'file-extract') {
+      content = await this.getContent(file.id);
+    }
+
+    return { file: currentFile, content };
+  }
+
+  /**
+   * Get file information.
+   */
+  async getFile(fileId: string): Promise<KimiFile> {
+    const fetchFn = this.config.fetch ?? fetch;
+    const headers = this.config.headers();
+
+    const response = await fetchFn(`${this.config.baseURL}/files/${fileId}`, {
+      method: 'GET',
+      headers: Object.fromEntries(
+        Object.entries(headers).filter((entry): entry is [string, string] => entry[1] !== undefined)
+      )
+    });
+
+    if (!response.ok) {
+      const errorBody = await response.text();
+      throw new Error(`Failed to get file: ${response.status} ${response.statusText} - ${errorBody}`);
+    }
+
+    return (await response.json()) as KimiFile;
+  }
+
+  /**
+   * List all uploaded files.
+   */
+  async listFiles(): Promise<KimiFile[]> {
+    const fetchFn = this.config.fetch ?? fetch;
+    const headers = this.config.headers();
+
+    const response = await fetchFn(`${this.config.baseURL}/files`, {
+      method: 'GET',
+      headers: Object.fromEntries(
+        Object.entries(headers).filter((entry): entry is [string, string] => entry[1] !== undefined)
+      )
+    });
+
+    if (!response.ok) {
+      const errorBody = await response.text();
+      throw new Error(`Failed to list files: ${response.status} ${response.statusText} - ${errorBody}`);
+    }
+
+    const result = (await response.json()) as { data: KimiFile[] };
+    return result.data;
+  }
+
+  /**
+   * Delete a file.
+   */
+  async deleteFile(fileId: string): Promise<void> {
+    const fetchFn = this.config.fetch ?? fetch;
+    const headers = this.config.headers();
+
+    const response = await fetchFn(`${this.config.baseURL}/files/${fileId}`, {
+      method: 'DELETE',
+      headers: Object.fromEntries(
+        Object.entries(headers).filter((entry): entry is [string, string] => entry[1] !== undefined)
+      )
+    });
+
+    if (!response.ok) {
+      const errorBody = await response.text();
+      throw new Error(`Failed to delete file: ${response.status} ${response.statusText} - ${errorBody}`);
+    }
+  }
+}
+
+// ============================================================================
+// Helper Functions
+// ============================================================================
+
+function base64ToUint8Array(base64: string): Uint8Array {
+  // Handle data URLs
+  const base64Data = base64.includes(',') ? base64.split(',')[1] : base64;
+  const binaryString = atob(base64Data);
+  const bytes = new Uint8Array(binaryString.length);
+  for (let i = 0; i < binaryString.length; i++) {
+    bytes[i] = binaryString.charCodeAt(i);
+  }
+  return bytes;
+}

package/src/index.ts

@@ -0,0 +1,122 @@
+/**
+ * Kimi (Moonshot AI) Provider for Vercel AI SDK
+ *
+ * A native implementation of the Kimi AI provider for the Vercel AI SDK,
+ * supporting all Kimi-specific features including web search, code interpreter,
+ * reasoning/thinking models, and Kimi Code premium coding service.
+ *
+ * @packageDocumentation
+ * @module kimi-vercel-ai-sdk-provider
+
+ */
+
+// ============================================================================
+// Kimi Provider (Standard API)
+// ============================================================================
+
+export type { KimiProvider, KimiProviderSettings } from './kimi-provider';
+export { createKimi, kimi } from './kimi-provider';
+
+// ============================================================================
+// Kimi Code Provider (Premium Coding API)
+// ============================================================================
+
+export type {
+  ExtendedThinkingConfig,
+  KimiCodeCapabilities,
+  KimiCodeModelId,
+  KimiCodeProvider,
+  KimiCodeProviderOptions,
+  KimiCodeProviderSettings,
+  KimiCodeSettings,
+  ReasoningEffort
+} from './code';
+export {
+  KIMI_CODE_BASE_URL,
+  KIMI_CODE_DEFAULT_MODEL,
+  KIMI_CODE_THINKING_MODEL,
+  KimiCodeLanguageModel,
+  createKimiCode,
+  inferKimiCodeCapabilities,
+  kimiCode,
+  kimiCodeProviderOptionsSchema
+} from './code';
+
+// ============================================================================
+// Chat Model
+// ============================================================================
+
+export type {
+  KimiCachingConfig,
+  KimiChatModelId,
+  KimiChatSettings,
+  KimiExtendedUsage,
+  KimiModelCapabilities,
+  KimiProviderOptions
+} from './chat';
+export {
+  KimiChatLanguageModel,
+  inferModelCapabilities,
+  kimiCachingConfigSchema,
+  kimiProviderOptionsSchema
+} from './chat';
+
+// ============================================================================
+// File Handling
+// ============================================================================
+
+export type {
+  Attachment,
+  FileUploadOptions,
+  FileUploadResult,
+  KimiFile,
+  KimiFileClientConfig,
+  ProcessedAttachment
+} from './files';
+export {
+  KimiFileClient,
+  SUPPORTED_FILE_EXTENSIONS,
+  SUPPORTED_MIME_TYPES,
+  getMediaTypeFromExtension,
+  getPurposeFromMediaType,
+  isDocumentMediaType,
+  isFileExtractMediaType,
+  isImageMediaType,
+  isVideoMediaType,
+  processAttachments
+} from './files';
+
+// ============================================================================
+// Built-in Tools
+// ============================================================================
+
+export type {
+  KimiBuiltinTool,
+  KimiCodeInterpreterConfig,
+  KimiCodeInterpreterToolOptions,
+  KimiWebSearchConfig,
+  KimiWebSearchToolConfig,
+  KimiWebSearchToolOptions
+} from './tools';
+export {
+  KIMI_CODE_INTERPRETER_TOOL_NAME,
+  KIMI_WEB_SEARCH_TOOL_NAME,
+  createCodeInterpreterTool,
+  createKimiWebSearchTool,
+  createWebSearchTool,
+  kimiTools
+} from './tools';
+
+// ============================================================================
+// Errors
+// ============================================================================
+
+export {
+  KimiAuthenticationError,
+  KimiContentFilterError,
+  KimiContextLengthError,
+  KimiError,
+  KimiModelNotFoundError,
+  KimiRateLimitError,
+  KimiValidationError
+} from './core';