o
    hhv                     @   s<   d dl mZ d dlmZ d dlmZ d dlmZ dd ZdS )    )RecursiveCharacterTextSplitter)OpenAIEmbeddings)create_vector_search_index)get_mongo_clientc           
      C   sh   | d }t ddd}||}t }|| }|D ]}||j}	||j|	d qt| d| dS )zMSplit text & store embeddings in MongoDB collection named after character_id.Knowledge_banki     )
chunk_sizechunk_overlap)text	embeddingz.Text split & embeddings stored in collection: z. and vector search index created successfully.)r   split_documentsr   embed_querypage_content
insert_oner   )
client	documentscharacter_iddbtext_splitterchunks
embeddings
collectionchunkvector r   R/var/www/html/aiverses.in/BackEnd/app/database/vector_embedings/split_and_store.pysplit_and_store   s   

r   N)	langchain.text_splitterr   langchain_openair   Z8app.database.vector_embedings.create_vector_search_indexr   app.database.fetch_datar   r   r   r   r   r   <module>   s
    