dominodatalab-data 5.10.0.dev3__tar.gz → 5.11.0.dev1__tar.gz

{dominodatalab_data-5.10.0.dev3 → dominodatalab_data-5.11.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dominodatalab-data
-Version: 5.10.0.dev3
+Version: 5.11.0.dev1
 Summary: Domino Data API for interacting with Domino Data features
 Home-page: https://github.com/dominodatalab/domino-data
 License: Apache Software License 2.0

{dominodatalab_data-5.10.0.dev3 → dominodatalab_data-5.11.0.dev1}/domino_data/vectordb.py

@@ -1,3 +1,5 @@
+from typing import Dict
+
 import os
 
 _import_error_message = (
@@ -14,6 +16,9 @@ except ImportError as e:
     else:
         raise
 
+HEADER_DOMINO_DATASOURCE = "X-Domino-Datasource"
+HEADER_PINECONE_INDEX = "X-Domino-Pinecone-Index"
+
 
 class DominoPineconeConfiguration(OpenApiConfiguration):
     def __init__(
@@ -51,8 +56,43 @@ class DominoPineconeConfiguration(OpenApiConfiguration):
         )
 
         self.proxy = os.getenv("DOMINO_DATA_API_GATEWAY", "http://127.0.0.1:8766")
-        self.proxy_headers = {"X-Domino-Datasource": datasource}
+        self.proxy_headers = {HEADER_DOMINO_DATASOURCE: datasource}
 
     def get_host_from_settings(self, index, variables=None, servers=None):
         url = super().get_host_from_settings(index, variables, servers)
         return url.replace("https://", "http://")
+
+
+def domino_pinecone3x_init_params(datasource_name: str) -> Dict[str, str]:
+    """Wrap the parameters to initialize a Pinecone 3.x client
+
+    Args:
+        datasource_name: the name of the Pinecone data source
+
+    Returns:
+        A dictionary of parameters to initialize the Pinecone client
+    """
+    return {
+        "api_key": "domino",
+        "host": os.getenv("DOMINO_DATA_API_GATEWAY", "http://127.0.0.1:8766"),
+        "additional_headers": {HEADER_DOMINO_DATASOURCE: datasource_name},
+    }
+
+
+def domino_pinecone3x_index_params(datasource_name: str, index_name: str) -> Dict[str, str]:
+    """Wrap the parameters to target an index in the Pinecone 3.x client
+
+    Args:
+        datasource_name: the name of the Pinecone data source
+        index_name: the name of the index
+
+    Returns:
+        A dictionary of parameters to target the Pinecone index for vector operations
+    """
+    return {
+        "host": os.getenv("DOMINO_DATA_API_GATEWAY", "http://127.0.0.1:8766"),
+        "additional_headers": {
+            HEADER_DOMINO_DATASOURCE: datasource_name,
+            HEADER_PINECONE_INDEX: index_name,
+        },
+    }

{dominodatalab_data-5.10.0.dev3 → dominodatalab_data-5.11.0.dev1}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "dominodatalab-data"
-version = "5.10.0.dev3"
+version = "5.11.0.dev1"
 description = "Domino Data API for interacting with Domino Data features"
 readme = "README.md"
 authors = [