Updated

smebuilder_vector.py

@@ -11,6 +11,7 @@ COLLECTION_NAME = "landing_page_generation_examples"
 EMBEDDING_MODEL = os.getenv("HF_EMBEDDING_MODEL", "intfloat/e5-large-v2")
 HF_CACHE_DIR = os.getenv("HF_CACHE_DIR", "/app/huggingface_cache")
 
+
 os.makedirs(HF_CACHE_DIR, exist_ok=True)
 os.makedirs(DB_LOCATION, exist_ok=True)
 
@@ -46,12 +47,12 @@ if add_documents:
             str(row.get("js_code", "")),
             str(row.get("sector", ""))
         ]).strip()
-        documents.append(Document(page_content=content, id=str(i)))
+        documents.append(Document(page_content=content, metadata={"id": str(i)}))
     
     if documents:
-        vector_store.add_documents(documents=documents, ids=[doc.id for doc in documents])
+        vector_store.add_documents(documents=documents)
 
 # ----------------- RETRIEVER -----------------
 retriever = vector_store.as_retriever(search_kwargs={"k": 20})
 
-print(f"Vector store ready with {len(df)} documents.")
+print(f"Vector store ready with {vector_store._collection.count()} documents.")