whale_core/agents.py

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 = "sk-z4C5B-O9hxSANw9drYaooVsp7SdXPw4Z29LUGkJEQhT3BlbkFJ0o-Gh0ALs1R5Zhiuchz-tn3bB_HizEuCx7s2xO_acA"

# 🧪 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..."

# ✅ 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]}"  # Use full text or chunk

        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