youssefleb commited on
Commit
3bd9a83
·
verified ·
1 Parent(s): 1d67f6d

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +12 -5
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",