fix hashing

zzstoatzz · Aug 31, 2024 · ba7be56 · ba7be56
1 parent 6d6089d
commit ba7be56
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/refresh_chroma/refresh_collection.py b/examples/refresh_chroma/refresh_collection.py
@@ -69,7 +69,7 @@ async def refresh_chroma(
             await chroma.reset_collection()
             docs = await chroma.add(documents)
         elif mode == "upsert":
-            docs = await chroma.upsert(documents)
+            docs = await task(chroma.upsert)(documents)
         else:
             raise ValueError(f"Unknown mode: {mode!r} (expected 'upsert' or 'reset')")
 

diff --git a/src/raggy/documents.py b/src/raggy/documents.py
@@ -130,4 +130,4 @@ async def _create_excerpt(
 
 def get_distinct_documents(documents: Iterable["Document"]) -> Iterable["Document"]:
     """Return a list of distinct documents."""
-    return distinct(documents, key=lambda doc: doc.hash)
+    return distinct(documents, key=lambda doc: hash(doc.text))