lawliet/CS224-knowledge-discovery

1a

2�2d|�@s>dd
lZddl
mZ
ejdd�

e�d�
Zded�
d	d3�Zd
S)�N�
)
�TextEncoderz$3078e00a-e1eb-4d97-860d-2410bc2ab27f)
�api_keyznlu-background�)
�_qc�sNt�}|�
|g
�
IdH\}}|d
}tjd|
dd|id�}d|di
}||fS)Nr
znlu-background-cs224nT)�	namespace�top_k�include_values�include_metadata�vector�filter�matches)r�encode_text�index�query)r�
k�encoder�	query_vec�usage�query_response�query_response_dict�r�R/Users/sri/CS/Academia/CS224U/projects/search_background/cs224u_qa/src/retrieve.py�get_pinecone_resultss









�	�r)
r)�pineconerr�init�Indexr�strrrrrr�<module>
s

4