o
    Fgs	                     @   s   d dl mZmZ d dlmZ d dlmZ d dlmZ e Z	G dd deZ
e	dedfd	efd
dZe	dedfd	efddZe	dedfdefddZe	dedfdefddZdS )    )FastAPIForm)	BaseModel)procesar_colecciones)actualizar_mulvusc                   @   s   e Zd ZU eed< eed< dS )RespuestaMilvus	contenido	documentoN)__name__
__module____qualname__str__annotations__ r   r   @/home/marcatoma/Documents/GitHub/back-agrota-serv-ia/app/main.pyr   
   s   
 r   z/pregunta_milvus_maquinaria.preguntac                 C   s   g d}d}g }t || }g }|D ]}|| qt|dd dd}t| |D ]\}}	|	dkr?||jd 7 }||jd	  q(t|d
krOt||d
 d}
ntddd}
t|
 |
S )N)Catalogo_Maquinarialistado_productosinformacion_garantialinks_catalogos c                 S      | d S N   r   xr   r   r   <lambda>       'envio_pregunta_milvus.<locals>.<lambda>Tkeyreverse皙?

documentr   r   r	   	ADICIONAL	r   extendsortedprintpage_contentappendmetadatalenr   r   coleccionesr   titulosdatadocssublistdocs_ordenadosr	   puntajemodelor   r   r   envio_pregunta_milvus   s&   
r8   z/pregunta_milvus_quimicosc                 C   s   ddg}d}g }t || }g }|D ]}|| qt|dd dd}t| |D ]\}}	|	dkr?||jd	 7 }||jd
  q(t|dkrOt||d d}
ntddd}
t|
 |
S )Ncatalogos_quimicoslistados_quimicosr   c                 S   r   r   r   r   r   r   r   r   6   r   r   Tr   r"   r#   r$   r   r%   r&   r'   r/   r   r   r   r8   *   s*   
z/actualizar_productos_milvuscollectionNamec                 C   s   t d t| S )Nactualizar_productos_milvus)r*   r   actualizar_productosr;   r   r   r   r<   C   s   
r<   z/separar_colecciones_milvusc                 C   s   t d t S )Nseparar_colecciones_milvus)r*   r   genear_sub_coleccionesr>   r   r   r   r?   H   s   r?   N)fastapir   r   openair   app.filtro_vectoresr   app.actualizacion_productosr   appr   postr   r8   r<   r?   r   r   r   r   <module>   s    