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

Track gpt-4.1 model costs. Set prompt size of new gemini, openai models

Browse Source
This commit is contained in:
Debanjum
2025-04-23 17:31:29 +05:30
parent 87262d15bb
commit dd46bcabc2
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -53,10 +53,17 @@ model_to_prompt_size = {
# OpenAI Models
"gpt-4o": 60000,
"gpt-4o-mini": 120000,
"gpt-4.1": 60000,
"gpt-4.1-mini": 120000,
"gpt-4.1-nano": 120000,
"o1": 20000,
"o3": 30000,
"o1-mini": 60000,
"o3-mini": 60000,
"o4-mini": 60000,
# Google Models
"gemini-2.5-flash-preview-04-17": 120000,
"gemini-2.5-pro-preview-03-25": 60000,
"gemini-2.0-flash": 120000,
"gemini-2.0-flash-lite": 120000,
"gemini-1.5-flash": 120000,

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

@@ -38,6 +38,9 @@ model_to_cost: Dict[str, Dict[str, float]] = {
# OpenAI Pricing: https://openai.com/api/pricing/
"gpt-4o": {"input": 2.50, "output": 10.00},
"gpt-4o-mini": {"input": 0.15, "output": 0.60},
"gpt-4.1": {"input": 2.00, "output": 8.00},
"gpt-4.1-mini": {"input": 0.40, "output": 1.60},
"gpt-4.1-nano": {"input": 0.10, "output": 0.40},
"o1": {"input": 15.0, "output": 60.00},
"o3": {"input": 10.0, "output": 40.00},
"o1-mini": {"input": 3.0, "output": 12.0},