Feat : agent router system + casual agent
This commit is contained in:
@@ -8,11 +8,11 @@ import configparser
|
||||
from sources.llm_provider import Provider
|
||||
from sources.interaction import Interaction
|
||||
from sources.code_agent import CoderAgent
|
||||
|
||||
from sources.casual_agent import CasualAgent
|
||||
|
||||
parser = argparse.ArgumentParser(description='Deepseek AI assistant')
|
||||
parser.add_argument('--speak', action='store_true',
|
||||
help='Make AI use text-to-speech')
|
||||
parser.add_argument('--no-speak', action='store_true',
|
||||
help='Make AI not use text-to-speech')
|
||||
args = parser.parse_args()
|
||||
|
||||
config = configparser.ConfigParser()
|
||||
@@ -31,12 +31,18 @@ def main():
|
||||
model=config["MAIN"]["provider_model"],
|
||||
server_address=config["MAIN"]["provider_server_address"])
|
||||
|
||||
agent = CoderAgent(model=config["MAIN"]["provider_model"],
|
||||
agents = [
|
||||
CoderAgent(model=config["MAIN"]["provider_model"],
|
||||
name=config["MAIN"]["agent_name"],
|
||||
prompt_path="prompts/coder_agent.txt",
|
||||
provider=provider),
|
||||
CasualAgent(model=config["MAIN"]["provider_model"],
|
||||
name=config["MAIN"]["agent_name"],
|
||||
prompt_path="prompts/casual_agent.txt",
|
||||
provider=provider)
|
||||
]
|
||||
|
||||
interaction = Interaction([agent], tts_enabled=config.getboolean('MAIN', 'speak'),
|
||||
interaction = Interaction(agents, tts_enabled=config.getboolean('MAIN', 'speak'),
|
||||
recover_last_session=config.getboolean('MAIN', 'recover_last_session'))
|
||||
while interaction.is_active:
|
||||
interaction.get_user()
|
||||
|
||||
Reference in New Issue
Block a user