@serii84/vertex-partner-provider 1.0.6 → 1.0.8

package/index.js

@@ -1,6 +1,6 @@
 /**
  * Vertex Partner Provider for OpenCode
- * No logging - clean output
+ * Transforms streaming responses to handle reasoning_content
  */
 
 const { createOpenAICompatible } = require('@ai-sdk/openai-compatible');
@@ -20,6 +20,74 @@ async function getAuthToken(googleAuthOptions) {
   return token.token;
 }
 
+/**
+ * Transform SSE stream to handle reasoning_content
+ * Converts reasoning_content to content so SDK can parse it
+ */
+function transformStream(response) {
+  const reader = response.body.getReader();
+  const decoder = new TextDecoder();
+  const encoder = new TextEncoder();
+  
+  let buffer = '';
+  
+  const transformedStream = new ReadableStream({
+    async pull(controller) {
+      const { done, value } = await reader.read();
+      
+      if (done) {
+        controller.close();
+        return;
+      }
+      
+      buffer += decoder.decode(value, { stream: true });
+      const lines = buffer.split('\n');
+      buffer = lines.pop() || ''; // Keep incomplete line in buffer
+      
+      for (const line of lines) {
+        if (line.startsWith('data: ')) {
+          const data = line.slice(6);
+          if (data === '[DONE]') {
+            controller.enqueue(encoder.encode(line + '\n'));
+            continue;
+          }
+          
+          try {
+            const parsed = JSON.parse(data);
+            
+            // Transform: move reasoning_content to content if content is null
+            if (parsed.choices) {
+              for (const choice of parsed.choices) {
+                if (choice.delta) {
+                  // If content is null but reasoning_content exists, use reasoning_content
+                  if (choice.delta.content === null && choice.delta.reasoning_content) {
+                    choice.delta.content = choice.delta.reasoning_content;
+                  }
+                  // Remove reasoning_content to avoid confusion
+                  delete choice.delta.reasoning_content;
+                }
+              }
+            }
+            
+            controller.enqueue(encoder.encode('data: ' + JSON.stringify(parsed) + '\n'));
+          } catch (e) {
+            // If JSON parse fails, pass through unchanged
+            controller.enqueue(encoder.encode(line + '\n'));
+          }
+        } else if (line.trim()) {
+          controller.enqueue(encoder.encode(line + '\n'));
+        }
+      }
+    }
+  });
+  
+  return new Response(transformedStream, {
+    headers: response.headers,
+    status: response.status,
+    statusText: response.statusText,
+  });
+}
+
 function createVertexPartner(options = {}) {
   const {
     project = process.env.GOOGLE_VERTEX_PROJECT,
@@ -41,7 +109,16 @@ function createVertexPartner(options = {}) {
     const token = await getAuthToken(googleAuthOptions);
     const headers = new Headers(init?.headers);
     headers.set('Authorization', `Bearer ${token}`);
-    return fetch(url, { ...init, headers });
+    
+    const response = await fetch(url, { ...init, headers });
+    
+    // Check if this is a streaming request
+    const contentType = response.headers.get('content-type') || '';
+    if (contentType.includes('text/event-stream')) {
+      return transformStream(response);
+    }
+    
+    return response;
   };
 
   const provider = createOpenAICompatible({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@serii84/vertex-partner-provider",
-  "version": "1.0.6",
+  "version": "1.0.8",
   "description": "Vertex AI partner models (GLM, Kimi, DeepSeek) for OpenCode",
   "main": "index.js",
   "scripts": {

package/test.js

@@ -1,78 +0,0 @@
-/**
- * Test script for Vertex Partner Provider
- * 
- * Run with: node test.js
- * 
- * Requires: gcloud auth application-default login
- */
-
-const { createGLM, createDeepSeek, createKimi } = require('./index.js');
-const { generateText, tool } = require('ai');
-const { z } = require('zod');
-
-const PROJECT = process.env.GOOGLE_VERTEX_PROJECT || 'noter-1c2a1';
-
-async function testBasicChat() {
-  console.log('\n📝 Test 1: Basic Chat');
-  console.log('-'.repeat(40));
-
-  const glm = createGLM({ project: PROJECT, location: 'global' });
-  const model = glm('glm-4.7-maas');
-  
-  console.log('Model ID:', model.modelId);
-  
-  const result = await generateText({
-    model,
-    prompt: 'Say "Hello from GLM!" and nothing else.',
-  });
-  
-  console.log('✅ Response:', result.text);
-  return true;
-}
-
-async function testToolCalling() {
-  console.log('\n🔧 Test 2: Tool Calling');
-  console.log('-'.repeat(40));
-
-  const glm = createGLM({ project: PROJECT, location: 'global' });
-  
-  const result = await generateText({
-    model: glm('glm-4.7-maas'),
-    prompt: 'What is the weather in San Francisco? Use the weather tool.',
-    tools: {
-      weather: tool({
-        description: 'Get the weather for a location',
-        parameters: z.object({
-          location: z.string().describe('The city name'),
-        }),
-        execute: async ({ location }) => {
-          return { location, temperature: 72, condition: 'sunny' };
-        },
-      }),
-    },
-    maxSteps: 3,
-  });
-  
-  console.log('✅ Response:', result.text);
-  console.log('Tool calls:', result.toolCalls?.length || 0);
-  return true;
-}
-
-async function main() {
-  console.log('='.repeat(50));
-  console.log('Vertex Partner Provider Test Suite');
-  console.log('='.repeat(50));
-  console.log(`Project: ${PROJECT}`);
-
-  try {
-    await testBasicChat();
-    await testToolCalling();
-    console.log('\n✅ All tests passed!');
-  } catch (error) {
-    console.error('\n❌ Test failed:', error.message);
-    if (error.cause) console.error('Cause:', JSON.stringify(error.cause, null, 2));
-    process.exit(1);
-  }
-}
-
-main();