@contractspec/lib.ai-providers 1.57.0 → 1.58.0

package/dist/browser/models.js

@@ -0,0 +1,248 @@
+// 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];
+}
+export {
+  getRecommendedModels,
+  getModelsForProvider,
+  getModelInfo,
+  getDefaultModel,
+  MODELS,
+  DEFAULT_MODELS
+};