introduce constants module

hookla · Oct 22, 2023 · ab7e113 · ab7e113
1 parent aed859a
commit ab7e113
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 47 deletions.
diff --git a/constants/__init__.py b/constants/__init__.py
@@ -0,0 +1,2 @@
+from .default_sme import DEFAULT_SME_DICT
+from .strings import NO_COMMENT
diff --git a/constants/default_sme.py b/constants/default_sme.py
@@ -0,0 +1,42 @@
+DEFAULT_SME_DICT = (
+    {
+        "name": "CEO",
+        "expertise": "Corporate Strategy",
+        "concerns": ["Market Entry", "Competitive Positioning"],
+    },
+    {
+        "name": "CFO",
+        "expertise": "Financial Products",
+        "concerns": ["Rate Management", "Regulatory Compliance"],
+    },
+    {
+        "name": "COO",
+        "expertise": "Operational Efficiency",
+        "concerns": ["Scalability", "Cost Optimization"],
+    },
+    {
+        "name": "CMO",
+        "expertise": "Customer Acquisition",
+        "concerns": ["Target Market", "Onboarding Experience"],
+    },
+    {
+        "name": "CTO",
+        "expertise": "Technical Infrastructure",
+        "concerns": ["Data Security", "System Integration"],
+    },
+    {
+        "name": "CRO",
+        "expertise": "Risk Management",
+        "concerns": ["Fraud Detection", "Compliance"],
+    },
+    {
+        "name": "CCO",
+        "expertise": "Customer Experience",
+        "concerns": ["UX/UI Design", "Customer Support"],
+    },
+    {
+        "name": "CPO",
+        "expertise": "Product Management",
+        "concerns": ["Feature Rollout", "Customer Feedback"],
+    },
+)
diff --git a/constants.py → constants/strings.py b/constants.py → constants/strings.py
diff --git a/main.py b/main.py
@@ -8,57 +8,11 @@
 from agents.idea_refiner import IdeaRefiner
 from agents.sme import SME
 from clients import AIClientConfig, AIClientType, get_ai_client
-from constants import NO_COMMENT
+from constants import DEFAULT_SME_DICT, NO_COMMENT
 from utils.logging import configure_logging
 from utils.parse_config import parse_yaml_config
 from utils.print_with_wrap import print_with_wrap
 
-load_dotenv()
-
-# typical C-suite of executives
-DEFAULT_SME_DICT = (
-    {
-        "name": "CEO",
-        "expertise": "Corporate Strategy",
-        "concerns": ["Market Entry", "Competitive Positioning"],
-    },
-    {
-        "name": "CFO",
-        "expertise": "Financial Products",
-        "concerns": ["Rate Management", "Regulatory Compliance"],
-    },
-    {
-        "name": "COO",
-        "expertise": "Operational Efficiency",
-        "concerns": ["Scalability", "Cost Optimization"],
-    },
-    {
-        "name": "CMO",
-        "expertise": "Customer Acquisition",
-        "concerns": ["Target Market", "Onboarding Experience"],
-    },
-    {
-        "name": "CTO",
-        "expertise": "Technical Infrastructure",
-        "concerns": ["Data Security", "System Integration"],
-    },
-    {
-        "name": "CRO",
-        "expertise": "Risk Management",
-        "concerns": ["Fraud Detection", "Compliance"],
-    },
-    {
-        "name": "CCO",
-        "expertise": "Customer Experience",
-        "concerns": ["UX/UI Design", "Customer Support"],
-    },
-    {
-        "name": "CPO",
-        "expertise": "Product Management",
-        "concerns": ["Feature Rollout", "Customer Feedback"],
-    },
-)
-
 
 @click.command()
 @click.option(
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		from .default_sme import DEFAULT_SME_DICT
		from .strings import NO_COMMENT