more debugging

Author: slice4e

engine/base_client/search.py

@@ -272,9 +272,7 @@ def process_chunk(chunk, search_one, insert_one, insert_fraction):
     for i, query in enumerate(chunk):
         if random.random() < insert_fraction:
             result = insert_one(query)
-            insert_count += 1
         else:
             result = search_one(query)
-            search_count += 1
         results.append(result)
     return results

engine/clients/redis/search.py

@@ -138,13 +138,16 @@ def insert_one(cls, doc_id: int, vector, meta_conditions):
                 if isinstance(v, dict)
             }
 
-        #print(f"DEBUG: Redis inserting doc_id={doc_id}, vector_size={len(vec_param)} bytes")
-        cls.client.hset(
-            str(doc_id),
-            mapping={
-                "vector": vec_param,
-                **payload,
-                **geopoints,
-            },
-        )
-        #print(f"DEBUG: Redis insert complete for doc_id={doc_id}")
+        try:
+            res = cls.client.hset(
+                str(doc_id),
+                mapping={
+                    "vector": vec_param,
+                    **payload,
+                    **geopoints,
+                },
+            )
+            if res == 0:
+                print(f"ERROR: Redis hset did not create a new key for doc_id={doc_id}")
+        except Exception as e:
+            print(f"ERROR: Redis hset failed for doc_id={doc_id}: {e}")