Data_Analyst_Assistant / src /agents /orchestrator_agent.py
Daksh Chaudhary
Update src/agents/orchestrator_agent.py
1823c1d verified
import sys
import logging
from llama_index.core.agent import ReActAgent
from .agents.models import get_base_agent_model
#Logging
logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
class OrchestratorAgent:
def __init__(self):
self.tools = []
self.llm = get_base_agent_model()
self.agent = ReActAgent.from_tools(llm=self.llm, tools=self.tools, context=self.system_prompt, verbose=True)
async def __call__(self, question: str) -> str:
response_object = self.agent.chat(question)
full_response_text = str(response_object)
return full_response_text