Spaces:
Sleeping
Sleeping
Update agents/historian.py
Browse files- 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="
|
8 |
|
9 |
-
def
|
|
|
|
|
10 |
try:
|
11 |
-
|
12 |
-
return query_func(query)
|
13 |
except Exception as e:
|
14 |
-
return f"[
|
|
|
|
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 |
+
|