Sahil1694/Atlan
0
1�
2 |�h� � � � d dl mZ d dlmZ dZdZedk �ru ed� � ed� � ee�� � Z ed e� d3�� � eee� � � Z ed � � n*# e4$ r"Z ed
e � �� � e � � Y dZ [ ndZ [ ww xY wdZ
ede
� d�� � e � e
� � Zes ed� � n ed ee� � � d�� � n*# e5$ r"Z ede � �� � e � � Y dZ [ ndZ [ ww xY w ed� � ee� � D ]d\ ZZ ededz � d�� � edej � dd � � � �� � ed � � eej � � ed � � �e ed � � dS dS ) � )�HuggingFaceEmbeddings)�Chromazdata/chroma_dbzall-MiniLM-L6-v2�__main__u" 🚀 Starting Vector Store Test...u$ 🧠 Initializing embedding model...)�6model_nameu 📚 Loading vector store from 'z'...)�persist_directory�embedding_functionu% ✅ Vector store loaded successfully!u" 🚨 Failed to load vector store: Nz5How do I add a resource or link to an existing asset?u. 7❓ Performing similarity search for query: '�'u" ⚠️ Search returned no results.u8 ✅ Found z relevant chunks.u& 🚨 An error occurred during search: z9--- Top Search Results ---z 10--- Result � z ---z**Source URL:** �sourcezN/Az
11**Content:**z--------------------u 12✨ Test complete!)� langchain_community.embeddingsr � langchain_community.vectorstoresr �VECTOR_STORE_SAVE_PATH�EMBEDDING_MODEL_NAME�__name__�print�13embeddings� vector_store� Exception�e�exit� sample_query�similarity_search�search_results�len� enumerate�i�doc�metadata�get� page_content� � �RC:\Users\Sahil\OneDrive\Desktop\Atlan Project\backend\scripts\test_vector_store.py�<module>r$ s� �� @� @� @� @� @� @� 3� 3� 3� 3� 3� 3� *� � *� � �z��� �E�14.�/�/�/� 15�E�160�1�1�1�&�&�2F�G�G�G�J� 17�E�18I�-C�19I�20I�21I�J�J�J�� �v� 4� )�22� 23� 24� � ��5�6�6�6�6�� � � � �
��6�1�6�6�7�7�7� ������������������ K�L� �E�25K�L�26K�27K�28K�L�L�L�29�%�7�7� �E�E�� � G� �E�6� 7� 7� 7� 7� �E�E�s�s�>� 2� 2�E�E�E� F� F� F��� � � � �
��:�q�:�:�;�;�;� ������������������30 31�E�32(�)�)�)��)�N�+�+� � ���3�
��'�a��c�'�'�'�(�(�(�
��D�� �!1�!1�(�E�!B�!B�D�D�E�E�E�
�� � � � �
��c� � � � �
��h����� �E�33 �!�!�!�!�!�] �s0 �34A# �#B35�(B�B36� ;C �D� C=�=D