BI Analytics Platform Overview

class AnalyticsOrchestrator:
    MAX_AGENT_VISITS = 10  # Prevent infinite loops
 
    def __init__(self, sql_generator, query_executor, llm_client, schema_provider, session_store):
        self._router = RouterAgent(llm_client=llm_client)
        self._sql_agent = SQLAgent(sql_generator=sql_generator, query_executor=query_executor)
        self._analysis_agent = AnalysisAgent(llm_client=llm_client)
        self._viz_agent = VisualizationAgent()
        self._docs_agent = DocumentationAgent(schema_provider=schema_provider)
 
        self._agents = {
            "router": self._router.classify,
            "sql": self._sql_agent.generate,
            "analysis": self._analysis_agent.analyze,
            "visualization": self._viz_agent.generate,
            "docs": self._docs_agent.respond,
        }

User Message -> Router Agent (classify intent)
    |
    +-- QUERY_DATA -> SQL Agent -> Visualization Agent
    |
    +-- ANALYZE -> SQL Agent -> Analysis Agent -> Visualization Agent
    |
    +-- VISUALIZE -> Visualization Agent
    |
    +-- HELP/SCHEMA -> Documentation Agent

async def create_session(self, tenant_id, user_id=None, session_id=None):
    session = ConversationSession(
        session_id=session_id or str(uuid.uuid4()),
        tenant_id=tenant_id,
        user_id=user_id,
        state=AgentState(session_id=session_id, tenant_id=tenant_id),
    )
    await store.save(session.to_session_data())
    return session
 
async def process_message(self, session_id, message):
    session = await self.get_session(session_id)
    state = session.state
    state.current_question = message
    state.next_agent = "router"
    state = await self._run_workflow(state)
    return {
        "success": len(state.errors) == 0,
        "response": state.build_response(),
        "sql": state.sql_context.sql,
        "visualization": state.visualization_context.chart_spec,
        "data": state.sql_context.result,
        "insights": state.analysis_context.insights,
    }