Update config.py

config.py

@@ -1,6 +1,5 @@
 # config.py
 # This is the central "control panel" for MudabbirAI.
-# We are using the most cost-effective models with large context windows.
 
 # --- 1. MODEL CONFIGURATION ---
 MODELS = {
@@ -16,15 +15,20 @@ MODELS = {
         "default": "claude-3-5-haiku-20241022" 
     },
     "SambaNova": {
-        # --- THIS IS THE FIX ---
-        # Using the Maverick model. It is available on your dashboard
-        # and has a 512K context window, meeting our requirement.
+        # Using the Maverick model. 
         "default": "Llama-4-Maverick-17B-128E-Instruct"
-        # --- END OF FIX ---
     }
 }
 
-# --- 2. CALIBRATION CONFIGURATION ---
+# --- 2. PRICING (USD per 1 Million Tokens) ---
+# Based on standard public pricing (approximate)
+PRICING = {
+    "Gemini":     {"input": 0.10, "output": 0.40},  # Flash tier
+    "Anthropic":  {"input": 0.25, "output": 1.25},  # Haiku tier
+    "SambaNova":  {"input": 0.00, "output": 0.00}   # Often free in hackathons/preview
+}
+
+# --- 3. CALIBRATION CONFIGURATION ---
 CALIBRATION_CONFIG = {
     "roles_to_test": {
         "Plant": "Culture_5",
@@ -38,13 +42,13 @@ CALIBRATION_CONFIG = {
     }
 }
 
-# --- 3. PROMPT FILE PATHS ---
+# --- 4. PROMPT FILE PATHS ---
 PROMPT_FILES = {
-    "evaluator": "prompts/evaluator_judge.txt",
+    "evaluator": "evaluator_judge.txt",
     "classifier": "prompts/classifier.txt",
     "manager_homogeneous": "prompts/manager_homogeneous.txt",
     "manager_heterogeneous": "prompts/manager_heterogeneous.txt"
 }
 
-# --- 4. STATIC CONFIGS ---
+# --- 5. STATIC CONFIGS ---
 DEFAULT_PERSONA_KEY = "Culture_Expert"