
    j                        d dl mZm Z  d dlmZ  G d de          Z G d de          Z G d de          Z G d	 d
e          Z G d de          Z G d de          Z	 G d de	          Z
 G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d d e          Z G d! d"e          Z G d# d$e          Z G d% d&e          Z G d' d(e          Z G d) d*e          Z G d+ d,e          Z G d- d.e          Z G d/ d0e          Z G d1 d2e          Z G d3 d4e          Zd5S )6    )datedatetime)	BaseModelc                       e Zd ZU dZedz  ed<   eed<   dZedz  ed<   dZedz  ed<   dZ	edz  ed<   dZ
edz  ed<   dZeed	<   dS )
CategoryCreateNknowledge_base_idnamedescription	parent_idiconcolorr   
sort_order)__name__
__module____qualname__r   int__annotations__strr
   r   r   r   r        F/lsinfo/ai/hellotax_ai/base_platform/app/schemas/knowledge_document.pyr   r      s         $(sTz(((
III"Kt""" IsTz   D#*E3:Jr   r   c                       e Zd ZU eed<   edz  ed<   eed<   edz  ed<   edz  ed<   edz  ed<   edz  ed<   d	Zeed
<   d	Zeed<   d	Zeed<   e	ed<   e	ed<    G d d          Z
dS )CategoryResponseidNr   r	   r
   r   r   r   r   r   	doc_countdocument_count
created_at
updated_atc                       e Zd ZdZdS )CategoryResponse.ConfigTNr   r   r   from_attributesr   r   r   Configr               r   r#   )r   r   r   r   r   r   r   r   r   r   r#   r   r   r   r   r      s         GGGTz!!!
IIItTz
*:JIsNC         r   r   c                       e Zd ZU eed<   eed<   eed<   edz  ed<   eed<   dZeed<   edz  ed	<   edz  ed
<   eed<   eed<    G d d          ZdS )TagCategoryResponser   r	   codeNr
   part_numberr   r   r   r   r   r   c                       e Zd ZdZdS )TagCategoryResponse.ConfigTNr!   r   r   r   r#   r*   .   r$   r   r#   )	r   r   r   r   r   r   r   r   r#   r   r   r   r&   r&   "   s         GGG
III
IIItJ
*:         r   r&   c                       e Zd ZU eed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZ	edz  ed<   d	Z
eed
<   dZeed<   dZedz  ed<   dZedz  ed<   dZeed<   dZee         dz  ed<   dZeed<   d	Zeed<   dS )	TagCreater	   Nr   r
   u   🏷️r   z#6366f1r   r   r   enabledstatustag_category_idtag_codeFis_predefinedkeywords      ?search_weightsearch_priority)r   r   r   r   r   r   r   r
   r   r   r   r.   r/   r0   r1   boolr2   listr4   floatr5   r   r   r   r,   r,   2   s         
III IsTz   "Kt""" D#*   !E3:!!!JFC"&OS4Z&&&HcDjM4!%Hd3i$%%%M5OSr   r,   c                       e Zd ZU dZedz  ed<   dZedz  ed<   dZedz  ed<   dZ	edz  ed<   dZ
edz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed	<   dZee         dz  ed
<   dZedz  ed<   dZedz  ed<   dS )	TagUpdateNr	   r   r
   r   r   r   r.   r/   r2   r4   r5   )r   r   r   r	   r   r   r   r   r
   r   r   r   r.   r/   r2   r7   r4   r8   r5   r   r   r   r:   r:   B   s         D#* IsTz   "Kt"""D#*E3:!Jd
!!!FC$J"&OS4Z&&&!%Hd3i$%%%"&M54<&&&"&OS4Z&&&&&r   r:   c                   8   e Zd ZU eed<   eed<   edz  ed<   edz  ed<   edz  ed<   edz  ed<   dZeed	<   eed
<   dZeed<   dZedz  ed<   dZ	edz  ed<   dZ
eed<   dZeed<   dZee         dz  ed<   dZeed<   dZeed<   eed<   eed<    G d d          ZdS )TagResponser   r	   Nr   r
   r   r   r   r   r.   r   r/   r0   Fr1   	is_systemr2   r3   r4   r5   r   r   c                       e Zd ZdZdS )TagResponse.ConfigTNr!   r   r   r   r#   r?   d   r$   r   r#   )r   r   r   r   r   r   r   r   r/   r0   r1   r6   r=   r2   r7   r4   r8   r5   r   r#   r   r   r   r<   r<   P   sR        GGG
