@contractspec/lib.ai-providers 3.0.0 → 3.1.1

package/dist/node/legacy.js

@@ -288,22 +288,30 @@ function getDefaultModel(provider) {
 }
 
 // src/factory.ts
-import { anthropic } from "@ai-sdk/anthropic";
-import { google } from "@ai-sdk/google";
-import { mistral } from "@ai-sdk/mistral";
-import { openai } from "@ai-sdk/openai";
-import { ollama } from "ollama-ai-provider";
+import { createAnthropic } from "@ai-sdk/anthropic";
+import { createGoogleGenerativeAI } from "@ai-sdk/google";
+import { createMistral } from "@ai-sdk/mistral";
+import { createOpenAI } from "@ai-sdk/openai";
+import { createOllama } from "ollama-ai-provider";
 class BaseProvider {
   name;
   model;
   mode;
   config;
+  transport;
+  authMethod;
+  apiVersion;
+  customHeaders;
   cachedModel = null;
   constructor(config) {
     this.name = config.provider;
     this.model = config.model ?? DEFAULT_MODELS[config.provider];
     this.mode = this.determineMode(config);
     this.config = config;
+    this.transport = config.transport;
+    this.authMethod = config.authMethod;
+    this.apiVersion = config.apiVersion;
+    this.customHeaders = config.customHeaders;
   }
   getModel() {
     if (!this.cachedModel) {
@@ -343,81 +351,33 @@ class BaseProvider {
     return "managed";
   }
   createModel() {
-    const { baseUrl, proxyUrl } = this.config;
+    const { baseUrl, proxyUrl, apiKey } = this.config;
+    const headers = this.customHeaders;
+    if (this.name === "ollama") {
+      const provider = createOllama({ baseURL: baseUrl, headers });
+      return provider(this.model);
+    }
+    if (this.mode === "managed" && proxyUrl) {
+      const provider = createOpenAI({ baseURL: proxyUrl, apiKey, headers });
+      return provider(this.model);
+    }
     switch (this.name) {
-      case "ollama": {
-        const originalBaseUrl = process.env.OLLAMA_BASE_URL;
-        if (baseUrl && baseUrl !== "http://localhost:11434") {
-          process.env.OLLAMA_BASE_URL = baseUrl;
-        }
-        const ollamaModel = ollama(this.model);
-        if (originalBaseUrl !== undefined) {
-          process.env.OLLAMA_BASE_URL = originalBaseUrl;
-        } else if (baseUrl && baseUrl !== "http://localhost:11434") {
-          delete process.env.OLLAMA_BASE_URL;
-        }
-        return ollamaModel;
+      case "openai": {
+        const provider = createOpenAI({ apiKey, headers });
+        return provider(this.model);
+      }
+      case "anthropic": {
+        const provider = createAnthropic({ apiKey, headers });
+        return provider(this.model);
+      }
+      case "mistral": {
+        const provider = createMistral({ apiKey, headers });
+        return provider(this.model);
+      }
+      case "gemini": {
+        const provider = createGoogleGenerativeAI({ apiKey, headers });
+        return provider(this.model);
       }
-      case "openai":
-        if (this.mode === "managed") {
-          const originalBaseUrl = process.env.OPENAI_BASE_URL;
-          if (proxyUrl) {
-            process.env.OPENAI_BASE_URL = proxyUrl;
-          }
-          const model = openai(this.model);
-          if (originalBaseUrl !== undefined) {
-            process.env.OPENAI_BASE_URL = originalBaseUrl;
-          } else if (proxyUrl) {
-            delete process.env.OPENAI_BASE_URL;
-          }
-          return model;
-        }
-        return openai(this.model);
-      case "anthropic":
-        if (this.mode === "managed") {
-          const originalBaseUrl = process.env.OPENAI_BASE_URL;
-          if (proxyUrl) {
-            process.env.OPENAI_BASE_URL = proxyUrl;
-          }
-          const model = openai(this.model);
-          if (originalBaseUrl !== undefined) {
-            process.env.OPENAI_BASE_URL = originalBaseUrl;
-          } else if (proxyUrl) {
-            delete process.env.OPENAI_BASE_URL;
-          }
-          return model;
-        }
-        return anthropic(this.model);
-      case "mistral":
-        if (this.mode === "managed") {
-          const originalBaseUrl = process.env.OPENAI_BASE_URL;
-          if (proxyUrl) {
-            process.env.OPENAI_BASE_URL = proxyUrl;
-          }
-          const model = openai(this.model);
-          if (originalBaseUrl !== undefined) {
-            process.env.OPENAI_BASE_URL = originalBaseUrl;
-          } else if (proxyUrl) {
-            delete process.env.OPENAI_BASE_URL;
-          }
-          return model;
-        }
-        return mistral(this.model);
-      case "gemini":
-        if (this.mode === "managed") {
-          const originalBaseUrl = process.env.OPENAI_BASE_URL;
-          if (proxyUrl) {
-            process.env.OPENAI_BASE_URL = proxyUrl;
-          }
-          const model = openai(this.model);
-          if (originalBaseUrl !== undefined) {
-            process.env.OPENAI_BASE_URL = originalBaseUrl;
-          } else if (proxyUrl) {
-            delete process.env.OPENAI_BASE_URL;
-          }
-          return model;
-        }
-        return google(this.model);
       default:
         throw new Error(`Unknown provider: ${this.name}`);
     }
@@ -499,13 +459,17 @@ function createProviderFromEnv() {
     case "ollama":
       break;
   }
+  const transport = process.env.CONTRACTSPEC_AI_TRANSPORT;
+  const apiVersion = process.env.CONTRACTSPEC_AI_API_VERSION;
   return createProvider({
     provider,
     model,
     apiKey,
     baseUrl: process.env.OLLAMA_BASE_URL,
     proxyUrl: process.env.CONTRACTSPEC_AI_PROXY_URL,
-    organizationId: process.env.CONTRACTSPEC_ORG_ID
+    organizationId: process.env.CONTRACTSPEC_ORG_ID,
+    transport,
+    apiVersion
   });
 }
 function getAvailableProviders() {
@@ -513,35 +477,45 @@ function getAvailableProviders() {
   providers.push({
     provider: "ollama",
     available: true,
-    mode: "local"
+    mode: "local",
+    transports: ["rest", "sdk"],
+    authMethods: []
   });
   const openaiKey = process.env.OPENAI_API_KEY;
   providers.push({
     provider: "openai",
     available: Boolean(openaiKey) || Boolean(process.env.CONTRACTSPEC_AI_PROXY_URL),
     mode: openaiKey ? "byok" : "managed",
-    reason: !openaiKey ? "Set OPENAI_API_KEY for BYOK mode" : undefined
+    reason: !openaiKey ? "Set OPENAI_API_KEY for BYOK mode" : undefined,
+    transports: ["rest", "sdk"],
+    authMethods: ["api-key"]
   });
   const anthropicKey = process.env.ANTHROPIC_API_KEY;
   providers.push({
     provider: "anthropic",
     available: Boolean(anthropicKey) || Boolean(process.env.CONTRACTSPEC_AI_PROXY_URL),
     mode: anthropicKey ? "byok" : "managed",
-    reason: !anthropicKey ? "Set ANTHROPIC_API_KEY for BYOK mode" : undefined
+    reason: !anthropicKey ? "Set ANTHROPIC_API_KEY for BYOK mode" : undefined,
+    transports: ["rest", "sdk"],
+    authMethods: ["api-key"]
   });
   const mistralKey = process.env.MISTRAL_API_KEY;
   providers.push({
     provider: "mistral",
     available: Boolean(mistralKey) || Boolean(process.env.CONTRACTSPEC_AI_PROXY_URL),
     mode: mistralKey ? "byok" : "managed",
-    reason: !mistralKey ? "Set MISTRAL_API_KEY for BYOK mode" : undefined
+    reason: !mistralKey ? "Set MISTRAL_API_KEY for BYOK mode" : undefined,
+    transports: ["rest", "sdk"],
+    authMethods: ["api-key"]
   });
   const geminiKey = process.env.GOOGLE_API_KEY ?? process.env.GEMINI_API_KEY;
   providers.push({
     provider: "gemini",
     available: Boolean(geminiKey) || Boolean(process.env.CONTRACTSPEC_AI_PROXY_URL),
     mode: geminiKey ? "byok" : "managed",
-    reason: !geminiKey ? "Set GOOGLE_API_KEY for BYOK mode" : undefined
+    reason: !geminiKey ? "Set GOOGLE_API_KEY for BYOK mode" : undefined,
+    transports: ["rest", "sdk"],
+    authMethods: ["api-key"]
   });
   return providers;
 }