@contractspec/lib.ai-providers 1.57.0 → 1.58.0

package/dist/legacy.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"legacy.d.ts","names":[],"sources":["../src/legacy.ts"],"mappings":";;;;;;;;;;iBA8BgB,aAAA,CAAc,MAAA,EAAQ,YAAA,GAAe,aAAA;;;;;;iBAmC/B,gBAAA,CACpB,MAAA,EAAQ,YAAA,GACP,OAAA;EAAU,OAAA;EAAkB,KAAA;AAAA"}
+{"version":3,"file":"legacy.d.ts","sourceRoot":"","sources":["../src/legacy.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,YAAY,EAAgB,MAAM,SAAS,CAAC;AAE1D,OAAO,EAAE,oBAAoB,IAAI,SAAS,EAAE,MAAM,UAAU,CAAC;AAgB7D;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,MAAM,EAAE,YAAY,GAAG,aAAa,CA4BjE;AAED;;;;GAIG;AACH,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,YAAY,GACnB,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CA0C/C;AAED;;;;GAIG;AACH,OAAO,EAAE,SAAS,IAAI,oBAAoB,EAAE,CAAC"}

package/dist/legacy.js

@@ -1,73 +1,573 @@
-import { getRecommendedModels } from "./models.js";
-import { createProvider } from "./factory.js";
+// @bun
+// src/models.ts
+var DEFAULT_MODELS = {
+  ollama: "llama3.2",
+  openai: "gpt-4o",
+  anthropic: "claude-sonnet-4-20250514",
+  mistral: "mistral-large-latest",
+  gemini: "gemini-2.0-flash"
+};
+var MODELS = [
+  {
+    id: "llama3.2",
+    name: "Llama 3.2",
+    provider: "ollama",
+    contextWindow: 128000,
+    capabilities: {
+      vision: false,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    }
+  },
+  {
+    id: "codellama",
+    name: "Code Llama",
+    provider: "ollama",
+    contextWindow: 16000,
+    capabilities: {
+      vision: false,
+      tools: false,
+      reasoning: false,
+      streaming: true
+    }
+  },
+  {
+    id: "deepseek-coder",
+    name: "DeepSeek Coder",
+    provider: "ollama",
+    contextWindow: 16000,
+    capabilities: {
+      vision: false,
+      tools: false,
+      reasoning: false,
+      streaming: true
+    }
+  },
+  {
+    id: "mistral",
+    name: "Mistral 7B",
+    provider: "ollama",
+    contextWindow: 32000,
+    capabilities: {
+      vision: false,
+      tools: false,
+      reasoning: false,
+      streaming: true
+    }
+  },
+  {
+    id: "gpt-4o",
+    name: "GPT-4o",
+    provider: "openai",
+    contextWindow: 128000,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 2.5, output: 10 }
+  },
+  {
+    id: "gpt-4o-mini",
+    name: "GPT-4o Mini",
+    provider: "openai",
+    contextWindow: 128000,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 0.15, output: 0.6 }
+  },
+  {
+    id: "o1",
+    name: "o1",
+    provider: "openai",
+    contextWindow: 200000,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: true,
+      streaming: true
+    },
+    costPerMillion: { input: 15, output: 60 }
+  },
+  {
+    id: "o1-mini",
+    name: "o1 Mini",
+    provider: "openai",
+    contextWindow: 128000,
+    capabilities: {
+      vision: false,
+      tools: true,
+      reasoning: true,
+      streaming: true
+    },
+    costPerMillion: { input: 3, output: 12 }
+  },
+  {
+    id: "claude-sonnet-4-20250514",
+    name: "Claude Sonnet 4",
+    provider: "anthropic",
+    contextWindow: 200000,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: true,
+      streaming: true
+    },
+    costPerMillion: { input: 3, output: 15 }
+  },
+  {
+    id: "claude-3-5-sonnet-20241022",
+    name: "Claude 3.5 Sonnet",
+    provider: "anthropic",
+    contextWindow: 200000,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 3, output: 15 }
+  },
+  {
+    id: "claude-3-5-haiku-20241022",
+    name: "Claude 3.5 Haiku",
+    provider: "anthropic",
+    contextWindow: 200000,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 0.8, output: 4 }
+  },
+  {
+    id: "mistral-large-latest",
+    name: "Mistral Large",
+    provider: "mistral",
+    contextWindow: 128000,
+    capabilities: {
+      vision: false,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 2, output: 6 }
+  },
+  {
+    id: "codestral-latest",
+    name: "Codestral",
+    provider: "mistral",
+    contextWindow: 32000,
+    capabilities: {
+      vision: false,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 0.2, output: 0.6 }
+  },
+  {
+    id: "mistral-small-latest",
+    name: "Mistral Small",
+    provider: "mistral",
+    contextWindow: 32000,
+    capabilities: {
+      vision: false,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 0.2, output: 0.6 }
+  },
+  {
+    id: "gemini-2.0-flash",
+    name: "Gemini 2.0 Flash",
+    provider: "gemini",
+    contextWindow: 1e6,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: false,
+      streaming: true
+    },
+    costPerMillion: { input: 0.075, output: 0.3 }
+  },
+  {
+    id: "gemini-2.5-pro-preview-06-05",
+    name: "Gemini 2.5 Pro",
+    provider: "gemini",
+    contextWindow: 1e6,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: true,
+      streaming: true
+    },
+    costPerMillion: { input: 1.25, output: 10 }
+  },
+  {
+    id: "gemini-2.5-flash-preview-05-20",
+    name: "Gemini 2.5 Flash",
+    provider: "gemini",
+    contextWindow: 1e6,
+    capabilities: {
+      vision: true,
+      tools: true,
+      reasoning: true,
+      streaming: true
+    },
+    costPerMillion: { input: 0.15, output: 0.6 }
+  }
+];
+function getModelsForProvider(provider) {
+  return MODELS.filter((m) => m.provider === provider);
+}
+function getModelInfo(modelId) {
+  return MODELS.find((m) => m.id === modelId);
+}
+function getRecommendedModels(provider) {
+  const normalizedProvider = provider === "claude" ? "anthropic" : provider === "custom" ? "openai" : provider;
+  return getModelsForProvider(normalizedProvider).map((m) => m.id);
+}
+function getDefaultModel(provider) {
+  return DEFAULT_MODELS[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";
+class BaseProvider {
+  name;
+  model;
+  mode;
+  config;
+  cachedModel = null;
+  constructor(config) {
+    this.name = config.provider;
+    this.model = config.model ?? DEFAULT_MODELS[config.provider];
+    this.mode = this.determineMode(config);
+    this.config = config;
+  }
+  getModel() {
+    if (!this.cachedModel) {
+      this.cachedModel = this.createModel();
+    }
+    return this.cachedModel;
+  }
+  async listModels() {
+    if (this.name === "ollama") {
+      return this.listOllamaModels();
+    }
+    return getModelsForProvider(this.name);
+  }
+  async validate() {
+    if (this.name === "ollama") {
+      return this.validateOllama();
+    }
+    if (this.mode === "byok" && !this.config.apiKey) {
+      return {
+        valid: false,
+        error: `API key required for ${this.name}`
+      };
+    }
+    if (this.mode === "managed" && !this.config.proxyUrl && !this.config.organizationId) {
+      return {
+        valid: false,
+        error: "Managed mode requires proxyUrl or organizationId"
+      };
+    }
+    return { valid: true };
+  }
+  determineMode(config) {
+    if (config.provider === "ollama")
+      return "local";
+    if (config.apiKey)
+      return "byok";
+    return "managed";
+  }
+  createModel() {
+    const { baseUrl, proxyUrl } = this.config;
+    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":
+        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}`);
+    }
+  }
+  async listOllamaModels() {
+    try {
+      const baseUrl = this.config.baseUrl ?? "http://localhost:11434";
+      const response = await fetch(`${baseUrl}/api/tags`);
+      if (!response.ok) {
+        return getModelsForProvider("ollama");
+      }
+      const data = await response.json();
+      const models = data.models ?? [];
+      return models.map((m) => ({
+        id: m.name,
+        name: m.name,
+        provider: "ollama",
+        contextWindow: 8000,
+        capabilities: {
+          vision: false,
+          tools: false,
+          reasoning: false,
+          streaming: true
+        }
+      }));
+    } catch {
+      return getModelsForProvider("ollama");
+    }
+  }
+  async validateOllama() {
+    try {
+      const baseUrl = this.config.baseUrl ?? "http://localhost:11434";
+      const response = await fetch(`${baseUrl}/api/tags`);
+      if (!response.ok) {
+        return {
+          valid: false,
+          error: `Ollama server returned ${response.status}`
+        };
+      }
+      const data = await response.json();
+      const models = data.models ?? [];
+      const hasModel = models.some((m) => m.name === this.model);
+      if (!hasModel) {
+        return {
+          valid: false,
+          error: `Model "${this.model}" not found. Available: ${models.map((m) => m.name).join(", ")}`
+        };
+      }
+      return { valid: true };
+    } catch (error) {
+      const baseUrl = this.config.baseUrl ?? "http://localhost:11434";
+      return {
+        valid: false,
+        error: `Cannot connect to Ollama at ${baseUrl}: ${error instanceof Error ? error.message : String(error)}`
+      };
+    }
+  }
+}
+function createProvider(config) {
+  return new BaseProvider(config);
+}
+function createProviderFromEnv() {
+  const provider = process.env.CONTRACTSPEC_AI_PROVIDER ?? "openai";
+  const model = process.env.CONTRACTSPEC_AI_MODEL;
+  let apiKey;
+  switch (provider) {
+    case "openai":
+      apiKey = process.env.OPENAI_API_KEY;
+      break;
+    case "anthropic":
+      apiKey = process.env.ANTHROPIC_API_KEY;
+      break;
+    case "mistral":
+      apiKey = process.env.MISTRAL_API_KEY;
+      break;
+    case "gemini":
+      apiKey = process.env.GOOGLE_API_KEY ?? process.env.GEMINI_API_KEY;
+      break;
+    case "ollama":
+      break;
+  }
+  return createProvider({
+    provider,
+    model,
+    apiKey,
+    baseUrl: process.env.OLLAMA_BASE_URL,
+    proxyUrl: process.env.CONTRACTSPEC_AI_PROXY_URL,
+    organizationId: process.env.CONTRACTSPEC_ORG_ID
+  });
+}
+function getAvailableProviders() {
+  const providers = [];
+  providers.push({
+    provider: "ollama",
+    available: true,
+    mode: "local"
+  });
+  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
+  });
+  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
+  });
+  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
+  });
+  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
+  });
+  return providers;
+}
 
-//#region src/legacy.ts
-/**
-* Map legacy provider names to new ones
-*/
+// src/legacy.ts
 function mapLegacyProvider(legacy) {
-	switch (legacy) {
-		case "claude": return "anthropic";
-		case "custom": return "openai";
-		default: return legacy;
-	}
+  switch (legacy) {
+    case "claude":
+      return "anthropic";
+    case "custom":
+      return "openai";
+    default:
+      return legacy;
+  }
 }
-/**
-* Get AI provider from legacy Config type
-*
-* @deprecated Use createProvider() instead
-*/
 function getAIProvider(config) {
-	const provider = mapLegacyProvider(config.aiProvider);
-	let apiKey;
-	switch (provider) {
-		case "openai":
-			apiKey = process.env.OPENAI_API_KEY;
-			break;
-		case "anthropic":
-			apiKey = process.env.ANTHROPIC_API_KEY;
-			break;
-		case "mistral":
-			apiKey = process.env.MISTRAL_API_KEY;
-			break;
-		case "gemini":
-			apiKey = process.env.GOOGLE_API_KEY ?? process.env.GEMINI_API_KEY;
-			break;
-	}
-	return createProvider({
-		provider,
-		model: config.aiModel,
-		apiKey,
-		baseUrl: config.customEndpoint
-	}).getModel();
+  const provider = mapLegacyProvider(config.aiProvider);
+  let apiKey;
+  switch (provider) {
+    case "openai":
+      apiKey = process.env.OPENAI_API_KEY;
+      break;
+    case "anthropic":
+      apiKey = process.env.ANTHROPIC_API_KEY;
+      break;
+    case "mistral":
+      apiKey = process.env.MISTRAL_API_KEY;
+      break;
+    case "gemini":
+      apiKey = process.env.GOOGLE_API_KEY ?? process.env.GEMINI_API_KEY;
+      break;
+  }
+  const instance = createProvider({
+    provider,
+    model: config.aiModel,
+    apiKey,
+    baseUrl: config.customEndpoint
+  });
+  return instance.getModel();
 }
-/**
-* Validate provider from legacy Config type
-*
-* @deprecated Use validateProvider() from './validation' instead
-*/
 async function validateProvider(config) {
-	const provider = mapLegacyProvider(config.aiProvider);
-	if (provider === "ollama") return { success: true };
-	if (provider === "anthropic" && !process.env.ANTHROPIC_API_KEY) return {
-		success: false,
-		error: "ANTHROPIC_API_KEY environment variable not set"
-	};
-	if (provider === "openai" && !process.env.OPENAI_API_KEY) return {
-		success: false,
-		error: "OPENAI_API_KEY environment variable not set"
-	};
-	if (provider === "mistral" && !process.env.MISTRAL_API_KEY) return {
-		success: false,
-		error: "MISTRAL_API_KEY environment variable not set"
-	};
-	if (provider === "gemini" && !process.env.GOOGLE_API_KEY && !process.env.GEMINI_API_KEY) return {
-		success: false,
-		error: "GOOGLE_API_KEY or GEMINI_API_KEY environment variable not set"
-	};
-	return { success: true };
+  const provider = mapLegacyProvider(config.aiProvider);
+  if (provider === "ollama") {
+    return { success: true };
+  }
+  if (provider === "anthropic" && !process.env.ANTHROPIC_API_KEY) {
+    return {
+      success: false,
+      error: "ANTHROPIC_API_KEY environment variable not set"
+    };
+  }
+  if (provider === "openai" && !process.env.OPENAI_API_KEY) {
+    return {
+      success: false,
+      error: "OPENAI_API_KEY environment variable not set"
+    };
+  }
+  if (provider === "mistral" && !process.env.MISTRAL_API_KEY) {
+    return {
+      success: false,
+      error: "MISTRAL_API_KEY environment variable not set"
+    };
+  }
+  if (provider === "gemini" && !process.env.GOOGLE_API_KEY && !process.env.GEMINI_API_KEY) {
+    return {
+      success: false,
+      error: "GOOGLE_API_KEY or GEMINI_API_KEY environment variable not set"
+    };
+  }
+  return { success: true };
 }
-
-//#endregion
-export { getAIProvider, getRecommendedModels, validateProvider };
-//# sourceMappingURL=legacy.js.map
+export {
+  validateProvider,
+  getRecommendedModels,
+  getAIProvider
+};

package/dist/models.d.ts

@@ -1,30 +1,29 @@
-import { ModelInfo, ProviderName } from "./types.js";
-
-//#region src/models.d.ts
+/**
+ * Model definitions and recommendations
+ */
+import type { ModelInfo, ProviderName } from './types';
 /**
  * Default models per provider
  */
-declare const DEFAULT_MODELS: Record<ProviderName, string>;
+export declare const DEFAULT_MODELS: Record<ProviderName, string>;
 /**
  * All recommended models with metadata
  */
-declare const MODELS: ModelInfo[];
+export declare const MODELS: ModelInfo[];
 /**
  * Get models for a specific provider
  */
-declare function getModelsForProvider(provider: ProviderName): ModelInfo[];
+export declare function getModelsForProvider(provider: ProviderName): ModelInfo[];
 /**
  * Get model info by ID
  */
-declare function getModelInfo(modelId: string): ModelInfo | undefined;
+export declare function getModelInfo(modelId: string): ModelInfo | undefined;
 /**
  * Get recommended models for a provider (legacy format)
  */
-declare function getRecommendedModels(provider: ProviderName | 'claude' | 'custom'): string[];
+export declare function getRecommendedModels(provider: ProviderName | 'claude' | 'custom'): string[];
 /**
  * Get default model for a provider
  */
-declare function getDefaultModel(provider: ProviderName): string;
-//#endregion
-export { DEFAULT_MODELS, MODELS, getDefaultModel, getModelInfo, getModelsForProvider, getRecommendedModels };
+export declare function getDefaultModel(provider: ProviderName): string;
 //# sourceMappingURL=models.d.ts.map

package/dist/models.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"models.d.ts","names":[],"sources":["../src/models.ts"],"mappings":";;;;;;cAQa,cAAA,EAAgB,MAAA,CAAO,YAAA;AAWpC;;;AAAA,cAAa,MAAA,EAAQ,SAAA;;AAwOrB;;iBAAgB,oBAAA,CAAqB,QAAA,EAAU,YAAA,GAAe,SAAA;;;;iBAO9C,YAAA,CAAa,OAAA,WAAkB,SAAA;;;AAA/C;iBAOgB,oBAAA,CACd,QAAA,EAAU,YAAA;;;;iBAkBI,eAAA,CAAgB,QAAA,EAAU,YAAA"}
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAEvD;;GAEG;AACH,eAAO,MAAM,cAAc,EAAE,MAAM,CAAC,YAAY,EAAE,MAAM,CAMvD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,MAAM,EAAE,SAAS,EAmO7B,CAAC;AAEF;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,YAAY,GAAG,SAAS,EAAE,CAExE;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,CAEnE;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,YAAY,GAAG,QAAQ,GAAG,QAAQ,GAC3C,MAAM,EAAE,CAYV;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,YAAY,GAAG,MAAM,CAE9D"}