kimi-vercel-ai-sdk-provider 0.2.0 → 0.3.0

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "kimi-vercel-ai-sdk-provider",
   "description": "Kimi (Moonshot AI) provider for Vercel AI SDK v6",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -15,6 +15,9 @@
   "scripts": {
     "build": "tsup",
     "clean": "del-cli dist *.tsbuildinfo",
+    "format": "biome format --write",
+    "lint": "biome lint --write",
+    "lint:fix": "biome lint --write",
     "type-check": "tsc --noEmit",
     "test": "vitest run",
     "test:watch": "vitest",

package/src/__tests__/code-integration.test.ts

@@ -1,9 +1,27 @@
-import { describe, expect, it, vi, beforeEach, afterEach } from 'vitest';
+import type { LanguageModelV3StreamPart } from '@ai-sdk/provider';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { createKimiCode } from '../code';
 
 // Mock fetch for API testing
 const mockFetch = vi.fn();
 
+async function readStreamParts<T>(stream: ReadableStream<T>) {
+  const reader = stream.getReader();
+  const parts: T[] = [];
+
+  while (true) {
+    const { value, done } = await reader.read();
+    if (done) {
+      break;
+    }
+    if (value !== undefined) {
+      parts.push(value);
+    }
+  }
+
+  return parts;
+}
+
 describe('KimiCodeLanguageModel Integration', () => {
   const originalEnv = process.env;
 
@@ -147,7 +165,7 @@ describe('KimiCodeLanguageModel Integration', () => {
 
       const result = await model.doGenerate({
         prompt: [{ role: 'user', content: [{ type: 'text', text: 'Read the index file' }] }],
-        
+
         tools: [
           {
             type: 'function',
@@ -198,8 +216,7 @@ describe('KimiCodeLanguageModel Integration', () => {
       });
 
       await model.doGenerate({
-        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Think hard' }] }],
-        
+        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Think hard' }] }]
       });
 
       const body = JSON.parse(mockFetch.mock.calls[0][1].body);
@@ -239,8 +256,7 @@ describe('KimiCodeLanguageModel Integration', () => {
       const model = provider('kimi-for-coding');
 
       const result = await model.doGenerate({
-        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Test' }] }],
-        
+        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Test' }] }]
       });
 
       expect(result.usage.inputTokens.total).toBe(100);
@@ -273,8 +289,7 @@ describe('KimiCodeLanguageModel Integration', () => {
 
       await expect(
         model.doGenerate({
-          prompt: [{ role: 'user', content: [{ type: 'text', text: 'Test' }] }],
-          
+          prompt: [{ role: 'user', content: [{ type: 'text', text: 'Test' }] }]
         })
       ).rejects.toThrow('Invalid API key');
     });
@@ -305,7 +320,7 @@ describe('KimiCodeLanguageModel Integration', () => {
 
       const result = await model.doGenerate({
         prompt: [{ role: 'user', content: [{ type: 'text', text: 'Write a long story' }] }],
-        
+
         maxOutputTokens: 100
       });
 
@@ -350,14 +365,10 @@ describe('KimiCodeLanguageModel Integration', () => {
       const model = provider('kimi-for-coding');
 
       const result = await model.doStream({
-        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Say hello' }] }],
-        
+        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Say hello' }] }]
       });
 
-      const parts: any[] = [];
-      for await (const part of result.stream) {
-        parts.push(part);
-      }
+      const parts = await readStreamParts<LanguageModelV3StreamPart>(result.stream);
 
       // Verify streaming worked
       expect(parts.length).toBeGreaterThan(0);
@@ -409,14 +420,10 @@ describe('KimiCodeLanguageModel Integration', () => {
       const model = provider('kimi-k2-thinking');
 
       const result = await model.doStream({
-        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Think about this' }] }],
-        
+        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Think about this' }] }]
       });
 
-      const parts: any[] = [];
-      for await (const part of result.stream) {
-        parts.push(part);
-      }
+      const parts = await readStreamParts<LanguageModelV3StreamPart>(result.stream);
 
       const partTypes = parts.map((p) => p.type);
       expect(partTypes).toContain('reasoning-start');
@@ -463,7 +470,7 @@ describe('KimiCodeLanguageModel Integration', () => {
 
       const result = await model.doStream({
         prompt: [{ role: 'user', content: [{ type: 'text', text: 'Read the file' }] }],
-        
+
         tools: [
           {
             type: 'function',
@@ -474,10 +481,7 @@ describe('KimiCodeLanguageModel Integration', () => {
         ]
       });
 
-      const parts: any[] = [];
-      for await (const part of result.stream) {
-        parts.push(part);
-      }
+      const parts = await readStreamParts<LanguageModelV3StreamPart>(result.stream);
 
       const partTypes = parts.map((p) => p.type);
       expect(partTypes).toContain('tool-input-start');
@@ -485,9 +489,12 @@ describe('KimiCodeLanguageModel Integration', () => {
       expect(partTypes).toContain('tool-input-end');
       expect(partTypes).toContain('tool-call');
 
-      const toolCallPart = parts.find((p) => p.type === 'tool-call');
-      expect(toolCallPart.toolName).toBe('read_file');
-      expect(toolCallPart.input).toContain('/src/index.ts');
+      const toolCallPart = parts.find(
+        (p): p is Extract<LanguageModelV3StreamPart, { type: 'tool-call' }> => p.type === 'tool-call'
+      );
+      expect(toolCallPart).toBeDefined();
+      expect(toolCallPart?.toolName).toBe('read_file');
+      expect(toolCallPart?.input).toContain('/src/index.ts');
     });
   });
 
@@ -517,8 +524,7 @@ describe('KimiCodeLanguageModel Integration', () => {
       const model = provider('kimi-for-coding');
 
       await model.doGenerate({
-        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Test' }] }],
-        
+        prompt: [{ role: 'user', content: [{ type: 'text', text: 'Test' }] }]
       });
 
       const headers = mockFetch.mock.calls[0][1].headers;

package/src/__tests__/code-provider.test.ts

@@ -1,12 +1,11 @@
-import { describe, expect, it, vi, beforeEach, afterEach } from 'vitest';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import {
-  createKimiCode,
-  kimiCode,
-  KimiCodeLanguageModel,
-  KIMI_CODE_BASE_URL,
   KIMI_CODE_DEFAULT_MODEL,
   KIMI_CODE_THINKING_MODEL,
-  inferKimiCodeCapabilities
+  KimiCodeLanguageModel,
+  createKimiCode,
+  inferKimiCodeCapabilities,
+  kimiCode
 } from '../code';
 
 describe('createKimiCode', () => {

package/src/__tests__/code.test.ts

@@ -171,9 +171,7 @@ describe('toAnthropicThinking', () => {
 
 describe('convertToKimiCodePrompt', () => {
   it('should convert simple user message', async () => {
-    const result = await convertToKimiCodePrompt([
-      { role: 'user', content: [{ type: 'text', text: 'Hello!' }] }
-    ]);
+    const result = await convertToKimiCodePrompt([{ role: 'user', content: [{ type: 'text', text: 'Hello!' }] }]);
 
     expect(result.system).toBeUndefined();
     expect(result.messages).toEqual([{ role: 'user', content: 'Hello!' }]);

package/src/__tests__/provider.test.ts

@@ -42,8 +42,9 @@ describe('createKimi', () => {
   it('should throw when called with new', () => {
     const provider = createKimi();
     // The error can be either our custom message or the native constructor error
-    // biome-ignore lint/suspicious/noExplicitAny: testing edge case requires any
-    expect(() => new (provider as any)('kimi-k2.5')).toThrow();
+    // Test that the provider function throws when used as a constructor
+    const ProviderAsConstructor = provider as unknown as new (modelId: string) => unknown;
+    expect(() => new ProviderAsConstructor('kimi-k2.5')).toThrow();
   });
 
   it('should throw for unsupported model types', () => {

package/src/chat/kimi-chat-language-model.ts

@@ -765,6 +765,26 @@ const kimiTokenUsageSchema = z
   })
   .nullish();
 
+/**
+ * Schema for content parts in response messages.
+ * Can be text, image, or other content types.
+ */
+const kimiContentPartSchema = z.union([
+  z.object({
+    type: z.literal('text'),
+    text: z.string()
+  }),
+  z.object({
+    type: z.literal('image_url'),
+    image_url: z.object({
+      url: z.string()
+    })
+  }),
+  z.looseObject({
+    type: z.string()
+  })
+]);
+
 const kimiChatResponseSchema = z.looseObject({
   id: z.string().nullish(),
   created: z.number().nullish(),
@@ -773,7 +793,7 @@ const kimiChatResponseSchema = z.looseObject({
     z.object({
       message: z.object({
         role: z.string().nullish(),
-        content: z.union([z.string(), z.array(z.any())]).nullish(),
+        content: z.union([z.string(), z.array(kimiContentPartSchema)]).nullish(),
         reasoning_content: z.string().nullish(),
         reasoning: z.string().nullish(),
         tool_calls: z

package/src/core/errors.ts

@@ -19,7 +19,7 @@ export const kimiErrorSchema = z.union([
     error: z.object({
       message: z.string(),
       type: z.string().nullish(),
-      param: z.any().nullish(),
+      param: z.string().nullish(),
       code: z.union([z.string(), z.number()]).nullish(),
       request_id: z.string().nullish()
     })

package/src/files/attachment-processor.ts

@@ -257,7 +257,8 @@ function guessFilename(attachment: Attachment, contentType: string): string {
     const urlPath = attachment.url.split('?')[0];
     const segments = urlPath.split('/');
     const lastSegment = segments[segments.length - 1];
-    if (lastSegment && lastSegment.includes('.')) {
+
+    if (lastSegment.includes('.')) {
       return lastSegment;
     }
   }