Update graph_config.py

graph_config.py

@@ -7,7 +7,7 @@ Modify these values to change graph behavior.
 
 Author: AI Lab Team
 Last Updated: 2025-10-10
-Version: 3.0 - Simplified 3-Tier System
+Version: 3.1 - Added legacy compatibility constants
 """
 
 import os
@@ -78,25 +78,22 @@ GPT4O_INPUT_COST_PER_1K_TOKENS = 0.005
 GPT4O_OUTPUT_COST_PER_1K_TOKENS = 0.015
 AVG_TOKENS_PER_CALL = 2000
 
-# Budget multipliers (removed - using tier-specific values)
-BUDGET_BUFFER_MULTIPLIER = 1.20  # Legacy - for compatibility
-MAX_COST_MULTIPLIER = 1.20  # Legacy - for compatibility
+# Budget multipliers (legacy compatibility)
+BUDGET_BUFFER_MULTIPLIER = 1.20
+MAX_COST_MULTIPLIER = 1.20
 
 
 # ============================================================================
 # LANGUAGE & CODING CONFIGURATION
 # ============================================================================
 
-# Default language
 DEFAULT_LANGUAGE = "python"
 
-# Supported languages
 SUPPORTED_LANGUAGES = [
     "python", "javascript", "typescript", "java", 
     "go", "rust", "cpp", "csharp", "php", "ruby"
 ]
 
-# Language preference keywords
 LANGUAGE_PREFERENCE_KEYWORDS = {
     "python": ["python", "py", "prefer python", "use python"],
     "javascript": ["javascript", "js", "node", "prefer javascript"],
@@ -106,14 +103,12 @@ LANGUAGE_PREFERENCE_KEYWORDS = {
     "rust": ["rust", "prefer rust"]
 }
 
-# No-code indicators (requests that should NOT generate code)
 NO_CODE_KEYWORDS = [
     'what is', 'what are', 'explain', 'describe', 
     'summarize', 'list', 'tell me about',
     'definition of', 'meaning of', 'overview of'
 ]
 
-# Code indicators (requests that SHOULD generate code)
 CODE_KEYWORDS = [
     'script', 'function', 'api', 'backend', 'frontend',
     'algorithm', 'program', 'code', 'implement',
@@ -125,7 +120,6 @@ CODE_KEYWORDS = [
 # RESEARCH MODE CONFIGURATION
 # ============================================================================
 
-# Research detection keywords
 RESEARCH_KEYWORDS = [
     'research', 'analyze', 'study', 'investigate', 'explore',
     'survey', 'review', 'compare', 'evaluate', 'assess',
@@ -133,21 +127,18 @@ RESEARCH_KEYWORDS = [
     'comprehensive', 'detailed analysis', 'in-depth'
 ]
 
-# Report indicators
 REPORT_KEYWORDS = [
     'report', 'summary', 'findings', 'document',
     'write up', 'present', 'deliverable', 'brief'
 ]
 
-# Citation configuration
 REQUIRE_CITATIONS_FOR_RESEARCH = True
 MIN_SOURCES_PER_RESEARCH = 3
 MAX_SOURCES_PER_RESEARCH = 10
-CITATION_FORMAT = "APA"  # Options: "APA", "MLA", "Chicago"
+CITATION_FORMAT = "APA"
 
-# Web search settings
 WEB_SEARCH_ENABLED = True
-WEB_SEARCH_API = "brave"  # Options: "brave", "google", "bing", "duckduckgo"
+WEB_SEARCH_API = "brave"
 WEB_SEARCH_MAX_RESULTS = 10
 WEB_SEARCH_TIMEOUT = 30  # seconds
 
@@ -156,13 +147,11 @@ WEB_SEARCH_TIMEOUT = 30  # seconds
 # ARTIFACT TYPE CONFIGURATION
 # ============================================================================
 
-# Known artifact types
 KNOWN_ARTIFACT_TYPES = {
     "notebook", "script", "repo", "word", 
     "excel", "pdf", "image", "data"
 }
 
-# Artifact type detection keywords
 ARTIFACT_TYPE_KEYWORDS = {
     'notebook': ['notebook', 'jupyter', 'ipynb', 'visualization', 'chart', 'graph'],
     'script': ['script', 'function', 'utility', 'tool'],
@@ -172,10 +161,7 @@ ARTIFACT_TYPE_KEYWORDS = {
     'pdf': ['pdf', 'printable']
 }
 
-# Default artifact type for coding
 DEFAULT_CODE_ARTIFACT = "script"
-
-# Default artifact type for research
 DEFAULT_RESEARCH_ARTIFACT = "word"
 
 
@@ -183,19 +169,16 @@ DEFAULT_RESEARCH_ARTIFACT = "word"
 # FILE & ARTIFACT MANAGEMENT
 # ============================================================================
 
-# Directories
 OUT_DIR = os.environ.get("OUT_DIR", "/tmp")
 EXPORTS_DIR = os.path.join(OUT_DIR, "exports")
 USER_ARTIFACTS_DIR = "outputs/user_artifacts"
 FEEDBACK_STORAGE_DIR = "outputs/feedback"
 
-# Artifact registry
 ARTIFACT_REGISTRY_FILE = "outputs/artifact_registry.json"
 ENSURE_ALL_ARTIFACTS_EXPORTED = True
 ARTIFACT_RETENTION_DAYS = 30
 MAX_ARTIFACT_SIZE_MB = 100
 
-# File naming
 UUID_LENGTH = 8
 FILENAME_SEPARATOR = "_"
 TIMESTAMP_FORMAT = "%Y%m%d_%H%M%S"
@@ -205,16 +188,12 @@ TIMESTAMP_FORMAT = "%Y%m%d_%H%M%S"
 # CONVERSATION CONTEXT MANAGEMENT
 # ============================================================================
 
-# Context window settings
-MAX_CONVERSATION_HISTORY = 10  # Last N exchanges
-CONTEXT_TOKEN_LIMIT = 4000  # Max tokens for context
+MAX_CONVERSATION_HISTORY = 10
+CONTEXT_TOKEN_LIMIT = 4000
 INCLUDE_ARTIFACTS_IN_CONTEXT = True
-
-# Context summarization
 AUTO_SUMMARIZE_LONG_CONVERSATIONS = True
 SUMMARIZE_AFTER_EXCHANGES = 5
 
-# Follow-up detection keywords
 FOLLOW_UP_KEYWORDS = [
     'also', 'additionally', 'now', 'then', 'next',
     'can you', 'please', 'what about', 'how about',
@@ -222,10 +201,8 @@ FOLLOW_UP_KEYWORDS = [
     'update', 'improve', 'enhance', 'refine'
 ]
 
-# Reference pronouns (indicate follow-up)
 REFERENCE_PRONOUNS = ['it', 'that', 'this', 'they', 'them', 'those', 'these']
 
-# Context indicator messages
 SHOW_CONTEXT_INDICATOR = True
 CONTEXT_INDICATOR_NEW = "📊 **New conversation**"
 CONTEXT_INDICATOR_FOLLOW_UP = "🔄 Follow-up detected"
@@ -272,7 +249,6 @@ EXECUTION_MODES = {
 # NODE STATUS MESSAGES
 # ============================================================================
 
-# Node status templates
 NODE_STATUS = {
     "memory": "Retrieving context...",
     "intent": "Clarifying objective...",
@@ -286,7 +262,6 @@ NODE_STATUS = {
     "archive": "Saving to memory..."
 }
 
-# Completion messages
 COMPLETION_SUCCESS = "✅ Task completed successfully"
 COMPLETION_WITH_WARNING = "⚠️ Task completed with limitations"
 COMPLETION_FAILED = "❌ Task could not be completed"
@@ -296,14 +271,12 @@ COMPLETION_FAILED = "❌ Task could not be completed"
 # ERROR HANDLING
 # ============================================================================
 
-# Error messages
 ERROR_NO_LLM = "LLM not available"
 ERROR_TIMEOUT = "Operation timed out"
 ERROR_BUDGET_EXCEEDED = "Budget limit exceeded"
 ERROR_PARSE_FAILED = "Failed to parse LLM response"
 ERROR_ARTIFACT_CREATION = "Failed to create artifact"
 
-# Retry settings
 MAX_RETRIES = 3
 RETRY_DELAY_SECONDS = 2
 EXPONENTIAL_BACKOFF = True
@@ -313,18 +286,14 @@ EXPONENTIAL_BACKOFF = True
 # LOGGING CONFIGURATION
 # ============================================================================
 
-# Log levels by tier
 TIER_LOG_LEVELS = {
     TIER_LITE: "WARNING",
     TIER_STANDARD: "INFO",
     TIER_FULL: "DEBUG"
 }
 
-# Log separators
 LOG_SEPARATOR = "=" * 60
 LOG_SEPARATOR_SHORT = "-" * 40
-
-# Performance logging
 LOG_PERFORMANCE_METRICS = True
 LOG_TOKEN_USAGE = True
 LOG_COST_TRACKING = True
@@ -340,78 +309,56 @@ def get_tier_config(tier: str) -> Dict[str, Any]:
 
 
 def calculate_cost_per_call(tokens: int = None) -> float:
-    """Calculate cost for an LLM call."""
     tokens = tokens or AVG_TOKENS_PER_CALL
     avg_cost_per_1k = (GPT4O_INPUT_COST_PER_1K_TOKENS + GPT4O_OUTPUT_COST_PER_1K_TOKENS) / 2.0
     return (tokens / 1000.0) * avg_cost_per_1k
 
 
 def is_tier_unlimited(tier: str) -> bool:
-    """Check if tier has unlimited budget."""
     config = get_tier_config(tier)
     return config["max_cost"] is None
 
 
 def get_max_rework_cycles(tier: str) -> int:
-    """Get max rework cycles for tier."""
     config = get_tier_config(tier)
     return config["qa_rework_cycles"]
 
 
 def should_include_monitoring(tier: str) -> bool:
-    """Check if tier includes performance monitoring."""
     config = get_tier_config(tier)
     return config["includes_monitoring"]
 
 
 def validate_tier(tier: str) -> str:
-    """Validate and normalize tier name."""
     tier_lower = tier.lower() if tier else DEFAULT_TIER
     if tier_lower not in TIER_CONFIGS:
         return DEFAULT_TIER
     return tier_lower
 
 
+# ============================================================================
+# LEGACY COMPATIBILITY CONSTANTS (to prevent AttributeError)
+# ============================================================================
+
+# Maintain backward compatibility with older modules (e.g., app_gradio.py)
+INITIAL_MAX_REWORK_CYCLES = get_max_rework_cycles(DEFAULT_TIER)
+# Explicitly re-export legacy cost multipliers for older modules
+LEGACY_BUDGET_BUFFER_MULTIPLIER = BUDGET_BUFFER_MULTIPLIER
+LEGACY_MAX_COST_MULTIPLIER = MAX_COST_MULTIPLIER
+
+
 # ============================================================================
 # EXPORTS
 # ============================================================================
 
 __all__ = [
-    # Tier system
-    'TIER_LITE',
-    'TIER_STANDARD',
-    'TIER_FULL',
-    'TIER_CONFIGS',
-    'DEFAULT_TIER',
-    
-    # Execution control
-    'MAX_EXECUTION_PATH_LENGTH',
-    'NODE_TIMEOUT_SECONDS',
-    
-    # Language & coding
-    'DEFAULT_LANGUAGE',
-    'SUPPORTED_LANGUAGES',
-    'NO_CODE_KEYWORDS',
-    'CODE_KEYWORDS',
-    
-    # Research mode
-    'RESEARCH_KEYWORDS',
-    'REQUIRE_CITATIONS_FOR_RESEARCH',
-    'WEB_SEARCH_ENABLED',
-    
-    # Artifacts
-    'KNOWN_ARTIFACT_TYPES',
-    'USER_ARTIFACTS_DIR',
-    
-    # Context management
-    'MAX_CONVERSATION_HISTORY',
-    'FOLLOW_UP_KEYWORDS',
-    
-    # Helper functions
-    'get_tier_config',
-    'calculate_cost_per_call',
-    'is_tier_unlimited',
-    'get_max_rework_cycles',
-    'should_include_monitoring',
-    'validate_tier',
-]
+    'TIER_LITE', 'TIER_STANDARD', 'TIER_FULL', 'TIER_CONFIGS', 'DEFAULT_TIER',
+    'MAX_EXECUTION_PATH_LENGTH', 'NODE_TIMEOUT_SECONDS',
+    'DEFAULT_LANGUAGE', 'SUPPORTED_LANGUAGES', 'NO_CODE_KEYWORDS', 'CODE_KEYWORDS',
+    'RESEARCH_KEYWORDS', 'REQUIRE_CITATIONS_FOR_RESEARCH', 'WEB_SEARCH_ENABLED',
+    'KNOWN_ARTIFACT_TYPES', 'USER_ARTIFACTS_DIR', 'MAX_CONVERSATION_HISTORY',
+    'FOLLOW_UP_KEYWORDS', 'get_tier_config', 'calculate_cost_per_call',
+    'is_tier_unlimited', 'get_max_rework_cycles', 'should_include_monitoring',
+    'validate_tier', 'INITIAL_MAX_REWORK_CYCLES',
+    'LEGACY_BUDGET_BUFFER_MULTIPLIER', 'LEGACY_MAX_COST_MULTIPLIER'
+]