gaia_final_assignment / multi_agent.py
Dkapsis's picture
audio agent
a9182c5
raw
history blame
984 Bytes
import os
import pandas as pd
import requests
from smolagents import OpenAIServerModel, CodeAgent, InferenceClientModel
from smolagents.tools import tool
import markdownify
import prompts
import agents
def orchestrate(message, file_path):
final_prompt = prompts.get_manager_prompt(message, file_path)
initial_answer = agents.create_manager_agent(message).run(message)
final_answer = agents.create_final_answer_agent(message).run(prompts.get_final_answer_prompt(message, initial_answer))
return final_answer
# def run_manager_workflow(message, file_path=None):
# final_prompt = prompts.get_manager_prompt(message, file_path)
# initial_answer = agents.create_simple_web_search_agent(message).run(message)
# final_answer = agents.create_final_answer_agent(message).run(prompts.get_final_answer_prompt(message, initial_answer))
# return final_answer
# final_answer = run_manager_workflow(message)
# return final_answer