whale_core/agents.py

import yaml
import openai
import os

# ✅ 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 in your Hugging Face Space settings under 'Secrets'.")

# ✅ Load agent configurations from YAML
def load_agents(path="agents/config.yaml"):
    with open(path, "r") as f:
        return yaml.safe_load(f)

# ✅ Run each agent on the input document
def run_agents_on_text(agent_cfgs, doc):
    responses = {}

    for agent in agent_cfgs.get("agents", []):
        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]}"  # Truncate if needed

        try:
            completion = openai.ChatCompletion.create(
                model="gpt-4",
                messages=[
                    {"role": "system", "content": system_msg},
                    {"role": "user", "content": user_prompt}
                ],
                max_tokens=300,
                temperature=0.7
            )
            responses[name] = completion.choices[0].message["content"].strip()

        except Exception as e:
            responses[name] = f"⚠️ Error calling OpenAI: {str(e)}"

    return responses