@llumiverse/common 1.2.0 → 1.4.0-dev.20260629.090753Z

package/src/capability/vertexai.ts

@@ -1,46 +1,112 @@
-import { ModelModalities } from "../types.js";
+import type { ModelModalities } from '../types.js';
 
 // Record of Vertex AI model capabilities keyed by model ID (last path segment, lowercased)
-const RECORD_MODEL_CAPABILITIES: Record<string, { input: ModelModalities; output: ModelModalities; tool_support?: boolean }> = {
-    "gemini-2.0-flash-lite-001": { input: { text: true, image: true, video: true, audio: true, embed: false }, output: { text: true, image: false, video: false, audio: false, embed: false }, tool_support: false },
+const RECORD_MODEL_CAPABILITIES: Record<
+    string,
+    { input: ModelModalities; output: ModelModalities; tool_support?: boolean }
+> = {
+    'gemini-2.0-flash-lite-001': {
+        input: { text: true, image: true, video: true, audio: true, embed: false },
+        output: { text: true, image: false, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
 };
 
 // Populate RECORD_FAMILY_CAPABILITIES as a const record (lowest common denominator for each family)
-const RECORD_FAMILY_CAPABILITIES: Record<string, { input: ModelModalities; output: ModelModalities; tool_support?: boolean }> = {
-    "gemini":                   { input: { text: true, image: true, video: true, audio: true, embed: false }, output: { text: true, image: false, video: false, audio: false, embed: false }, tool_support: true },
-    "gemini-2.5-flash-image":   { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: true, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    "gemini-3.0-pro-image":     { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: true, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    "gemini-3.0-flash-image":   { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: true, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    'imagen':                   { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: false, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    "imagen-3.0":               { input: { text: true, image: false, video: false, audio: false, embed: false }, output: { text: false, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    "imagen-3.0-capability":    { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: false, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    "imagen-4.0":               { input: { text: true, image: false, video: false, audio: false, embed: false }, output: { text: false, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    "imagen-4.0-capability":    { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: false, image: true, video: false, audio: false, embed: false }, tool_support: false },
-    "claude":                   { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: true, image: false, video: false, audio: false, embed: false }, tool_support: true },
-    "llama":                    { input: { text: true, image: false, video: false, audio: false, embed: false }, output: { text: true, image: false, video: false, audio: false, embed: false }, tool_support: true },
-    "llama-2":                  { input: { text: true, image: false, video: false, audio: false, embed: false }, output: { text: true, image: false, video: false, audio: false, embed: false }, tool_support: false },
-    "llama-3.2":                { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: true, image: false, video: false, audio: false, embed: false }, tool_support: true },
-    "llama-4":                  { input: { text: true, image: true, video: false, audio: false, embed: false }, output: { text: true, image: false, video: false, audio: false, embed: false }, tool_support: true },
+const RECORD_FAMILY_CAPABILITIES: Record<
+    string,
+    { input: ModelModalities; output: ModelModalities; tool_support?: boolean }
+> = {
+    gemini: {
+        input: { text: true, image: true, video: true, audio: true, embed: false },
+        output: { text: true, image: false, video: false, audio: false, embed: false },
+        tool_support: true,
+    },
+    'gemini-2.5-flash-image': {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: true, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    'gemini-3.0-pro-image': {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: true, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    'gemini-3.0-flash-image': {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: true, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    imagen: {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: false, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    'imagen-3.0': {
+        input: { text: true, image: false, video: false, audio: false, embed: false },
+        output: { text: false, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    'imagen-3.0-capability': {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: false, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    'imagen-4.0': {
+        input: { text: true, image: false, video: false, audio: false, embed: false },
+        output: { text: false, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    'imagen-4.0-capability': {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: false, image: true, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    claude: {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: true, image: false, video: false, audio: false, embed: false },
+        tool_support: true,
+    },
+    llama: {
+        input: { text: true, image: false, video: false, audio: false, embed: false },
+        output: { text: true, image: false, video: false, audio: false, embed: false },
+        tool_support: true,
+    },
+    'llama-2': {
+        input: { text: true, image: false, video: false, audio: false, embed: false },
+        output: { text: true, image: false, video: false, audio: false, embed: false },
+        tool_support: false,
+    },
+    'llama-3.2': {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: true, image: false, video: false, audio: false, embed: false },
+        tool_support: true,
+    },
+    'llama-4': {
+        input: { text: true, image: true, video: false, audio: false, embed: false },
+        output: { text: true, image: false, video: false, audio: false, embed: false },
+        tool_support: true,
+    },
 };
 
 // Fallback pattern lists for inferring modalities and tool support
-const IMAGE_INPUT_MODELS = ["image"];
-const VIDEO_INPUT_MODELS = ["video"];
-const AUDIO_INPUT_MODELS = ["audio"];
-const TEXT_INPUT_MODELS = ["text"];
-const IMAGE_OUTPUT_MODELS = ["image"];
-const VIDEO_OUTPUT_MODELS = ["video"];
-const AUDIO_OUTPUT_MODELS = ["audio"];
-const TEXT_OUTPUT_MODELS = ["text"];
-const EMBEDDING_OUTPUT_MODELS = ["embed"];
-const TOOL_SUPPORT_MODELS = ["tool", "sonnet", "opus", "gemini", "claude"];
+const IMAGE_INPUT_MODELS = ['image'];
+const VIDEO_INPUT_MODELS = ['video'];
+const AUDIO_INPUT_MODELS = ['audio'];
+const TEXT_INPUT_MODELS = ['text'];
+const IMAGE_OUTPUT_MODELS = ['image'];
+const VIDEO_OUTPUT_MODELS = ['video'];
+const AUDIO_OUTPUT_MODELS = ['audio'];
+const TEXT_OUTPUT_MODELS = ['text'];
+const EMBEDDING_OUTPUT_MODELS = ['embed'];
+const TOOL_SUPPORT_MODELS = ['tool', 'sonnet', 'opus', 'gemini', 'claude'];
 
 function modelMatches(modelName: string, patterns: string[]): boolean {
-    return patterns.some(pattern => modelName.includes(pattern));
+    return patterns.some((pattern) => modelName.includes(pattern));
 }
 
 function normalizeVertexAIModelName(modelName: string): string {
-    const segments = modelName.toLowerCase().split("/");
+    const segments = modelName.toLowerCase().split('/');
     return segments[segments.length - 1];
 }
 
@@ -48,11 +114,15 @@ function normalizeVertexAIModelName(modelName: string): string {
  * Get the full ModelCapabilities for a Vertex AI model.
  * Checks RECORD_MODEL_CAPABILITIES first, then falls back to pattern-based inference.
  */
-export function getModelCapabilitiesVertexAI(model: string): { input: ModelModalities; output: ModelModalities; tool_support?: boolean } {
+export function getModelCapabilitiesVertexAI(model: string): {
+    input: ModelModalities;
+    output: ModelModalities;
+    tool_support?: boolean;
+} {
     const normalized = normalizeVertexAIModelName(model);
     const record = RECORD_MODEL_CAPABILITIES[normalized];
     if (record) return record;
-    let bestFamilyKey = undefined;
+    let bestFamilyKey: string | undefined;
     let bestFamilyLength = 0;
     for (const key of Object.keys(RECORD_FAMILY_CAPABILITIES)) {
         if (normalized.startsWith(key) && key.length > bestFamilyLength) {
@@ -68,15 +138,15 @@ export function getModelCapabilitiesVertexAI(model: string): { input: ModelModal
         image: modelMatches(normalized, IMAGE_INPUT_MODELS) || undefined,
         video: modelMatches(normalized, VIDEO_INPUT_MODELS) || undefined,
         audio: modelMatches(normalized, AUDIO_INPUT_MODELS) || undefined,
-        embed: false
+        embed: false,
     };
     const output: ModelModalities = {
         text: modelMatches(normalized, TEXT_OUTPUT_MODELS) || undefined,
         image: modelMatches(normalized, IMAGE_OUTPUT_MODELS) || undefined,
         video: modelMatches(normalized, VIDEO_OUTPUT_MODELS) || undefined,
         audio: modelMatches(normalized, AUDIO_OUTPUT_MODELS) || undefined,
-        embed: modelMatches(normalized, EMBEDDING_OUTPUT_MODELS) || undefined
+        embed: modelMatches(normalized, EMBEDDING_OUTPUT_MODELS) || undefined,
     };
     const tool_support = modelMatches(normalized, TOOL_SUPPORT_MODELS) || undefined;
     return { input, output, tool_support };
-}
+}

package/src/capability.ts

@@ -1,15 +1,16 @@
-import { getModelCapabilitiesAzureFoundry } from "./capability/azure_foundry.js";
-import { getModelCapabilitiesBedrock } from "./capability/bedrock.js";
-import { getModelCapabilitiesOpenAI } from "./capability/openai.js";
-import { getModelCapabilitiesVertexAI } from "./capability/vertexai.js";
-import { ModelCapabilities, ModelModalities, Providers } from "./types.js";
+import { getModelCapabilitiesAnthropic } from './capability/anthropic.js';
+import { getModelCapabilitiesAzureFoundry } from './capability/azure_foundry.js';
+import { getModelCapabilitiesBedrock } from './capability/bedrock.js';
+import { getModelCapabilitiesOpenAI } from './capability/openai.js';
+import { getModelCapabilitiesVertexAI } from './capability/vertexai.js';
+import { type ModelCapabilities, type ModelModalities, Providers } from './types.js';
 
 export function getModelCapabilities(model: string, provider?: string | Providers): ModelCapabilities {
     //Check for locations/<location>/ prefix and remove it
-    if (model.startsWith("locations/")) {
-        const parts = model.split("/");
+    if (model.startsWith('locations/')) {
+        const parts = model.split('/');
         if (parts.length >= 3) {
-            model = parts.slice(2).join("/");
+            model = parts.slice(2).join('/');
         }
     }
     const capabilities = _getModelCapabilities(model, provider);
@@ -25,6 +26,8 @@ export function getModelCapabilities(model: string, provider?: string | Provider
 
 function _getModelCapabilities(model: string, provider?: string | Providers): ModelCapabilities {
     switch (provider?.toLowerCase()) {
+        case Providers.anthropic:
+            return getModelCapabilitiesAnthropic(model);
         case Providers.vertexai:
             return getModelCapabilitiesVertexAI(model);
         case Providers.openai:
@@ -44,26 +47,28 @@ function _getModelCapabilities(model: string, provider?: string | Providers): Mo
         case Providers.xai:
             // xAI (Grok) models support tool use and are text-based
             return {
-                input: { text: true, image: model.includes("vision") },
+                input: { text: true, image: model.includes('vision') },
                 output: { text: true },
                 tool_support: true,
                 tool_support_streaming: false, // Conservative - may work but not tested
             };
         default:
             // Guess the provider based on the model name
-            if (model.startsWith("gpt")) {
+            if (model.startsWith('gpt')) {
                 return getModelCapabilitiesOpenAI(model);
-            } else if (model.startsWith("grok")) {
+            } else if (model.startsWith('claude')) {
+                return getModelCapabilitiesAnthropic(model);
+            } else if (model.startsWith('grok')) {
                 // xAI Grok models
                 return {
-                    input: { text: true, image: model.includes("vision") },
+                    input: { text: true, image: model.includes('vision') },
                     output: { text: true },
                     tool_support: true,
                     tool_support_streaming: false,
                 };
-            } else if (model.startsWith("publishers/")) {
+            } else if (model.startsWith('publishers/')) {
                 return getModelCapabilitiesVertexAI(model);
-            } else if (model.startsWith("arn:aws")) {
+            } else if (model.startsWith('arn:aws')) {
                 return getModelCapabilitiesBedrock(model);
             }
             // Fallback to a generic model with no capabilities
@@ -85,7 +90,7 @@ function getModelCapabilitiesOpenAICompatible(model: string): ModelCapabilities
         return caps;
     }
     const normalized = model.toLowerCase();
-    const isNonToolModel = NO_TOOL_SUPPORT_PATTERNS.some(p => normalized.includes(p));
+    const isNonToolModel = NO_TOOL_SUPPORT_PATTERNS.some((p) => normalized.includes(p));
     return {
         input: { text: true },
         output: { text: true },
@@ -103,4 +108,4 @@ export function modelModalitiesToArray(modalities: ModelModalities): string[] {
     return Object.entries(modalities)
         .filter(([_, isSupported]) => isSupported)
         .map(([modality]) => modality);
-}
+}

package/src/index.ts

@@ -1,11 +1,13 @@
-export * from "./types.js";
-export * from "./capability.js";
-export * from "./options.js";
-
-export * from "./options/bedrock.js";
-export * from "./options/fallback.js";
-export * from "./options/groq.js";
-export * from "./options/openai.js";
-export * from "./options/vertexai.js";
-export * from "./options/context-windows.js";
-export * from "./options/version-parsing.js";
+export * from './capability.js';
+export * from './options/anthropic.js';
+export * from './options/bedrock.js';
+export * from './options/context-windows.js';
+export * from './options/embedding.js';
+export * from './options/fallback.js';
+export * from './options/groq.js';
+export * from './options/openai.js';
+export * from './options/shared-parsing.js';
+export * from './options/version-parsing.js';
+export * from './options/vertexai.js';
+export * from './options.js';
+export * from './types.js';

package/src/options/anthropic.ts

@@ -0,0 +1,59 @@
+import { type ModelOptions, type ModelOptionsInfo, OptionType, SharedOptions } from '../types.js';
+import { textOptionsFallback } from './fallback.js';
+import {
+    buildClaudeCacheOptions,
+    buildClaudeCacheTtlOptions,
+    buildClaudeEffortOptions,
+    buildClaudeIncludeThoughtsOption,
+    buildClaudeThinkingBudgetOption,
+    getClaudeMaxTokensLimit,
+} from './shared-parsing.js';
+import { hasSamplingParameterRestriction } from './version-parsing.js';
+
+export interface AnthropicClaudeOptions {
+    _option_id: 'anthropic-claude';
+    max_tokens?: number;
+    temperature?: number;
+    top_p?: number;
+    top_k?: number;
+    stop_sequence?: string[];
+    effort?: 'low' | 'medium' | 'high' | 'xhigh' | 'max';
+    thinking_budget_tokens?: number;
+    include_thoughts?: boolean;
+    cache_enabled?: boolean;
+    cache_ttl?: '5m' | '1h';
+}
+
+export function getAnthropicOptions(model: string, option?: ModelOptions): ModelOptionsInfo {
+    const max_tokens_limit = getClaudeMaxTokensLimit(model);
+    const excludeOptions = ['max_tokens', 'presence_penalty', 'frequency_penalty'];
+    let commonOptions = textOptionsFallback.options.filter((o) => !excludeOptions.includes(o.name));
+
+    const hasSamplingRestriction = hasSamplingParameterRestriction(model);
+    if (hasSamplingRestriction) {
+        commonOptions = commonOptions.filter(
+            (o) => o.name !== SharedOptions.temperature && o.name !== SharedOptions.top_p && o.name !== 'top_k',
+        );
+    }
+
+    return {
+        _option_id: 'anthropic-claude',
+        options: [
+            {
+                name: SharedOptions.max_tokens,
+                type: OptionType.numeric,
+                min: 1,
+                max: max_tokens_limit,
+                integer: true,
+                step: 200,
+                description: 'The maximum number of tokens to generate',
+            },
+            ...commonOptions,
+            ...buildClaudeEffortOptions(model),
+            ...buildClaudeThinkingBudgetOption(model),
+            ...buildClaudeIncludeThoughtsOption(model),
+            ...buildClaudeCacheOptions(),
+            ...buildClaudeCacheTtlOptions((option as unknown as AnthropicClaudeOptions)?.cache_enabled),
+        ],
+    };
+}