Update agent/product_review_agent.py

agent/product_review_agent.py

@@ -20,6 +20,7 @@ from langchain_community.vectorstores import Chroma
 from langchain.prompts import ChatPromptTemplate
 from langchain.schema.output_parser import StrOutputParser
 from langchain.schema.runnable import RunnableLambda, RunnablePassthrough
+from clearml import StorageManager, Dataset
 
 import faiss
 import warnings
@@ -157,6 +158,10 @@ Remember: Always verify information against the provided context before respondi
         with open(documents_path, 'rb') as f:
             documents = pickle.load(f)
 
+    dataset = Dataset.create(dataset_name="embeddingsdataset", dataset_project="dataset_embeddings")
+    dataset.add_files(path=documents_path)
+    dataset.upload()
+    dataset.finalize()
 
     # Create FAISS index with faster search
     embeddings_np = np.array(embeddings_list).astype('float32')