o "i @sUddlZddlmZmZmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZddlmZddlmZddlmZddlmZdd lmZeeZeZGd d d Zda edBe!d <d efddZ"defddZ#ej$de dee#fde d efddZ%ej$de dee#fde d efddZ&ej$dedee#fded efddZ'ej(de dee#fde)d e)d efd!d"Z*dS)#N) APIRouterDepends HTTPException)INGESTED_CHUNKSRETRIEVAL_QUERIES)AgentReportResponseIngestCallRequestIngestCallResponse QueryRequest QueryResponseScoreCallRequestScoreCallResponse ContextChunk)IngestionService)RetrievalService) LLMService)ScoringService) ReportServicec@s6eZdZUeed<eed<eed<eed<eed<dS)ServiceContainer ingestion retrievalllmscoringreportN) __name__ __module__ __qualname__r__annotations__rrrrrr;/home/aiteam/pcaa-dev/ai-call-quality-rag/app/api/routes.pyrs  r _containerccCs|adS)N)r )r!rrr set_container$sr"returncCstdurtdtS)Nz!Service container not initialized)r RuntimeErrorrrrr get_container)sr%z /ingest_call)response_modelpayloadcCsT|jj|j|j|j|j|jdd|jDd}tj |jd |t |j|j|dS)NcSsg|]}|qSr) model_dump).0trrr 7szingest_call..) tenant_idcall_idagent_id timestamptags transcriptr,)r,r-chunks_ingested) r ingest_callr,r-r.r/r0r1rlabelsincr )r'r!chunksrrrr4/sr4z/querycCsv|jj|j|j|j|j|jd}tj|jd d dd|D}|j d|jd|}t |dd|Dd S) N)r,queryr0top_k use_keywordr2z cSs g|] }d|jd|jqS)[z] )speakertextr)xrrrr+Hs zquery..z3Use this context to answer the user query. Query: z Context: c Ss0g|]}t|j|j|j|j|j|j|jdqS))chunk_idr-r.r<r=scoremetadata)rr@r-r.r<r=rArBr>rrrr+Os )answercontext)rretriever,r8r0r9r:rr5r6joinrrCr )r'r!r7 context_textrCrrrr8=s$ r8z /score_callc Cs`z |j|j|j|j}Wnty&}z tdtdt |dd}~wwt |j|j|dS)Nzscore_call failedi) status_codedetail)r,r- score_payload) r score_callr,r-r. Exceptionlogger exceptionrstrr )r'r!rJexcrrrrK^s rKz /agent_reportr,r.cCs|j||}tdi|S)Nr)r agent_reportr)r,r.r!rrrrrQisrQ)+loggingfastapirrrapp.core.observabilityrrapp.models.schemasrrr r r r r rapp.services.ingestion_servicerapp.services.retrieval_servicerapp.services.llm_servicerapp.services.scoring_servicerapp.services.report_servicer getLoggerrrMrouterrr rr"r%postr4r8rKgetrOrQrrrrs. (         $