Refactor load_config to support multiple environment variable names for API key and update default model

agents.py

@@ -7,10 +7,15 @@ from together import Together
 
 # Load configuration from config.json or environment variables
 def load_config():
-    # First try to get API key from environment variable (for Hugging Face deployment)
+    # Try to get API key from environment variables (for Hugging Face deployment)
+    # Check both possible environment variable names
     together_api_key = os.environ.get("TOGETHER_API_KEY", "")
     
-    # If not found in environment, try to load from config.json (for local development)
+    # If not found with TOGETHER_API_KEY, try together_ai_token (the name used in your HF secret)
+    if not together_api_key:
+        together_api_key = os.environ.get("together_ai_token", "")
+    
+    # If still not found in environment, try to load from config.json (for local development)
     if not together_api_key:
         try:
             config_path = os.path.join(os.path.dirname(__file__), "config.json")
@@ -23,10 +28,13 @@ def load_config():
         except Exception as e:
             print(f"Error loading config.json: {str(e)}")
     
+    # Print debug information
+    print(f"API key found: {'Yes' if together_api_key else 'No'}")
+    
     # Return a config dictionary with the API key from environment variable
     return {
         "together_ai_token": together_api_key,
-        "model": "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8"  # Default model
+        "model": "meta-llama/Llama-3.3-70B-Instruct-Turbo"  # Using the model from your config.json
     }
 
 # Get API key and model from config