aymnsk commited on
Commit
eeadd63
·
verified ·
1 Parent(s): 4df82dc

Update agents/hacker.py

Browse files
Files changed (1) hide show
  1. 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="Tech Savvy Code Wizard")
8
 
9
- def run(self, prompt, query_func):
 
 
10
  try:
11
- query = f"As a hacker, analyze or respond to this from a technical perspective: {prompt}"
12
- return query_func(query)
13
  except Exception as e:
14
- return f"[Hacker ERROR] {str(e)}"
 
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}"