whalecore/agents.py

import yaml

class Agent:
    def __init__(self, name, persona, instructions):
        self.name = name
        self.persona = persona
        self.instructions = instructions

    def chat(self, message):
        # Placeholder logic — replace with real LLM call later
        return f"🧠 {self.name} says:\n{self.instructions}\n\n{self.persona}\n\nYou said: {message[:260]}..."

def load_agents(config_path="config.yaml"):
    with open(config_path, 'r') as f:
        config = yaml.safe_load(f)

    assert isinstance(config, dict), "YAML must contain a top-level 'agents:' key"
    assert 'agents' in config, "Missing 'agents' key in YAML file"

    print("🧠 YAML loaded successfully:", config)

    agents = []
    for agent_conf in config['agents']:
        agent = Agent(
            name=agent_conf['name'],
            persona=agent_conf['persona'],
            instructions=agent_conf['instructions']
        )
        agents.append(agent)
    return agents

def run_agents_on_text(agent_list, text):
    results = {}
    for agent in agent_list:
        results[agent.name] = agent.chat(text)
    return results