feat: update output limit

paradite · Jul 31, 2024 · f38e348 · f38e348
1 parent acbea40
commit f38e348
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "llm-info",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "Information on LLM models, context window token limit, output token limit, pricing and more",
   "main": "dist/index.js",
   "type": "module",

diff --git a/src/modelInfo.ts b/src/modelInfo.ts
@@ -9,6 +9,8 @@ export type ModelInfo = {
   pricePerMillionInputTokens: number | null;
   pricePerMillionOutputTokens: number | null;
   tokenizerId: string | null;
+  notes?: string;
+  notesUrl?: string;
 };
 
 export const ModelInfoMap: Record<ModelLike, ModelInfo> = {
@@ -43,10 +45,12 @@ export const ModelInfoMap: Record<ModelLike, ModelInfo> = {
     name: 'Claude 3.5 Sonnet',
     provider: AI_PROVIDERS.ANTHROPIC,
     contextWindowTokenLimit: 200000,
-    outputTokenLimit: 4096,
+    outputTokenLimit: 8192,
     pricePerMillionInputTokens: 3,
     pricePerMillionOutputTokens: 15,
     tokenizerId: 'Xenova/claude-tokenizer',
+    notes: '8192 output tokens is in beta.',
+    notesUrl: 'https://docs.anthropic.com/en/docs/about-claude/models',
   },
   [NonModelEnum['chatgpt']]: {
     name: 'ChatGPT',