dominodatalab-data 5.9.0__tar.gz → 5.10.0.dev2__tar.gz

{dominodatalab_data-5.9.0 → dominodatalab_data-5.10.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dominodatalab-data
-Version: 5.9.0
+Version: 5.10.0.dev2
 Summary: Domino Data API for interacting with Domino Data features
 Home-page: https://github.com/dominodatalab/domino-data
 License: Apache Software License 2.0
@@ -18,7 +18,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Dist: attrs (>=20.1.0,<22.0.0)
+Requires-Dist: attrs (>=20.1.0)
 Requires-Dist: backoff (>=1.11.1,<2.0.0)
 Requires-Dist: bson (>=0.5.10,<0.6.0)
 Requires-Dist: httpx (>=0.23.0,<0.24.0)

{dominodatalab_data-5.9.0 → dominodatalab_data-5.10.0.dev2}/domino_data/_feature_store/sync.py

@@ -29,7 +29,6 @@ _import_error_message = (
     "feast is not installed.\n\n"
     "Please pip install feast:\n\n"
     "  python -m pip install feast   # install directly\n"
-    '  python -m pip install "dominodatalab-data[feast]" --upgrade   # install via extra dependency'
 )
 
 try:

dominodatalab_data-5.10.0.dev2/domino_data/vectordb.py

@@ -0,0 +1,58 @@
+import os
+
+_import_error_message = (
+    "pinecone-client is not installed.\n\n"
+    "Please pip install pinecone-client:\n\n"
+    "  python -m pip install pinecone-client   # install directly\n"
+)
+
+try:
+    from pinecone.core.client.configuration import Configuration as OpenApiConfiguration
+except ImportError as e:
+    if e.msg == "No module named 'pinecone'":
+        raise ImportError(_import_error_message) from e
+    else:
+        raise
+
+
+class DominoPineconeConfiguration(OpenApiConfiguration):
+    def __init__(
+        self,
+        datasource=None,
+        host=None,
+        api_key=None,
+        api_key_prefix=None,
+        access_token=None,
+        username=None,
+        password=None,
+        discard_unknown_keys=False,
+        disabled_client_side_validations="",
+        server_index=None,
+        server_variables=None,
+        server_operation_index=None,
+        server_operation_variables=None,
+        ssl_ca_cert=None,
+    ):
+
+        super().__init__(
+            host,
+            api_key,
+            api_key_prefix,
+            access_token,
+            username,
+            password,
+            discard_unknown_keys,
+            disabled_client_side_validations,
+            server_index,
+            server_variables,
+            server_operation_index,
+            server_operation_variables,
+            ssl_ca_cert,
+        )
+
+        self.proxy = os.getenv("DOMINO_DATA_API_GATEWAY", "http://127.0.0.1:8766")
+        self.proxy_headers = {"X-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://")

{dominodatalab_data-5.9.0 → dominodatalab_data-5.10.0.dev2}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "dominodatalab-data"
-version = "5.9.0"
+version = "5.10.0.dev2"
 description = "Domino Data API for interacting with Domino Data features"
 readme = "README.md"
 authors = [
@@ -44,7 +44,7 @@ classifiers = [  #! Update me
 python = "^3.8"
 pandas = "^1.3.0"
 httpx = "^0.23.0"
-attrs = ">=20.1.0,<22.0.0"
+attrs = ">=20.1.0"
 python-dateutil = "^2.8.0"
 pyarrow = "^10.0.0"
 loguru = "^0.5.3"
@@ -79,9 +79,12 @@ types-python-dateutil = "^2.8.19.14"
 vcrpy = "^5.0.0"
 
 [tool.poetry.group.featurestore.dependencies]
-feast = "^0.31.1"
+feast = "^0.34.1"
 GitPython = "^3.1.32"
 
+[tool.poetry.group.vectordbs.dependencies]
+pinecone-client = ">=2.2.4"
+
 [tool.black]
 # https://github.com/psf/black
 target-version = ["py38"]