UNPKG

@llumiverse/common

Version:

Public types, enums and options used by Llumiverse API.

github.com/vertesia/llumiverse

vertesia/llumiverse

50 lines (42 loc) • 3.57 kB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import type { ModelCapabilities } from "../types.js"; // Explicit model exceptions const RECORD_MODEL_CAPABILITIES: Record<string, ModelCapabilities> = { // claude-3-5-haiku: no image input "claude-3-5-haiku-20241022": { 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, tool_support_streaming: true }, }; // Family-level capabilities (longest prefix match) const RECORD_FAMILY_CAPABILITIES: Record<string, ModelCapabilities> = { "claude-3-5-haiku": { 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, tool_support_streaming: true }, "claude-3-haiku": { 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, tool_support_streaming: true }, "claude-3-sonnet": { 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, tool_support_streaming: true }, "claude-3-opus": { 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, tool_support_streaming: true }, "claude-3-5": { 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, tool_support_streaming: true }, "claude-3-7": { 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, tool_support_streaming: true }, "claude-3": { 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, tool_support_streaming: true }, "claude-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, tool_support_streaming: true }, "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, tool_support_streaming: true }, }; const DEFAULT_CLAUDE_CAPABILITIES: ModelCapabilities = { 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, tool_support_streaming: true, }; export function getModelCapabilitiesAnthropic(model: string): ModelCapabilities { const lower = model.toLowerCase(); // Exact match first if (lower in RECORD_MODEL_CAPABILITIES) { return RECORD_MODEL_CAPABILITIES[lower]; } // Longest prefix family match let bestKey: string | undefined; for (const key of Object.keys(RECORD_FAMILY_CAPABILITIES)) { if (lower.startsWith(key) && (!bestKey || key.length > bestKey.length)) { bestKey = key; } } if (bestKey) { return RECORD_FAMILY_CAPABILITIES[bestKey]; } return DEFAULT_CLAUDE_CAPABILITIES; }