o
    f                     @   sD   d dl mZ d dlmZmZ defddZdededefd	d
ZdS )    )modelo_vector)vectoresvectors_namepreguntac                 C   s~   d}g }t | }|D ]\}}|dkr!||jd 7 }||jd  q
t|dkr2tj||d d}ntjddd}t| |S N g      ?z

	namespacer   )	contenido	documento	ADICIONAL)r   page_contentappendmetadatalenr   RespuestaPgVectorprint)r   r	   titulosdocsr
   puntajemodelo r   Z/home/marcatoma/Documents/GitHub/back-lhia-marcimex-ia/app/langchain_lhia/milvus_vector.pypregunta_milvusr   s   r   question
collectionamountc           	      C   s   d}g }t | ||}|D ]\}}|dkr#||jd 7 }||jd  qt|dkr4tj||d d}ntjddd}t| |S r   )r   r   r   r   r   r   r   r   )	r   r   r   r	   r   r   r
   r   r   r   r   r   question_milvusr   s   r   N)	
app.modelsr   6app.langchain_lhia.funciones_langchain.filtro_vectoresr   r   strr   intr   r   r   r   r   <module>   s    