Aigenthix/Graph_RAG
0
1+
2 R6j� � � � R t ^ RIt^ RIHt ^ RIHtHtHtHt ^ RI H t ^ RI3H t ^ RI t ] P ! ]4 t ! R R4 tR# )z*Document processing and management serviceN)�Path)�Optional�List�Dict�Any)�datetime)�uuid4c � a � ] tR t^
t o RtV 3R lR ltRV 3R lR lltV 3R lR ltV 3R lR4 ltV 3R lR lt V 3R
lR lt5Rt V t R# )�DocumentServicez Service for managing documentsc � <� V ^8� d QhRS[ /# )� �6upload_dir)�str)�format�
__classdict__s "��a/Users/ssp/Downloads/SSP/Claude folder/KnowledgeHut/POC1/backend/app/services/document_service.py� __annotate__� DocumentService.__annotate__ s �� � 7� 7�3� 7� c �n � \ V4 V n V P P R R R7 / V n R# )T)�parents�exist_okN)r r
�mkdir� documents)�selfr
s &&r �__init__�DocumentService.__init__ s, � � �z�*��� ��� � �d�T� �:�46��r Nc �t <� V ^8� d QhRS[ RS[RS[RS[S[S[S[3, , RS[S[S[3, /# )r � file_content�filename�
document_type�metadata�return)�bytesr r r r )r r s "�r r r sR �� � � �� � � � � �7 �4��S��>�*� � 8�c�3�h� �
r c � �\ " � \ \ 4 4 pV P V RV 2, pVP V4 RTRTRTR\ P9 ! 4 R\
V4 R\ V4 R^ R T;' g / /pWpP V& \ P R10V R V R 24 V# 5i)
zSave uploaded document�_�id�namer � upload_date� file_size� file_path� chunks_countr! zDocument saved: z (�))11r r r
� write_bytesr �now�lenr �logger�info)r r r r r! �doc_idr* �doc_infos &&&&& r �
save_document� DocumentService.save_document s� � � � �U�W� ���O�O����(��&<�<� � � � �l�+�
�&� �H� �]� �8�<�<�>� ��\�*� ��Y� � �A� ����B� 12�� "*���v� �� � �&�v�h�b��13�!�<�=���s �B*B,c �L <� V ^8� d QhRS[ RS[S[S[ S[3, , /# �r r2 r" )r r r r )r r s "�r r r 3 s( �� � *� *�3� *�8�D��c��N�+C� *r c �8 � V P P V4 # )zGet document info)r �get)r r2 s &&r � get_document� DocumentService.get_document3 s � ��~�~�!�!�&�)�)r c �F <� V ^8� d QhRS[ S[S[S[3, , /# )r r" )r r r r )r r s "�r r r 7 s! �� � -� -��T�#�s�(�^� 4� -r c �H � \ V P P 4 4 # )zList all documents)�listr �values)r s &r �list_documents� DocumentService.list_documents7 s � ��D�N�N�)�)�+�,�,r c �& <� V ^8� d QhRS[ RS[/# r7 )r �bool)r r s "�r r r ; s �� � 14� 15�c� 16�d� 17r c �18 � WP 9 ds V P V, p\ VR, 4 pVP 4 ' d VP 4 V P V \ P RV 24 R# R# )zDelete documentr* zDocument deleted: TF)r r �exists�unlinkr0 r1 )r r2 �docr* s && r �delete_document� DocumentService.delete_document; sj � � �^�^� #��.�.��(�C� �S�� -�.�I�� � �!�!�� � �"����v�&� �K�K�,�V�H�5� 6��r c �* <� V ^8� d QhRS[ RS[RR/# )r r2 �countr" N)r �int)r r s "�r r r G s" �� � ;� ;�#� ;�c� ;�d� ;r c �T � WP 9 d W P V, R& R# R# )z Update chunk count for documentr+ N)r )r r2 rK s &&&r �update_chunks_count�#DocumentService.update_chunks_countG s" � � �^�^� #�5:�N�N�6� "�>� 2� $r )r r
)N)
�__name__�19__module__� __qualname__�__firstlineno__�__doc__r r4 r: r@ rH rN �__static_attributes__�__classdictcell__)r s @r r20 r21
sD �� � �(�7� 7�22 � �<*� *�-� -�23� 24�;� ;r r25 )rT �os�pathlibr �typingr r r r r �uuidr �logging� getLoggerrP r0 r26 � r r �<module>r^ s7 �� 0� � � ,� ,� � � � � � �8� $��=;� =;r