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

Update agents/historian.py

Browse files
Files changed (1) hide show
  1. agents/historian.py +9 -6
agents/historian.py CHANGED
@@ -1,14 +1,17 @@
1
  # agents/historian.py
2
 
3
- from agents.base_agent import BaseAgent
 
4
 
5
  class HistorianAgent(BaseAgent):
6
  def __init__(self):
7
- super().__init__(name="👨‍🏫 Historian", role="Expert in Historical Knowledge")
8
 
9
- def run(self, prompt, query_func):
 
 
10
  try:
11
- query = f"As a historian, explain this with historical context: {prompt}"
12
- return query_func(query)
13
  except Exception as e:
14
- return f"[Historian ERROR] {str(e)}"
 
 
1
  # agents/historian.py
2
 
3
+ from agents.base_agent import BaseAgent, ACPMessage
4
+ from multi_inference import multi_query
5
 
6
  class HistorianAgent(BaseAgent):
7
  def __init__(self):
8
+ super().__init__(name="👨‍🏫 Historian", role="Provides historical context and analysis.")
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}"
17
+