klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-05 05:39:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Track cost of claude haiku 4.5 model

Browse Source
This commit is contained in:
Debanjum
2025-11-09 09:35:13 -08:00
parent a809de8970
commit 88a1fc75cc
3 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/khoj/processor/conversation/utils.py
View File

@@ -84,6 +84,7 @@ model_to_prompt_size = {
"claude-3-7-sonnet-20250219": 60000,
"claude-3-7-sonnet-latest": 60000,
"claude-3-5-haiku-20241022": 60000,
"claude-haiku-4-5-20251001": 60000,
"claude-sonnet-4-0": 60000,
"claude-sonnet-4-20250514": 60000,
"claude-opus-4-0": 60000,

2
src/khoj/utils/constants.py
View File

@@ -69,6 +69,8 @@ model_to_cost: Dict[str, Dict[str, float]] = {
"claude-opus-4@20250514": {"input": 15.0, "output": 75.0, "cache_read": 1.50, "cache_write": 18.75},
"claude-sonnet-4-5": {"input": 3.0, "output": 15.0, "cache_read": 0.3, "cache_write": 3.75},
"claude-sonnet-4-5-20250929": {"input": 3.0, "output": 15.0, "cache_read": 0.3, "cache_write": 3.75},
"claude-haiku-4-5": {"input": 1.0, "output": 5.0, "cache_read": 0.08, "cache_write": 1.0},
"claude-haiku-4-5-20251001": {"input": 1.0, "output": 5.0, "cache_read": 0.08, "cache_write": 1.0},
# Grok pricing: https://docs.x.ai/docs/models
"grok-3": {"input": 3.0, "output": 15.0},
"grok-3-latest": {"input": 3.0, "output": 15.0},