Update whale_core/agents.py

whale_core/agents.py

@@ -2,17 +2,12 @@ import yaml
 import openai
 import os
 
-# ✅ Set your OpenAI API key — use one of the two options below:
-
-# 🔒 Option 1: Load from environment variable (recommended)
-openai.api_key = os.environ["OPENAI_API_KEY"]
+# ✅ Set your OpenAI API key from environment variable (Hugging Face Secrets)
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # 🧪 Optional: fail early if key not set
 if not openai.api_key:
-    raise ValueError("❌ No OpenAI API key found. Set the environment variable OPENAI_API_KEY.")
-
-# 🔓 Option 2 (for development only): Hardcode the key
-# openai.api_key = "sk-...your full key..."
+    raise ValueError("❌ No OpenAI API key found. Set the environment variable OPENAI_API_KEY in your Hugging Face Space settings under 'Secrets'.")
 
 # ✅ Load agent configurations from YAML
 def load_agents(path="agents/config.yaml"):
@@ -27,7 +22,7 @@ def run_agents_on_text(agent_cfgs, doc):
         name = agent.get("name", "UnnamedAgent")
         system_msg = agent.get("persona", "You are a helpful assistant.")
         instructions = agent.get("instructions", "Please analyze this document.")
-        user_prompt = f"{instructions}\n\nDocument:\n{doc[:4000]}"  # Use full text or chunk
+        user_prompt = f"{instructions}\n\nDocument:\n{doc[:4000]}"  # Truncate if needed
 
         try:
             completion = openai.ChatCompletion.create(