a
    g                     @   s   d dl mZ d dlmZ d dlmZ d dlmZ edddZ	eddd	Z
edd
dZedddZedddZedddZdS )    )modelo_vector)vectores)Milvus)OpenAIEmbeddings)preguntac           
      C   s   t ddd}t|ddddd}d	}g }| }|j|d
d}|D ].\}}|dkr>||jd 7 }||jd  q>t|dkrtj||d d}	ntjd	dd}	t	|	 |	S )Ntext-embedding-3-large3sk-BJBMLzenOAiK9uGa5s5DT3BlbkFJdlwQPbVPKBFOkChjfD8rmodelopenai_api_keydanec74.208.60.7919530hostportcollection_nameZconnection_args    k皙?

	namespacer   	contenido	documento	ADICIONAL
r   r   similarity_search_with_scorepage_contentappendmetadatalenr   RespuestaPgVectorprint
r   
embeddingsZmilvusr   Ztitulosquerydocsr   Zpuntajemodelo r,   L/opt/spring_back/lhia/danec/python-danec/app/langchain_lhia/milvus_vector.pypregunta_milvusr   s,    r.   c           
      C   s   t ddd}t|ddddd}d	}g }| }|j|d
d}|D ].\}}|dkr>||jd 7 }||jd  q>t|dkrtj||d d}	ntjd	dd}	t	|	 |	S )Nr   r   r	   Zdanec_ventasr   r   r   r   r      r   r   r   r   r   r   r   r   r'   r,   r,   r-   pregunta_milvus_danec_ventas    s,    r0   c           
      C   s   t ddd}t|ddddd}d	}g }| }|j|d
d}|D ].\}}|dkr>||jd 7 }||jd  q>t|dkrtj||d d}	ntjd	dd}	t	|	 |	S )Nr   r   r	   Zvirgen_cisner   r   r   r   r   r   r   r   r   r   r   r   r   r   r'   r,   r,   r-   pregunta_milvus_vcisne:   s,    r1   c           
      C   s   t ddd}t|ddddd}d	}g }| }|j|d
d}|D ].\}}|dkr>||jd 7 }||jd  q>t|dkrtj||d d}	ntjd	dd}	t	|	 |	S )Nr   r   r	   Z	academicor   r   r   r   r   r   r   r   r   r   r   r   r   r   r'   r,   r,   r-   pregunta_milvus_academicoT   s,    r2   c           
      C   s   t ddd}t|ddddd}d	}g }| }|j|d
d}|D ].\}}|dkr>||jd 7 }||jd  q>t|dkrtj||d d}	ntjd	dd}	t	|	 |	S )Nr   r   r	   Zcoop_ambator   r   r   r   r   r   r   r   r   r   r   r   r   r   r'   r,   r,   r-   pregunta_coop_ambaton   s,    r3   c           
      C   s   t ddd}t|ddddd}d	}g }| }|j|d
d}|D ].\}}|dkr>||jd 7 }||jd  q>t|dkrtj||d d}	ntjd	dd}	t	|	 |	S )Nr   r   r	   Zcoop_andaluciaz74.249.12.122r   r   r   r   r   r   r   r   r   r   r   r   r   r'   r,   r,   r-   pregunta_coop_andalucia   s,    r4   N)
app.modelsr   Z6app.langchain_lhia.funciones_langchain.filtro_vectoresr    langchain_community.vectorstoresr   langchain_openair   strr.   r0   r1   r2   r3   r4   r,   r,   r,   r-   <module>   s   