@serii84/vertex-partner-provider 1.0.0

package/README.md

@@ -0,0 +1,104 @@
+# Vertex Partner Provider
+
+Vercel AI SDK provider for **Vertex AI partner models** (GLM, Kimi, DeepSeek, MiniMax, Qwen) using the OpenAI-compatible format.
+
+## The Problem
+
+Vertex AI partner models use OpenAI-compatible format, but `@ai-sdk/google-vertex` only supports Gemini format. This provider bridges that gap using the same pattern as `@ai-sdk/google-vertex/anthropic`.
+
+## How It Works
+
+```
+┌─────────────────────────────────────────────────────────────────┐
+│                    @ai-sdk/google-vertex                        │
+├─────────────────────┬───────────────────────────────────────────┤
+│  Main (Gemini)      │  /anthropic submodule                     │
+│                     │                                           │
+│  Uses:              │  Uses:                                    │
+│  GoogleGenerative   │  AnthropicMessagesLanguageModel           │
+│  AILanguageModel    │  (from @ai-sdk/anthropic/internal)        │
+│                     │                                           │
+│  Format: Gemini     │  Format: Anthropic (native)               │
+└─────────────────────┴───────────────────────────────────────────┘
+
+┌─────────────────────────────────────────────────────────────────┐
+│              vertex-partner-provider (this package)             │
+├─────────────────────────────────────────────────────────────────┤
+│  Uses:                                                          │
+│  OpenAICompatibleChatLanguageModel                              │
+│  (from @ai-sdk/openai-compatible)                               │
+│                                                                 │
+│  Format: OpenAI (native) + Google Cloud Auth                    │
+│                                                                 │
+│  Endpoint: /endpoints/openapi/chat/completions                  │
+└─────────────────────────────────────────────────────────────────┘
+```
+
+## Installation
+
+```bash
+npm install @ai-sdk/openai-compatible google-auth-library ai
+# Then copy index.js to your project
+```
+
+## Usage
+
+```javascript
+const { createGLM, createDeepSeek, createKimi } = require('./vertex-partner-provider');
+const { generateText } = require('ai');
+
+// Create provider for GLM
+const glm = createGLM({
+  project: 'your-gcp-project',
+  location: 'global',  // or specific region
+});
+
+// Use it
+const result = await generateText({
+  model: glm('glm-4.7-maas'),
+  prompt: 'Hello!',
+});
+
+console.log(result.text);
+```
+
+## Available Factory Functions
+
+| Function | Publisher | Models |
+|----------|-----------|--------|
+| `createGLM()` | zai-org | glm-4.7-maas |
+| `createKimi()` | moonshotai | kimi-k2-thinking-maas |
+| `createDeepSeek()` | deepseek-ai | deepseek-v3.2-maas |
+| `createMiniMax()` | minimaxai | minimax-m2-maas |
+| `createQwen()` | qwen | qwen3-coder-480b-a35b-instruct-maas |
+
+## Generic Usage
+
+```javascript
+const { createVertexPartner } = require('./vertex-partner-provider');
+
+const provider = createVertexPartner({
+  project: 'your-project',
+  location: 'global',
+  publisher: 'any-publisher',
+});
+
+const model = provider('model-id');
+```
+
+## Authentication
+
+Uses Google Cloud Application Default Credentials:
+
+```bash
+gcloud auth application-default login
+gcloud auth application-default set-quota-project YOUR_PROJECT
+```
+
+## Tool Calling Support
+
+⚠️ **Note**: Tool calling should work with this provider since it uses the native OpenAI-compatible format. The previous issues were caused by using the Gemini format class with OpenAI-format responses.
+
+## License
+
+MIT

package/index.js

