klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 13:18:18 +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-20250219": 60000,
"claude-3-7-sonnet-latest": 60000, "claude-3-7-sonnet-latest": 60000,
"claude-3-5-haiku-20241022": 60000, "claude-3-5-haiku-20241022": 60000,
"claude-haiku-4-5-20251001": 60000,
"claude-sonnet-4-0": 60000, "claude-sonnet-4-0": 60000,
"claude-sonnet-4-20250514": 60000, "claude-sonnet-4-20250514": 60000,
"claude-opus-4-0": 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-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": {"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-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 pricing: https://docs.x.ai/docs/models
"grok-3": {"input": 3.0, "output": 15.0}, "grok-3": {"input": 3.0, "output": 15.0},
"grok-3-latest": {"input": 3.0, "output": 15.0}, "grok-3-latest": {"input": 3.0, "output": 15.0},

2
tests/conftest.py
View File

@@ -248,7 +248,7 @@ def chat_client_builder(search_config, user, index_content=True, require_auth=Fa
elif chat_provider == ChatModel.ModelType.GOOGLE: elif chat_provider == ChatModel.ModelType.GOOGLE:
online_chat_model = ChatModelFactory(name="gemini-2.5-flash", model_type="google") online_chat_model = ChatModelFactory(name="gemini-2.5-flash", model_type="google")
elif chat_provider == ChatModel.ModelType.ANTHROPIC: elif chat_provider == ChatModel.ModelType.ANTHROPIC:
online_chat_model = ChatModelFactory(name="claude-3-5-haiku-20241022", model_type="anthropic") online_chat_model = ChatModelFactory(name="claude-haiku-4-5-20251001", model_type="anthropic")
if online_chat_model: if online_chat_model:
online_chat_model.ai_model_api = AiModelApiFactory(api_key=get_chat_api_key(chat_provider)) online_chat_model.ai_model_api = AiModelApiFactory(api_key=get_chat_api_key(chat_provider))
UserConversationProcessorConfigFactory(user=user, setting=online_chat_model) UserConversationProcessorConfigFactory(user=user, setting=online_chat_model)