o
    "iL                     @   s   d dl m Z  d dlmZ d dlmZmZ G dd deZG dd deZG dd	 d	eZG d
d deZ	G dd deZ
G dd deZG dd deZG dd deZG dd deZdS )    )datetime)Any)	BaseModelFieldc                   @   s.   e Zd ZU eed< eed< dZedB ed< dS )SpeakerTurnspeakertextN	timestamp)__name__
__module____qualname__str__annotations__r	   r    r   r   ?/home/aiteam/pcaa-dev/ai-call-quality-rag/app/models/schemas.pyr         
 r   c                   @   sP   e Zd ZU eed< eed< eed< eed< eedZee ed< ee	 ed< dS )	IngestCallRequest	tenant_idcall_idagent_idr	   default_factorytags
transcriptN)
r
   r   r   r   r   r   r   listr   r   r   r   r   r   r      s   
 r   c                   @   s&   e Zd ZU eed< eed< eed< dS )IngestCallResponser   r   chunks_ingestedN)r
   r   r   r   r   intr   r   r   r   r      s   
 r   c                   @   sX   e Zd ZU eed< eed< eed< eedZee ed< dZe	dB ed< dZ
eed	< dS )
QueryRequestr   user_idqueryr   r   Ntop_kTuse_keyword)r
   r   r   r   r   r   r   r   r!   r   r"   boolr   r   r   r   r      s   
 r   c                   @   sN   e Zd ZU eed< eed< eed< eed< eed< eed< eeef ed< dS )	ContextChunkchunk_idr   r   r   r   scoremetadataN)r
   r   r   r   r   floatdictr   r   r   r   r   r$   %   s   
 r$   c                   @   s"   e Zd ZU eed< ee ed< dS )QueryResponseanswercontextN)r
   r   r   r   r   r   r$   r   r   r   r   r*   /   s   
 r*   c                   @   s6   e Zd ZU eed< eed< eed< dZedB ed< dS )ScoreCallRequestr   r   r   N
query_hint)r
   r   r   r   r   r.   r   r   r   r   r-   4   s
   
 r-   c                   @   s.   e Zd ZU eed< eed< eeef ed< dS )ScoreCallResponser   r   score_payloadN)r
   r   r   r   r   r)   r   r   r   r   r   r/   ;   r   r/   c                   @   sB   e Zd ZU eed< eed< eed< eed< eed< ee ed< dS )AgentReportResponser   r   total_calls_scoredavg_quality_scorecompliance_raterecurring_issuesN)r
   r   r   r   r   r   r(   r   r   r   r   r   r1   A   s   
 r1   N)r   typingr   pydanticr   r   r   r   r   r   r$   r*   r-   r/   r1   r   r   r   r   <module>   s    		
