File size: 514 Bytes
d496e90
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# agents/lawyer.py

from agents.base_agent import BaseAgent, ACPMessage
from multi_inference import multi_query

class LawyerAgent(BaseAgent):
    def __init__(self):
        super().__init__(name="⚖️ Lawyer", role="Analyzes from a legal and ethical perspective.")

    def generate(self, messages):
        prompt = f"{self.role}:\n" + "\n".join([f"{m.role}: {m.content}" for m in messages])
        try:
            return multi_query(prompt)
        except Exception as e:
            return f"[ERROR] {e}"