@@ -0,0 +1,94 @@
+/**
+ * Vertex Partner Provider for OpenCode
+ * 
+ * Uses custom fetch to inject Google Cloud auth since 
+ * @ai-sdk/openai-compatible doesn't support async headers.
+ */
+
+const { createOpenAICompatible } = require('@ai-sdk/openai-compatible');
+const { GoogleAuth } = require('google-auth-library');
+
+let authClient = null;
+
+async function getAuthToken(googleAuthOptions) {
+  if (!authClient) {
+    authClient = new GoogleAuth({
+      scopes: ['https://www.googleapis.com/auth/cloud-platform'],
+      ...googleAuthOptions,
+    });
+  }
+  
+  const client = await authClient.getClient();
+  const token = await client.getAccessToken();
+  return token.token;
+}
+
+function createProvider(options = {}) {
+  const {
+    project = process.env.GOOGLE_VERTEX_PROJECT,
+    location = process.env.GOOGLE_VERTEX_LOCATION || 'global',
+    publisher,
+    googleAuthOptions,
+    ...rest
+  } = options;
+
+  if (!project) throw new Error('project is required');
+  if (!publisher) throw new Error('publisher is required');
+
+  const baseHost = location === 'global' 
+    ? 'aiplatform.googleapis.com' 
+    : `${location}-aiplatform.googleapis.com`;
+  
+  const baseURL = `https://${baseHost}/v1/projects/${project}/locations/${location}/endpoints/openapi`;
+
+  console.error(`[vertex-partner] baseURL: ${baseURL}`);
+
+  // Custom fetch that injects Google auth
+  const authFetch = async (url, init) => {
+    console.error(`[vertex-partner] Fetching: ${url}`);
+    
+    const token = await getAuthToken(googleAuthOptions);
+    console.error(`[vertex-partner] Got token: ${token.substring(0, 20)}...`);
+    
+    const headers = new Headers(init?.headers);
+    headers.set('Authorization', `Bearer ${token}`);
+    
+    return fetch(url, {
+      ...init,
+      headers,
+    });
+  };
+
+  const provider = createOpenAICompatible({
+    name: `vertex-${publisher}`,
+    baseURL,
+    fetch: authFetch,  // Use custom fetch for auth
+    ...rest,
+  });
+
+  // Wrap to auto-prefix model IDs
+  const wrappedProvider = (modelId) => {
+    const fullModelId = modelId.includes('/') ? modelId : `${publisher}/${modelId}`;
+    console.error(`[vertex-partner] Using model: ${fullModelId}`);
+    return provider(fullModelId);
+  };
+
+  wrappedProvider.specificationVersion = provider.specificationVersion;
+  wrappedProvider.languageModel = (id) => wrappedProvider(id);
+  wrappedProvider.chatModel = (id) => wrappedProvider(id);
+  wrappedProvider.completionModel = provider.completionModel;
+  wrappedProvider.embeddingModel = provider.embeddingModel;
+  wrappedProvider.textEmbeddingModel = provider.textEmbeddingModel;
+  wrappedProvider.imageModel = provider.imageModel;
+
+  return wrappedProvider;
+}
+
+module.exports = createProvider;
+module.exports.createProvider = createProvider;
+module.exports.createVertexPartner = createProvider;
+module.exports.createGLM = (opts) => createProvider({ ...opts, publisher: 'zai-org' });
+module.exports.createKimi = (opts) => createProvider({ ...opts, publisher: 'moonshotai' });
+module.exports.createDeepSeek = (opts) => createProvider({ ...opts, publisher: 'deepseek-ai' });
+module.exports.createMiniMax = (opts) => createProvider({ ...opts, publisher: 'minimaxai' });
+module.exports.createQwen = (opts) => createProvider({ ...opts, publisher: 'qwen' });

package/package.json

@@ -0,0 +1,19 @@
+{
+  "name": "@serii84/vertex-partner-provider",
+  "version": "1.0.0",
+  "description": "",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "@ai-sdk/openai-compatible": "^2.0.4",
+    "@ai-sdk/provider-utils": "^4.0.4",
+    "ai": "^6.0.27",
+    "google-auth-library": "^10.5.0",
+    "zod": "^4.3.5"
+  }
+}

package/test.js

@@ -0,0 +1,78 @@
+/**
+ * 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();