afulara/PythonicRAG-FastAPI-React
0
1�
2 ʱ�g2 � � � d dl Zd dlmZ d dlmZmZmZ d dlm Z d dl3Z4dej dej de fd�Z
G d � d5� Zed k( r�g d �Z e� Z e6j$ ej' e� � Zd
Zej+ de�� Z ede� d�e� ej1 d� Z ede� ej+ ded�� Z ede� d�e� yy)� N)� defaultdict)�List�Tuple�Callable)�EmbeddingModel�vector_a�vector_b�returnc � � t j | |� }t j j | � }t j j |� }|||z z S )z3Computes the cosine similarity between two vectors.)�np�dot�linalg�norm)r r � dot_product�norm_a�norm_bs �{/Users/aqsa/AI Projects/AIE6 Class Projects/AIE6-DeployPythonicRAG-Fastapi-react-app/backend/aimakerspace/vectordatabase.py�cosine_similarityr sE � ��&�&��8�,�K�
�Y�Y�^�^�H�
%�F�
�Y�Y�^�^�H�
%�F� �&�6�/� *�*� c �� � e Zd Zddefd�Zdedej ddfd�Ze fdej d e7d8e de e
eef fd �Ze d fd
ed e9d10e dede e
eef f11d�Zdedej fd�Zde e dd fd�Zy)�VectorDatabaseN�embedding_modelc �f � t t j � | _ |xs12 t � | _ y �N)r r �array�vectorsr r )�selfr s r �__init__zVectorDatabase.__init__ s" � �"�2�8�8�,�� � .� B�.�2B�� r �key�vectorr13 c �" � || j |< y r )r )r r r s r �insertzVectorDatabase.insert s � � "�� � �S�r � query_vector�k�distance_measurec � � | j j � D ��cg c] \ }}| |||� f�� }}}t |d� d�� d | S c c}}w )Nc � � | d S )N� � )�xs r �<lambda>z'VectorDatabase.search.<locals>.<lambda>" s14 � �A�a�D� r T)r �reverse)r �items�sorted)r r# r$ r% r r �scoress r �searchzVectorDatabase.search s\ � � $�|�|� 1� 1� 3�15���V� �"�<��8� 9�16�� 17� �f�.�$�?���C�C�� 18s �AF�19query_text�return_as_textc � � | j j |� }| j |||� }|r|D �cg c] }|d �� c}S |S c c}w )Nr )r �
get_embeddingr0 )r r1 r$ r% r2 r# �results�results r �search_by_textz VectorDatabase.search_by_text$ sL � � �+�+�9�9�*�E� ��+�+�l�A�/?�@��4B��0�f��q� �0�O��O��0s � Ac �: � | j j |d � S r )r �get)r r s r �retrieve_from_keyz VectorDatabase.retrieve_from_key/ s � ��|�|� � ��T�*�*r � list_of_textc � �� K � | j j |� � d { ��� }t ||� D ]* \ }}| j |t j20 |� � �, | S 7 �?�wr )r �async_get_embeddings�zipr" r r )r r; �21embeddings�text� embeddings r �abuild_from_listz VectorDatabase.abuild_from_list2 s[ � �� � �/�/�D�D�\�R�R�22� "�<�� <� 3�O�D�)� �K�K��b�h�h�y� 1� 2� 3�� � S�s � A#�A!�A A#r )�__name__�23__module__� __qualname__r r �strr r r" r �intr r r �floatr0 �boolr7 r: rB r) r r r r s� � �C�� C�#�#� #�r�x�x� #�D� #� &7� 24D��h�h�25D� �26D� #� 27D�28 29�e�C��J� � � 30D� &7� $� P�� P� � P� #� P�31 � P� 32�e�C��J� � �
P�+�S� +�R�X�X� +��4��9� �AQ� r r �__main__)�#I like to eat broccoli and bananas.z2I ate a banana and spinach smoothie for breakfast.z!Chinchillas and kittens are cute.z%My sister adopted a kitten yesterday.z:Look at this cute hamster munching on a piece of broccoli.� zI think fruit is awesome!)r$ zClosest z vector(s):rK zRetrieved vector:T)r$ r2 z text(s):)�numpyr � collectionsr �typingr r r �#aimakerspace.openai_utils.embeddingr �asyncior rH r r rC r; � vector_db�runrB r$ r7 �searched_vector�printr: �retrieved_vector�relevant_textsr) r r �<module>rX s �� � #� (� (� >� �+���� +�B�H�H� +�� +�&� &�R �z���L� � �I��� � �I� 6� 6�|� D�E�I� �A� �.�.�/J�a�.�P�O� �H�Q�C�{�33#�_�5� �2�2�-��� 34�35 � /�0� �-�-�#�q�� .� �N� 36�H�Q�C�y�37!�>�2�1 r