youssefleb commited on
Commit
f499a2f
·
verified ·
1 Parent(s): 4052248

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +44 -0
config.py ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # config.py
2
+ # This is the central "control panel" for MudabbirAI.
3
+
4
+ # --- 1. MODEL CONFIGURATION ---
5
+ MODELS = {
6
+ "Gemini": {
7
+ "default": "gemini-1.5-flash-latest",
8
+ "judge": "gemini-1.5-flash-latest",
9
+ "classifier": "gemini-1.5-flash-latest",
10
+ "manager": "gemini-1.5-flash-latest"
11
+ },
12
+ "Anthropic": {
13
+ "default": "claude-3-opus-20240229"
14
+ },
15
+ "SambaNova": {
16
+ "default": "Meta-Llama-3.1-8B-Instruct"
17
+ }
18
+ }
19
+
20
+ # --- 2. CALIBRATION CONFIGURATION ---
21
+ CALIBRATION_CONFIG = {
22
+ "roles_to_test": {
23
+ "Plant": "Culture_5",
24
+ "Implementer": "Culture_Expert",
25
+ "Monitor": "Culture_11"
26
+ },
27
+ "role_metrics": {
28
+ "Plant": "Novelty",
29
+ "Implementer": "Usefulness_Feasibility",
30
+ "Monitor": "Cultural_Appropriateness"
31
+ }
32
+ }
33
+
34
+ # --- 3. PROMPT FILE PATHS ---
35
+ # This new section points to our external prompt files.
36
+ PROMPT_FILES = {
37
+ "evaluator": "prompts/evaluator_judge.txt",
38
+ "classifier": "prompts/classifier.txt",
39
+ "manager_homogeneous": "prompts/manager_homogeneous.txt",
40
+ "manager_heterogeneous": "prompts/manager_heterogeneous.txt"
41
+ }
42
+
43
+ # --- 4. STATIC CONFIGS ---
44
+ DEFAULT_PERSONA_KEY = "Culture_Expert"