Fixed the test failure

dominodatalab · Apr 1, 2024 · 3a78c59 · 3a78c59
1 parent b674c41
commit 3a78c59
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 20 deletions.
diff --git a/domino_data/vectordb.py b/domino_data/vectordb.py
@@ -1,4 +1,4 @@
-from typing import Dict
+from typing import Dict, Union
 
 import os
 
@@ -63,7 +63,7 @@ def get_host_from_settings(self, index, variables=None, servers=None):
         return url.replace("https://", "http://")
 
 
-def domino_pinecone3x_init_params(datasource_name: str) -> Dict[str, str]:
+def domino_pinecone3x_init_params(datasource_name: str) -> Dict[str, Union[str, Dict[str, str]]]:
     """Wrap the parameters to initialize a Pinecone 3.x client
 
     Args:
@@ -79,7 +79,7 @@ def domino_pinecone3x_init_params(datasource_name: str) -> Dict[str, str]:
     }
 
 
-def domino_pinecone3x_index_params(datasource_name: str, index_name: str) -> Dict[str, str]:
+def domino_pinecone3x_index_params(datasource_name: str, index_name: str) -> Dict[str, Union[str, Dict[str, str]]]:
     """Wrap the parameters to target an index in the Pinecone 3.x client
 
     Args:

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -83,7 +83,7 @@ feast = "^0.35.0"
 GitPython = "^3.1.41"
 
 [tool.poetry.group.vectordbs.dependencies]
-pinecone-client = ">=2.2.4"
+pinecone-client = "^2.2.4"
 
 [tool.black]
 # https://github.com/psf/black