afulara/PythonicRAG-FastAPI-React
0
1�
2 ʱ�g2 � � � S SK rS SKJr S SKJrJrJr S SKJ r S SK3r4S\R S\R S\ 4S jr
" S S55 r\S :X a� / S Qr\" 5 r\6R$ " \R' \5 5 rS
r\R+ S\S9r\" S\ S3\5 \R1 S5 r\" S\5 \R+ S\SS9r\" S\ S3\5 gg)� N)� defaultdict)�List�Tuple�Callable)�EmbeddingModel�vector_a�vector_b�returnc � � [ R " X5 n[ R R U 5 n[ R R U5 nX#U- - $ )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 sA � ��&�&��,�K�
�Y�Y�^�^�H�
%�F�
�Y�Y�^�^�H�
%�F� �6�/� *�*� c � � \ rS rSrSS\4S jjrS\S\R SS4S jr \74S \R S8\ S \ S\
\\\4 4S jjr\9S
4S\S10\ S \ S\S\
\\\4 411S jjrS\S\R 4S jrS\
\ SS 4S jrSrg)�VectorDatabase� N�embedding_modelc �p � [ [ R 5 U l U=( d12 [ 5 U l g �N)r r �array�vectorsr r )�selfr s r �__init__�VectorDatabase.__init__ s" � �"�2�8�8�,�� � .� B�.�2B�� r �key�vectorr13 c � � X R U' g r )r )r r! r" s r �insert�VectorDatabase.insert s � � "� � �S�r � query_vector�k�distance_measurec � � U R R 5 VVs/ s H u pEXC" X5 4PM nnn[ US SS9S U $ s snnf )Nc � � U S $ )N� � )�xs r �<lambda>�'VectorDatabase.search.<locals>.<lambda>" s � �A�a�Dr T)r! �reverse)r �items�sorted)r r& r' r( r! r" �scoress r �search�VectorDatabase.search s[ � � $�|�|� 1� 1� 3�14� 3� �� �"�<�8� 9� 3� � 15� �f�.�$�?���C�C�� 16s �AF�17query_text�return_as_textc � � U R R U5 nU R XRU5 nU( a U Vs/ s H owS PM sn$ U$ s snf )Nr )r �
get_embeddingr4 )r r6 r' r( r7 r& �results�results r �search_by_text� VectorDatabase.search_by_text$ sL � � �+�+�9�9�*�E� ��+�+�l�/?�@��4B��0��f�q� ��0�O��O��0s �A c �: � U R R US 5 $ r )r �get)r r! s r �retrieve_from_key� VectorDatabase.retrieve_from_key/ s � ��|�|� � ��T�*�*r � list_of_textc � �� # � U R R U5 I S h v�N n[ X5 H+ u p4U R U[ R18 " U5 5 M- U $ N@7fr )r �async_get_embeddings�zipr$ r r )r rB �19embeddings�text� embeddings r �abuild_from_list� VectorDatabase.abuild_from_list2 sQ � � � �/�/�D�D�\�R�R�20� "�<� <�O�D� �K�K��b�h�h�y� 1� 2� =�� � S�s � A$�A"�AA$)r r r )�__name__�21__module__� __qualname__�__firstlineno__r r �strr r r$ r �intr r r �floatr4 �boolr< r@ rI �__static_attributes__r, r r r r s� � �C�� C�#�#� #�r�x�x� #�D� #� &7� 22D��h�h�23D� �24D� #� 25D�26 27�e�C��J� � � 28D� &7� $� P�� P� � P� #� P�29 � P� 30�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):rU zRetrieved vector:T)r' r7 z text(s):)�numpyr � collectionsr �typingr r r �#aimakerspace.openai_utils.embeddingr �asyncior rQ r r rK rB � vector_db�runrI r' r< �searched_vector�printr@ �retrieved_vector�relevant_textsr, r r �<module>rb s� �� � #� (� (� >� �+���� +�B�H�H� +�� +�&� &�R �z���L� � �I�� � �I� 6� 6�|� D�E�I� �A� �.�.�/J�a�.�P�O� �H�Q�C�{�31#�_�5� �2�2�-��� 32�33 � /�0� �-�-�#�q�� .� �N� 34�H�Q�C�y�35!�>�2�1 r