o
    6Úfhð  ã                   @   sh   d dl mZ d dlmZmZ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
dS )é    )Ú	BaseModel)ÚListÚOptionalÚDictÚAnyc                   @   sr   e Zd ZU dZeed< eed< dZee ed< dZee ed< dZ	ee ed< dZ
ee ed< dZee ed	< dS )
ÚSearchResultDataz+Search result with separate content and URLÚcontentÚurlNÚtitleÚsnippetÚsourceÚdomainÚrelevance_score)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚstrÚ__annotations__r
   r   r   r   r   r   Úfloat© r   r   ú]/var/www/eduai.edurigo.com/doc_train/edurigo_ai/my_career/testing/advanced_chatbot/schemas.pyr      s   
 r   c                   @   sr   e Zd ZU dZeed< g Zeee  ed< g Z	eee
  ed< dZeeeef  ed< dZeeeef  ed< dS )ÚAdvancedChatbotResponsez5Updated response model with structured search resultsÚresponseÚsearch_resultsÚsearch_results_dataNÚcontextÚmetadata)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      s   
 r   c                   @   s.   e Zd ZU dZdZeed< eeef ed< dS )ÚEnhancedAdvancedChatbotResponsez0Enhanced response with structured search resultsÚcompleteÚtypeÚdataN)	r   r   r   r   r    r   r   r   r   r   r   r   r   r      s   
 r   c                   @   sj   e Zd ZU dZeed< g Zee ed< g Zee	 ed< dZ
eeeef  ed< dZeeeef  ed< dS )ÚResponseDatazResponse data structurer   r   r   Nr   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r"      s   
 r"   N)Úpydanticr   Útypingr   r   r   r   r   r   r   r"   r   r   r   r   Ú<module>   s   
