Aigenthix/Graph_RAG
0
1�
2 IR6j� � �x � S r SSKJrJrJrJr SSKJr SSKJ r SSK3r4\5R " \ 5 r
" S S5 rg) z Vector database service wrapper� )�List�Dict�Any�Optional� )�VectorDBFactory)�VectorDBProviderNc 6 �� � \ rS rSrSrS\S\\\4 4S jrS\ \\\4 S\74S jr SS \ \ S8\
S\ \\\4 4S jjrS \S\94S
jrS\104S jrS\\\4 4S jrSrg)�VectorDBService� z/Service for managing vector database operations�db_type�configc �F � Xl [ R " X5 U l g )N)r
r �create�provider)�selfr
r s �c/Users/ssp/Downloads/SSP7/Claude folder/KnowledgeHut/POC1/backend/app/services/vector_db_service.py�__init__�VectorDBService.__init__ s � � � �*9�*@�*@��*Q��
� �chunks�returnc � � U Vs/ s H o"S PM nnU Vs/ s H nUS R SS5 SUS 3PM" nnU Vs/ s H' nUS US R SS5 US S .US EPM) nnU R R X4U5 $ s snf s snf s snf ! [ a" n[ R S11U 35 S nAg S nAff = f)
z&Add document chunks to vector database� embedding�metadata�doc_id�unknown�_�chunk_id�text� )r r r z)Failed to add chunks to vector database: NF)�getr � add_vectors� Exception�logger�error)r r �chunk�vectors�ids� metadatas�es r �12add_chunks�VectorDBService.add_chunks s � �13 �7=�>�v�e�[�)�v�G�>� $�� #�E� ��$�(�(��9�=�>�a��j�@Q�?R�S� #�
� � $�� $�E� "�&�M� #�J� /� 3� 3�H�b� A� %�j� 1�� �J�'� � $�
� � �=�=�,�,�W�9�E� E�� ?������ � � �L�L�D�Q�C�H� I��� �s> �B" �B�B" �'B�B" �.B �5 B" �B" �"14C�,C � C�query_embedding�top_kc � � U R R X5 nU$ ! [ a$ n[ R SU 35 / s SnA$ SnAff = f)zSearch for similar chunkszSearch failed: N)r �searchr$ r% r& )r r. r/ �resultsr+ s r r1 �VectorDBService.search, sH � � ��m�m�*�*�?�B�G��N��� � �L�L�?�1�#�.� /��I�� �s � �15A
�A�A
�A
r c � � U R R U5 $ ! [ a" n[ R SU 35 SnAgSnAff = f)z Delete all chunks for a documentzFailed to delete document: NF)r �deleter$ r% r& )r r r+ s r �delete_document� VectorDBService.delete_document9 sC � � ��=�=�'�'��/� /��� � �L�L�6�q�c�:� ;��� �s � �16A �A�A c �6 � U R R 5 $ )z#Check if vector database is healthy)r � health_check�r s r r9 � VectorDBService.health_checkA s � ��}�}�)�)�+�+r c �6 � U R R 5 $ )z Get vector database statistics)r � get_statsr: s r r= �VectorDBService.get_statsE s � ��}�}�&�&�(�(r )r
r N)� )�__name__�17__module__� __qualname__�__firstlineno__�__doc__�strr r r r �boolr, �float�intr1 r6 r9 r= �__static_attributes__� r r r r s� � �9�R�� R�T�#�s�(�^� R���T�#�s�(�^�$�� 18��: � � �e�� � � � 19�d�3��8�n� � ��c� �d� �,�d� ,�)�4��S��>� )r r )rD �typingr r r r �vector_dbs.factoryr �vector_dbs.baser �logging� getLoggerr@ r% r rJ r r �<module>rP s1 �� %� ,� ,� 0� .� � � � �8� $��<)� <)r