Update src/document_processor.py

src/document_processor.py

@@ -27,6 +27,8 @@ class DynamicRAGUpdater:
         vector_db_path: str,
         embedding_model: str,
     ):
+        self.model_name = embedding_model
+
         self.vector_db_path = Path(vector_db_path)
 
         self.embedding_model = SentenceTransformer(
@@ -53,7 +55,8 @@ class DynamicRAGUpdater:
                     "chunks": self.chunks,
                     "chunk_id_to_idx": self.chunk_id_to_idx,
                     "embedding_dim": self.embedding_dim,
-                    "model": self.embedding_model._model_card,
+                    "model": self.model_name,
+
                 },
                 f,
             )