from __future__ import annotations

from fastapi import Request

from stirling.agents import (
    DocumentExtractorAgent,
    ExecutionPlanningAgent,
    FormAnalyserAgent,
    FormFillerAgent,
    OrchestratorAgent,
    PdfEditAgent,
    PdfQuestionAgent,
    UserSpecAgent,
)
from stirling.agents.ledger import MathAuditorAgent
from stirling.rag import RagService
from stirling.services import AppRuntime


def get_runtime(request: Request) -> AppRuntime:
    return request.app.state.runtime


def get_orchestrator_agent(request: Request) -> OrchestratorAgent:
    return request.app.state.orchestrator_agent


def get_pdf_edit_agent(request: Request) -> PdfEditAgent:
    return request.app.state.pdf_edit_agent


def get_pdf_question_agent(request: Request) -> PdfQuestionAgent:
    return request.app.state.pdf_question_agent


def get_user_spec_agent(request: Request) -> UserSpecAgent:
    return request.app.state.user_spec_agent


def get_execution_planning_agent(request: Request) -> ExecutionPlanningAgent:
    return request.app.state.execution_planning_agent


def get_rag_service(request: Request) -> RagService:
    return request.app.state.runtime.rag_service


def get_rag_embedding_model(request: Request) -> str:
    return request.app.state.runtime.settings.rag_embedding_model


def get_math_auditor_agent(request: Request) -> MathAuditorAgent:
    return request.app.state.math_auditor_agent


def get_form_analyser_agent(request: Request) -> FormAnalyserAgent:
    return request.app.state.form_analyser_agent


def get_form_filler_agent(request: Request) -> FormFillerAgent:
    return request.app.state.form_filler_agent


def get_document_extractor_agent(request: Request) -> DocumentExtractorAgent:
    return request.app.state.document_extractor_agent