Fix sparse fetch over grpc

Author: jhamon

pinecone/grpc/utils.py

@@ -48,8 +48,8 @@ def parse_fetch_response(response: Message):
     for id, vec in vectors.items():
         vd[id] = _Vector(
             id=vec["id"],
-            values=vec["values"],
-            sparse_values=parse_sparse_values(vec.get("sparseValues")),
+            values=vec.get("values", None),
+            sparse_values=parse_sparse_values(vec.get("sparseValues", None)),
             metadata=vec.get("metadata", None),
             _check_type=False,
         )

tests/integration/data/test_fetch.py

@@ -143,7 +143,6 @@ def test_fetch_unspecified_namespace(self, idx):
         assert results.vectors["1"].values is not None
         assert results.vectors["4"].metadata is not None
 
-    @pytest.mark.skip(reason="Backend implementation not ready")
     def test_fetch_sparse_index(self, sparse_idx):
         sparse_idx.upsert(
             vectors=[
@@ -165,9 +164,11 @@ def test_fetch_sparse_index(self, sparse_idx):
         assert fetch_results.namespace == ""
         assert len(fetch_results.vectors) == 10
         for i in range(10):
+            logger.debug(fetch_results.vectors[str(i)])
             assert fetch_results.vectors[str(i)].id == str(i)
-            assert fetch_results.vectors[str(i)].values is not None
-            assert len(fetch_results.vectors[str(i)].values) == 2
+            assert fetch_results.vectors[str(i)].sparse_values is not None
+            assert len(fetch_results.vectors[str(i)].sparse_values.indices) == 2
+            assert len(fetch_results.vectors[str(i)].sparse_values.values) == 2
             assert fetch_results.vectors[str(i)].metadata is not None
             assert fetch_results.vectors[str(i)].metadata["genre"] == "action"
             assert fetch_results.vectors[str(i)].metadata["runtime"] == 120