mirror of
https://github.com/tcsenpai/agenticSeek.git
synced 2025-06-06 11:05:26 +00:00
62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
|
|
from sources.text_to_speech import Speech
|
|
from sources.utility import pretty_print
|
|
from sources.router import AgentRouter
|
|
|
|
class Interaction:
|
|
def __init__(self, agents, tts_enabled: bool = False, recover_last_session: bool = False):
|
|
self.tts_enabled = tts_enabled
|
|
self.agents = agents
|
|
self.current_agent = None
|
|
self.router = AgentRouter(self.agents)
|
|
self.speech = Speech()
|
|
self.is_active = True
|
|
self.last_query = None
|
|
self.last_answer = None
|
|
if tts_enabled:
|
|
self.speech.speak("Hello Sir, we are online and ready. What can I do for you ?")
|
|
if recover_last_session:
|
|
self.recover_last_session()
|
|
|
|
def recover_last_session(self):
|
|
for agent in self.agents:
|
|
agent.memory.load_memory()
|
|
|
|
def is_active(self):
|
|
return self.is_active
|
|
|
|
def read_stdin(self) -> str:
|
|
buffer = ""
|
|
|
|
while buffer == "" or buffer.isascii() == False:
|
|
try:
|
|
buffer = input(f">>> ")
|
|
except EOFError:
|
|
return None
|
|
if buffer == "exit" or buffer == "goodbye":
|
|
return None
|
|
return buffer
|
|
|
|
def get_user(self):
|
|
query = self.read_stdin()
|
|
if query is None:
|
|
self.is_active = False
|
|
self.last_query = "Goodbye (exit requested by user, dont think, make answer very short)"
|
|
return None
|
|
self.last_query = query
|
|
return query
|
|
|
|
def think(self):
|
|
agent = self.router.select_agent(self.last_query)
|
|
if self.current_agent != agent:
|
|
self.current_agent = agent
|
|
# get history from previous agent
|
|
self.current_agent.memory.push('user', self.last_query)
|
|
self.last_answer, _ = agent.process(self.last_query, self.speech)
|
|
|
|
def show_answer(self):
|
|
self.current_agent.show_answer()
|
|
if self.tts_enabled:
|
|
self.speech.speak(self.last_answer)
|
|
|