Spaces:
Running
Running
Update config.py
Browse files
config.py
CHANGED
|
@@ -4,16 +4,22 @@
|
|
| 4 |
# --- 1. MODEL CONFIGURATION ---
|
| 5 |
MODELS = {
|
| 6 |
"Gemini": {
|
| 7 |
-
"default": "gemini-2.0-flash",
|
| 8 |
"judge": "gemini-2.0-flash",
|
| 9 |
"classifier": "gemini-2.0-flash",
|
| 10 |
"manager": "gemini-2.0-flash"
|
| 11 |
},
|
| 12 |
"Anthropic": {
|
| 13 |
-
"default": "claude-3-5-haiku-20241022"
|
| 14 |
},
|
| 15 |
"SambaNova": {
|
| 16 |
-
"default": "Llama-4-Maverick-17B-128E-Instruct"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
}
|
| 18 |
}
|
| 19 |
|
|
@@ -21,7 +27,9 @@ MODELS = {
|
|
| 21 |
PRICING = {
|
| 22 |
"Gemini": {"input": 0.10, "output": 0.40},
|
| 23 |
"Anthropic": {"input": 0.25, "output": 1.25},
|
| 24 |
-
"SambaNova": {"input": 0.00, "output": 0.00}
|
|
|
|
|
|
|
| 25 |
}
|
| 26 |
|
| 27 |
# --- 3. CALIBRATION CONFIGURATION ---
|
|
@@ -39,7 +47,6 @@ CALIBRATION_CONFIG = {
|
|
| 39 |
}
|
| 40 |
|
| 41 |
# --- 4. PROMPT FILE PATHS ---
|
| 42 |
-
# Updated: Explicitly point to the 'prompts/' directory for the evaluator
|
| 43 |
PROMPT_FILES = {
|
| 44 |
"evaluator": "prompts/evaluator_judge.txt",
|
| 45 |
"classifier": "prompts/classifier.txt",
|
|
|
|
| 4 |
# --- 1. MODEL CONFIGURATION ---
|
| 5 |
MODELS = {
|
| 6 |
"Gemini": {
|
| 7 |
+
"default": "gemini-2.0-flash", # 1M Context
|
| 8 |
"judge": "gemini-2.0-flash",
|
| 9 |
"classifier": "gemini-2.0-flash",
|
| 10 |
"manager": "gemini-2.0-flash"
|
| 11 |
},
|
| 12 |
"Anthropic": {
|
| 13 |
+
"default": "claude-3-5-haiku-20241022" # 200K Context
|
| 14 |
},
|
| 15 |
"SambaNova": {
|
| 16 |
+
"default": "Llama-4-Maverick-17B-128E-Instruct" # 512K Context (Hackathon Special)
|
| 17 |
+
},
|
| 18 |
+
"OpenAI": {
|
| 19 |
+
"default": "gpt-4o-mini" # 128K Context (Very cost effective)
|
| 20 |
+
},
|
| 21 |
+
"Nebius": {
|
| 22 |
+
"default": "meta-llama/Meta-Llama-3.1-70B-Instruct" # 128K Context (Open Source Powerhouse)
|
| 23 |
}
|
| 24 |
}
|
| 25 |
|
|
|
|
| 27 |
PRICING = {
|
| 28 |
"Gemini": {"input": 0.10, "output": 0.40},
|
| 29 |
"Anthropic": {"input": 0.25, "output": 1.25},
|
| 30 |
+
"SambaNova": {"input": 0.00, "output": 0.00}, # Assuming free tier
|
| 31 |
+
"OpenAI": {"input": 0.15, "output": 0.60}, # gpt-4o-mini pricing
|
| 32 |
+
"Nebius": {"input": 0.20, "output": 0.80} # Approximate Llama 70B hosting cost
|
| 33 |
}
|
| 34 |
|
| 35 |
# --- 3. CALIBRATION CONFIGURATION ---
|
|
|
|
| 47 |
}
|
| 48 |
|
| 49 |
# --- 4. PROMPT FILE PATHS ---
|
|
|
|
| 50 |
PROMPT_FILES = {
|
| 51 |
"evaluator": "prompts/evaluator_judge.txt",
|
| 52 |
"classifier": "prompts/classifier.txt",
|