IIITzt
*:JKKKNC"&OS4Z&&&HcDjM4It!%Hd3i$%%%M5OS         r   r<   c                   D    e Zd ZU g Zed          ed<    G d d          ZdS )TagTreeNodechildrenc                       e Zd ZdZdS )TagTreeNode.ConfigTNr!   r   r   r   r#   rD   k   r$   r   r#   N)r   r   r   rB   r7   r   r#   r   r   r   rA   rA   h   sS         $&Hd=!&&&         r   rA   c                   T    e Zd ZU eed<   eed<   ee         ed<    G d d          ZdS )TagTreeGroupedr(   	part_name
categoriesc                       e Zd ZdZdS )TagTreeGrouped.ConfigTNr!   r   r   r   r#   rJ   t   r$   r   r#   N)	r   r   r   r   r   r   r7   dictr#   r   r   r   rF   rF   o   sc         NNNT
         r   rF   c                   T    e Zd ZU eed<   eed<   edz  ed<   eed<   ee         ed<   dS )AutoTagSuggestiontag_idtag_nameNr0   
confidencematched_keywords)r   r   r   r   r   r   r8   r7   r   r   r   rM   rM   x   sP         KKKMMMDj3ir   rM   c                   (    e Zd ZU eed<   dZeed<   dS )AutoTagRequestdocument_id333333?confidence_thresholdN)r   r   r   r   r   rV   r8   r   r   r   rS   rS      s/         "%%%%%%%r   rS   c                   :    e Zd ZU eed<   ee         ed<   eed<   dS )AutoTagResponserT   suggestionstotal_suggestionsN)r   r   r   r   r   r7   rM   r   r   r   rX   rX      s<         '((((r   rX   c                   B    e Zd ZU ee         ed<   dZeed<   dZe	ed<   dS )BatchAutoTagRequestdocument_idsrU   rV   F
auto_applyN)
r   r   r   r7   r   r   rV   r8   r^   r6   r   r   r   r\   r\      sC         s)"%%%%%Jr   r\   c                   :    e Zd ZU eed<   eed<   ee         ed<   dS )BatchAutoTagResponsetotal_documents	processedresultsN)r   r   r   r   r   r7   rX   r   r   r   r`   r`      s8         NNN/""""""r   r`   c                   t   e Zd ZU eed<   eed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZ	edz  ed<   dZ
eed	<   d
Zeed<   g Zee         ed<   dZedz  ed<   dZedz  ed<   dZeed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZee         dz  ed<   dZedz  ed<   dZee         dz  ed<   dZ eed<   dZ!ee         dz  ed <   dZ"e#dz  ed!<   dZ$edz  ed"<   d#Z%eed$<   dZ&edz  ed%<   dS )&DocumentCreatetitlecontentNsummarycategory_idsourcereference_urldraftr.   T	is_publictag_ids	file_type	file_path	automaticsegmentation_modecharacter_countdoc_type
doc_numberissuing_authoritydoc_number_yeardoc_number_serial
issue_dateeffective_dateexpire_date
doc_statussupersedes_doc_idssuperseded_by_doc_idtax_type_tagsFhas_attachmentattachment_typesparse_quality_scorecontent_hash   version_numberdata_center_doc_id)'r   r   r   r   r   rh   ri   r   rj   rk   r.   rm   r6   rn   r7   ro   rp   rr   rs   rt   ru   rv   rw   rx   ry   r   rz   r{   r|   r}   r~   r   r   r   r   r8   r   r   r   r   r   r   re   re      sk        JJJLLLGS4Z"Kt"""FC$J $M3:$$$FCItGT#Y IsTz    IsTz   (s((("&OS4Z&&&HcDj!Jd
!!!$(sTz((("&OS4Z&&&$(sTz((("Jt""""&ND4K&&&#K###!Jd
!!!+/S	D(///'+#*+++&*M49t#*** ND   )-d3i$&---(,,,,#L#*###NC%)d
)))))r   re   c                       e Zd ZU dZedz  ed<   dZedz  ed<   dZedz  ed<   dZe	dz  ed<   dZ
edz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed	<   dZee	         dz  ed
<   dZe	dz  ed<   dZe	dz  ed<   dZedz  ed<   dZedz  ed<   dS )DocumentUpdateNrf   rg   rh   ri   rj   rk   r.   rm   rn   
chunk_sizechunk_overlaprr   splitter_type)r   r   r   rf   r   r   rg   rh   ri   r   rj   rk   r.   rm   r6   rn   r7   r   r   rr   r   r   r   r   r   r      s        E3:GS4ZGS4Z"Kt"""FC$J $M3:$$$FC$J!Itd{!!! $GT#Y$$$!Jd
!!! $M3:$$$$(sTz((( $M3:$$$$$r   r   c                      e Zd ZU eed<   eed<   eed<   edz  ed<   edz  ed<   dZedz  ed<   eed<   edz  ed	<   edz  ed
<   eed<   eed<   eed<   eed<   edz  ed<   dZedz  ed<   dZ	eed<   dZ
eed<   dZeed<   dZeed<   dZeed<   dZeed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed<   dZedz  ed <   dZedz  ed!<   dZedz  ed"<   dZedz  ed#<   dZedz  ed$<   dZee         dz  ed%<   dZedz  ed&<   dZee         dz  ed'<   d(Zeed)<   dZee         dz  ed*<   dZe dz  ed+<   dZ!edz  ed,<   d-Z"eed.<   e#ed/<   e#ed0<    G d1 d2          Z$dS )3DocumentResponser   rf   rg   Nrh   ri   category_name	author_idrj   rk   r.   rm   
view_countis_vectorizedvector_modelro   rq   rr   i  r      r   	recursiver   r   rs   recall_countrt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   Fr   r   r   r   r   r   r   r   c                       e Zd ZdZdS )DocumentResponse.ConfigTNr!   r   r   r   r#   r      r$   r   r#   )%r   r   r   r   r   r   r   r6   ro   rr   r   r   r   rs   r   rt   ru   rv   rw   rx   ry   r   rz   r{   r|   r}   r7   r~   r   r   r   r   r8   r   r   r   r#   r   r   r   r   r      s        GGGJJJLLL4Zt $M3:$$$NNN$J:KKKOOOOOO* IsTz   (s(((JM3$M3$$$OSL#HcDj!Jd
!!!$(sTz((("&OS4Z&&&$(sTz((("Jt""""&ND4K&&&#K###!Jd
!!!+/S	D(///'+#*+++&*M49t#*** ND   )-d3i$&---(,,,,#L#*###NC         r   r   c                   X    e Zd ZU eed<   eed<   dZedz  ed<   dZee         dz  ed<   dS )QACreatequestionanswerNri   related_documents)	r   r   r   r   r   ri   r   r   r7   r   r   r   r   r      sR         MMMKKK"Kt"""*.tCy4'.....r   r   c                       e Zd ZU eed<   eed<   eed<   edz  ed<   ee         dz  ed<   eed<   eed<   eed	<   eed
<   eed<    G d d          ZdS )
QAResponser   r   r   Nri   r   	use_counthelpful_countr.   r   r   c                       e Zd ZdZdS )QAResponse.ConfigTNr!   r   r   r   r#   r   
  r$   r   r#   )	r   r   r   r   r   r   r7   r   r#   r   r   r   r   r      s         GGGMMMKKKtCy4''''NNNKKK         r   r   c                   ^    e Zd ZU eed<   dZedz  ed<   dZedz  ed<   dZeed<   dZ	eed<   dS )	SearchRequestqueryNr   ri   keywordsearch_type
   limit)
r   r   r   r   r   r   r   ri   r   r   r   r   r   r   r     s_         JJJ$(sTz((("Kt""" K   E3OOOOOr   r   c                   P    e Zd ZU eed<   dZedz  ed<   dZedz  ed<   dZeed<   dS )VectorSearchRequestr   Nr   ri      top_k)	r   r   r   r   r   r   r   ri   r   r   r   r   r   r     sO         JJJ$(sTz((("Kt"""E3NNNNNr   r   c                   L    e Zd ZU eed<   eed<   eed<   eed<   dZedz  ed<   dS )VectorizationStatusResponserT   r.   progressr   Nr   )	r   r   r   r   r   r   r8   r6   r   r   r   r   r   r     sP         KKKOOO#L#*#####r   r   c                   0    e Zd ZU ee         ed<   eed<   dS )VectorizationTasksResponsetaskstotalN)r   r   r   r7   rK   r   r   r   r   r   r   r   %  s)         :JJJJJr   r   c                   `    e Zd ZU eed<   eed<   dZedz  ed<   dZedz  ed<   dZ	edz  ed<   dS )BatchUploadResultfilenamesuccessNrT   r   error)
r   r   r   r   r   r6   rT   r   r   r   r   r   r   r   r   *  s`         MMMMMM"Kt"""!%M4$;%%%E3:r   r   c                   D    e Zd ZU eed<   eed<   eed<   ee         ed<   dS )BatchUploadResponser   r   failedrc   N)r   r   r   r   r   r7   r   r   r   r   r   r   2  s?         JJJLLLKKK#$$$$$$r   r   c                   B    e Zd ZU eed<   eed<   eed<   eed<   eed<   dS )KnowledgeStatsResponsera   published_documentstotal_categories
total_tagstotal_qaN)r   r   r   r   r   r   r   r   r   r   9  sF         OOOMMMMMr   r   N)r   r   pydanticr   r   r   r&   r,   r:   r<   rA   rF   rM   rS   rX   r\   r`   re   r   r   r   r   r   r   r   r   r   r   r   r   r   r   <module>r      sU   # # # # # # # #          Y       y   $    )        	    ' ' ' ' '	 ' ' '    )   0    +       Y            	      & & & & &Y & & &
    i       )   # # # # #9 # # #* * * * *Y * * *D% % % % %Y % % % + + + + +y + + +\/ / / / /y / / /            I       )   $ $ $ $ $) $ $ $       
    	   % % % % %) % % %    Y     r   