Fix error querying PineconeVectorStore using sparse query mode (run-l…

…lama#12967) * fix sparse query pinecone * none checking
mattf · Apr 25, 2024 · 3028f7d · 3028f7d
1 parent 77459a7
commit 3028f7d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/...ctor_stores/llama-index-vector-stores-pinecone/llama_index/vector_stores/pinecone/base.py b/...ctor_stores/llama-index-vector-stores-pinecone/llama_index/vector_stores/pinecone/base.py
@@ -433,7 +433,13 @@ def query(self, query: VectorStoreQuery, **kwargs: Any) -> VectorStoreQueryResul
                     "values": [v * (1 - query.alpha) for v in sparse_vector["values"]],
                 }
 
-        query_embedding = None
+        # pinecone requires a query embedding, so default to 0s if not provided
+        if query.query_embedding is not None:
+            dimension = len(query.query_embedding)
+        else:
+            dimension = self._pinecone_index.describe_index_stats()["dimension"]
+        query_embedding = [0.0] * dimension
+
         if query.mode in (VectorStoreQueryMode.DEFAULT, VectorStoreQueryMode.HYBRID):
             query_embedding = cast(List[float], query.query_embedding)
             if query.alpha is not None:

diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-pinecone/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-pinecone/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-vector-stores-pinecone"
 readme = "README.md"
-version = "0.1.4"
+version = "0.1.5"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<3.13"