Spaces:
Running
Running
| # config.py | |
| # This is the central "control panel" for MudabbirAI. | |
| # --- 1. MODEL CONFIGURATION --- | |
| MODELS = { | |
| "Gemini": { | |
| "default": "gemini-2.0-flash", # 1M Context | |
| "judge": "gemini-2.0-flash", | |
| "classifier": "gemini-2.0-flash", | |
| "manager": "gemini-2.0-flash" | |
| }, | |
| "Anthropic": { | |
| "default": "claude-3-5-haiku-20241022" # 200K Context | |
| }, | |
| "SambaNova": { | |
| "default": "Llama-4-Maverick-17B-128E-Instruct" # 512K Context | |
| }, | |
| "OpenAI": { | |
| "default": "gpt-4o-mini" # 128K Context | |
| }, | |
| # Updated: Using Qwen via Nebius | |
| "Nebius": { | |
| "default": "Qwen/Qwen3-235B-A22B-Thinking-2507" # 262K Context | |
| } | |
| } | |
| # --- 2. PRICING (USD per 1 Million Tokens) --- | |
| PRICING = { | |
| "Gemini": {"input": 0.10, "output": 0.40}, | |
| "Anthropic": {"input": 0.25, "output": 1.25}, | |
| "SambaNova": {"input": 0.00, "output": 0.00}, | |
| "OpenAI": {"input": 0.15, "output": 0.60}, | |
| # Updated Pricing for Qwen (based on your screenshot) | |
| "Nebius": {"input": 0.20, "output": 0.60} | |
| } | |
| # --- 3. CALIBRATION CONFIGURATION --- | |
| CALIBRATION_CONFIG = { | |
| "roles_to_test": { | |
| "Plant": "Culture_5", | |
| "Implementer": "Culture_Expert", | |
| "Monitor": "Culture_11" | |
| }, | |
| "role_metrics": { | |
| "Plant": "Novelty", | |
| "Implementer": "Usefulness_Feasibility", | |
| "Monitor": "Cultural_Appropriateness" | |
| } | |
| } | |
| # --- 4. PROMPT FILE PATHS --- | |
| PROMPT_FILES = { | |
| "evaluator": "prompts/evaluator_judge.txt", | |
| "classifier": "prompts/classifier.txt", | |
| "manager_homogeneous": "prompts/manager_homogeneous.txt", | |
| "manager_heterogeneous": "prompts/manager_heterogeneous.txt" | |
| } | |
| # --- 5. STATIC CONFIGS --- | |
| DEFAULT_PERSONA_KEY = "Culture_Expert" |