gaia_final_assignment / multi_agent.py
Dkapsis's picture
use env variables as inputs
5f2bd4a
raw
history blame
527 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