File size: 1,129 Bytes
0835b53
 
 
 
 
 
 
 
 
dddee86
 
0835b53
6208c46
0835b53
 
6208c46
dddee86
3688a56
 
dddee86
6208c46
0835b53
9086bc8
0835b53
 
 
 
 
 
3688a56
 
 
 
 
 
f24f363
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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