Update settings.py

settings.py

@@ -1,71 +1,49 @@
-# settings.py
 import os
+from typing import Dict, Any
 
-# Snapshot settings
-SNAPSHOT_PATH = os.getenv("SNAPSHOT_PATH", "./snapshots")
-PERSIST_CONTENT = os.getenv("PERSIST_CONTENT", "false").lower() == "true"
+CANADA_RESIDENCY_REQUIRED = True
+CLOUD_REGION = os.getenv("CLOUD_REGION", "ca-central-1")
 
-# Healthcare settings
-HEALTHCARE_SETTINGS = {
-    "healthcare_keywords": [
-        "hospital", "clinic", "patient", "doctor", "nurse", "medical",
-        "healthcare", "diagnosis", "treatment", "pharmacy", "surgery",
-        "emergency", "icu", "ward", "bed", "capacity", "occupancy"
-    ],
-    "supported_file_types": [".csv", ".json", ".txt", ".xlsx", ".xls"],
-    "max_file_size_mb": 50
-}
+COHERE_API_KEY = os.getenv("COHERE_API_KEY", "")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
+ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", "")
 
-# Model settings
-MODEL_SETTINGS = {
-    "max_new_tokens": 2048,
-    "temperature": 0.3,
-    "top_p": 0.9,
-    "repetition_penalty": 1.15
-}
+COHERE_MODEL_PRIMARY = os.getenv("COHERE_MODEL_PRIMARY", "command-r")
+COHERE_EMBED_MODEL = os.getenv("COHERE_EMBED_MODEL", "embed-english-v3.0")
 
-# System prompts
-HEALTHCARE_SYSTEM_PROMPT = """
-You are a specialized healthcare analytics AI with expertise in:
-- Healthcare facility operations and capacity planning
-- Medical resource allocation and optimization
-- Health data analysis and trend identification
-- Healthcare policy and operational recommendations
+OPENAI_MODEL_FALLBACK = os.getenv("OPENAI_MODEL_FALLBACK", "gpt-4o-mini")
+ANTHROPIC_MODEL_FALLBACK = os.getenv("ANTHROPIC_MODEL_FALLBACK", "claude-3-5-sonnet-latest")
 
-When analyzing healthcare scenarios:
-1. Always structure your response with clear sections:
-   - Executive Summary
-   - Data Analysis (with subsections)
-   - Key Findings
-   - Operational Recommendations
-   - Future Integration Opportunities
-   - Provenance
+LOCAL_MODEL_ID = os.getenv("LOCAL_MODEL_ID", "meta-llama/Meta-Llama-3.1-8B-Instruct")
+LOCAL_ENABLE = os.getenv("LOCAL_ENABLE", "0") not in ("0", "false", "False")
 
-2. For data analysis:
-   - Include specific metrics and calculations
-   - Provide context and interpretation
-   - Identify trends and patterns
-   - Highlight significant findings
+USE_SCENARIO_ENGINE = True
 
-3. For recommendations:
-   - Prioritize by impact and feasibility
-   - Include implementation considerations
-   - Reference supporting data
+MODEL_SETTINGS: Dict[str, Any] = {
+    "temperature": float(os.getenv("TEMP", "0.2")),
+    "top_p": float(os.getenv("TOP_P", "0.9")),
+    "repetition_penalty": float(os.getenv("REP_PEN", "1.1")),
+    "max_new_tokens": int(os.getenv("MAX_NEW_TOKENS", "1500")),
+}
 
-4. Maintain strict privacy standards:
-   - Aggregate data appropriately
-   - Suppress small cohorts (<10)
-   - Never infer individual data
+HEALTHCARE_SETTINGS = {
+    "supported_file_types": [".csv", ".txt", ".md", ".pdf"],
+    "healthcare_keywords": [
+        "hospital", "clinic", "surgery", "wait time", "consult",
+        "beds", "icu", "zone", "health authority", "triage",
+    ],
+}
+
+GENERAL_CONVERSATION_PROMPT = "You are a helpful assistant. Be concise, accurate, and friendly."
 
-5. Use precise healthcare terminology and concepts.
-"""
+HEALTHCARE_SYSTEM_PROMPT = (
+    "You are a Canadian healthcare operations analysis copilot. "
+    "Read the entire scenario. Output only a structured JSON plan of tasks. "
+    "Each task must define dataset, filters, grouping, aggregations, joins, pivots, "
+    "and formatting exactly as the scenario requires. "
+    "Do not invent numbers; computations will be executed deterministically."
+)
 
-GENERAL_CONVERSATION_PROMPT = """
-You are a helpful AI assistant with broad knowledge. When responding:
-1. Be conversational and engaging
-2. Provide accurate, well-researched information
-3. Structure responses clearly with headings and bullet points
-4. Cite sources when possible
-5. Admit when you don't know something
-6. Maintain a professional yet approachable tone
-"""
+DATA_DIR = os.getenv("DATA_DIR", "./data")
+RAG_INDEX_DIR = os.getenv("RAG_INDEX_DIR", "./rag_index")
+SNAPSHOT_PATH = os.getenv("SNAPSHOT_PATH", "./snapshots")