Update utils.py

utils.py

@@ -28,22 +28,20 @@ def embedding_function(items_to_embed: list[str]):
     print(len(embedded_items))
     print(type(embedded_items[0]))
     print(type(embedded_items[0][0]))
-    return embedded_items
+    return list(embedded_items)
 
 def chroma_upserting(collection, payload:list[dict]):
     print('upserting')
-    
-    #for idx, item in enumerate(payload):
-    #    print("printing item:")
-    #    print(type(item))
+    print("printing item:")
+    print(type(item))
     embedding = embedding_function([item['doc'] for item in payload])
-    #    print(type(embedding))
-    #    collection.add(
-    #        documents=item['doc'],
-    #        embeddings=[embedding],
-    #        #metadatas=item,
-    #        ids=f"id_{idx}"
-    #    )
+    print(type(embedding))
+    collection.add(
+        documents=[item['doc'] for item in payload],
+        embeddings=[embedding],
+        #metadatas=item,
+        ids=[f"id_{idx}" for idx, _ in enumerate(payload)]
+        )
 
 def search_chroma(collection, query:str):
     results = collection.query(