Fix backward compatibility with test code

OpenAgentsInc · Apr 20, 2024 · 06bfed6 · 06bfed6
1 parent acaa6c1
commit 06bfed6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -51,7 +51,9 @@ def deserializeFromJSON( self, data, marker, index_vectors ,index_content, searc
         shape=None
         data=json.loads(data)
         for part in data:
-            [text,embeddings_b64,_dtype,_shape, part_marker] = part
+            [text,embeddings_b64,_dtype,_shape] = part
+            part_marker=part[4] if len(part)>3 else None
+
             if dtype is None: dtype = _dtype
             elif dtype != _dtype: raise Exception("Data type mismatch")
             if shape is None: shape = _shape