gaia_final_assignment / multi_agent.py
Dkapsis's picture
minor promt fix
11ae913
raw
history blame contribute delete
526 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