Add long, double to field type (#217)

* Add long and double support to structured field * Bump version to 3.1.0
marqo-ai · Feb 8, 2024 · e718e08 · e718e08
1 parent afc1839
commit e718e08
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@
         "tox"
     ],
     name="marqo",
-    version="3.0.1",
+    version="3.1.0",
     author="marqo org",
     author_email="org@marqo.io",
     description="Tensor search for humans",

diff --git a/src/marqo/models/marqo_index.py b/src/marqo/models/marqo_index.py
@@ -14,10 +14,14 @@ class FieldType(str, Enum):
     Text = 'text'
     Bool = 'bool'
     Int = 'int'
+    Long = 'long'
     Float = 'float'
+    Double = 'double'
     ArrayText = 'array<text>'
     ArrayInt = 'array<int>'
+    ArrayLong = 'array<long>'
     ArrayFloat = 'array<float>'
+    ArrayDouble = 'array<double>'
     ImagePointer = 'image_pointer'
     MultimodalCombination = 'multimodal_combination'
 

diff --git a/tests/v2_tests/test_add_documents.py b/tests/v2_tests/test_add_documents.py
@@ -727,4 +727,4 @@ def test_add_empty_docs_batched(self):
 
             res = self.client.index(test_index_name).add_documents(documents=[], client_batch_size=5,
                                                                    tensor_fields="field a")
-            assert res == []
+            assert res == []