Merge pull request #138 from pinecone-io/upsert_numpy_values

support numpy arrays

Author: igiloh-pinecone

pinecone/index.py

@@ -4,6 +4,7 @@
 from tqdm.autonotebook import tqdm
 from importlib.util import find_spec
 import numbers
+import numpy as np
 
 from collections.abc import Iterable, Mapping
 from typing import Union, List, Tuple, Optional, Dict, Any
@@ -184,6 +185,11 @@ def _dict_to_vector(item):
                         f"Column `sparse_values` is expected to be a dictionary, found {type(item['sparse_values'])}")
                 indices = item['sparse_values'].get('indices', None)
                 values = item['sparse_values'].get('values', None)
+
+                if isinstance(values, np.ndarray):
+                    values = values.tolist()
+                if isinstance(indices, np.ndarray):
+                    indices = indices.tolist()
                 try:
                     sparse_values = SparseValues(indices=indices, values=values)
                 except TypeError as e:
@@ -195,8 +201,11 @@ def _dict_to_vector(item):
             if not isinstance(metadata, Mapping):
                 raise TypeError(f"Column `metadata` is expected to be a dictionary, found {type(metadata)}")
 
+            if isinstance(item['values'], np.ndarray):
+                item['values'] = item['values'].tolist()
+
             try:
-                return Vector(id=item['id'], values=item['values'], sparse_values=sparse_values, metadata=metadata)
+                return Vector(id=item['id'],values=item['values'], sparse_values=sparse_values, metadata=metadata)
 
             except TypeError as e:
                 # if not isinstance(item['values'], Iterable) or not isinstance(item['values'][0], numbers.Real):

requirements.txt

@@ -6,4 +6,5 @@ dnspython>=2.0.0
 # openapi generated client:
 python_dateutil >= 2.5.3
 urllib3 >= 1.21.1
-tqdm >= 4.64.1
+tqdm >= 4.64.1
+numpy