pc.cpython-311.pyc•3.39 kB
�
?f�h� �
� � d dl Z d dlZej � d e j � e j � e j � e� � � � � � � � d dlmZ d dl m
Z
e
d�� � d dl m Z m
Z
G d� d� � Zed k r# ed
� � � Ze� d � � dS dS )
� N)�Pinecone)� load_dotenvT)�override)�embed�build_rerank_documentsc �< � e Zd Zd
defd�Zd� Zddeded ed
efd �Zd S )� Pinecone_DB�sec-embeddings�
index_namec � � || _ d| _ t t j � d� � �� � | _ d S )N� __default__�PINECONE_API_KEY)�api_key)r � namespacer �os�environ�get�pc)�selfr s � /Users/sharhad/mcp/server/pc.py�__init__zPinecone_DB.__init__ s5 � �$���&����R�Z�^�^�4F�%G�%G�H�H�H����� c �R � t | j � � � � � d S )N)�printr � list_indexes)r s r r zPinecone_DB.list_indexes s$ � �
�d�g�"�"�$�$�%�%�%�%�%r � F�query�top_k�top_n�rerankc
�� � g }t |� � }| j � | j � � � | j ||dd�� � }|ri| j j � d|t |d � � |dgdddi� � � }|j D ]$} |�
| j | j d
�� � �%n5|d D ],}
|�
|
d d |
d
d
�� � �-|S )NFT)r �vectorr �include_values�include_metadatazbge-reranker-v2-m3�matches�
chunk_text�truncate�END)�modelr � documentsr � rank_fields�return_documents�
parameters)�document�score�metadata�
original_textr/ )
r r �Indexr r r � inferencer r �data�appendr. r/ ) r r r r r �
query_results� query_embed�results�ranked_results�
ranked_result�results r r zPinecone_DB.query sF � ��
��E�l�l� ��'�-�-���0�0�6�6� �� �� "� #� 7�
�
�� � � !�W� .� 5� 5�,� � 2�7�9�3E� F� F� � +�n�#'� �� � 6�
�
�N� "0�!4�
�
�
� �$�$� -� 6� *� 0�&� &� � � � �
� "�)�,�
�
�� �$�$� &�z� 2�?� C� #�G�_�&� &� � � � � � r N)r
)r r F) �__name__�
__module__� __qualname__�strr r �int�boolr � r r r r s~ � � � � � �I� I�3� I� I� I� I�
&� &� &�! � ! �3� ! �s� ! �� ! �$� ! � ! � ! � ! � ! � ! r r �__main__r
)r z'What is the latest 10-K for Apple Inc.?)r �sys�path�insert�dirname�abspath�__file__�pineconer �dotenvr �
server.helperr r r r< r r rB r r �<module>rM s �� � � � �
�
�
�
� �����2�7�?�?�2�7�?�?�2�7�?�?�8�3L�3L�#M�#M�N�N� O� O� O� � � � � � � � � � � � � � �t� � � � � 7� 7� 7� 7� 7� 7� 7� 7�* � * � * � * � * � * � * � * �X �z��� ��"2� 3� 3� 3�B��H�H�
6�7�7�7�7�7� �r