Update agents/hacker.py
Browse files- agents/hacker.py +8 -6
agents/hacker.py
CHANGED
@@ -1,14 +1,16 @@
|
|
1 |
# agents/hacker.py
|
2 |
|
3 |
-
from agents.base_agent import BaseAgent
|
|
|
4 |
|
5 |
class HackerAgent(BaseAgent):
|
6 |
def __init__(self):
|
7 |
-
super().__init__(name="💻 Hacker", role="
|
8 |
|
9 |
-
def
|
|
|
|
|
10 |
try:
|
11 |
-
|
12 |
-
return query_func(query)
|
13 |
except Exception as e:
|
14 |
-
return f"[
|
|
|
1 |
# agents/hacker.py
|
2 |
|
3 |
+
from agents.base_agent import BaseAgent, ACPMessage
|
4 |
+
from multi_inference import multi_query
|
5 |
|
6 |
class HackerAgent(BaseAgent):
|
7 |
def __init__(self):
|
8 |
+
super().__init__(name="💻 Hacker", role="Solves problems with code and logic.")
|
9 |
|
10 |
+
def generate(self, messages):
|
11 |
+
prompt = f"{self.role}:
|
12 |
+
" + "\n".join([f"{m.role}: {m.content}" for m in messages])
|
13 |
try:
|
14 |
+
return multi_query(prompt)
|
|
|
15 |
except Exception as e:
|
16 |
+
return f"[ERROR] {e}"
|