all_the_food_public

Sleeping

Mikeplockhart commited on Apr 5, 2024

Commit

15233ff

verified ·

1 Parent(s): 06d377d

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -30,7 +30,7 @@ def chroma_upserting(collection, payload:list[dict]):
         print(type(item))
         collection.add(
             documents=item['doc'],
-            embeddings=embedding_function(item['doc']).tolist,
             #metadatas=item,
             ids=f"id_{idx}"
         )

         print(type(item))
         collection.add(
             documents=item['doc'],
+            embeddings=[list(embedding_function(item['doc']))],
             #metadatas=item,
             ids=f"id_{idx}"
         )