snowflake-ml-python 1.7.1__tar.gz → 1.7.3__tar.gz

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/CHANGELOG.md

@@ -1,6 +1,47 @@
 # Release History
 
-## 1.7.1
+## 1.7.3
+
+- Added lowercase versions of Cortex functions, added deprecation warning to Capitalized versions.
+- Bumped the requirements of `fsspec` and `s3fs` to `>=2024.6.1,<2026`
+- Bumped the requirement of `mlflow` to `>=2.16.0, <3`
+- Registry: Support 500+ features for model registry
+
+### Bug Fixes
+
+- Registry: Fixed a bug when providing non-range index pandas DataFrame as the input to a `ModelVersion.run`.
+- Registry: Improved random model version name generation to prevent collisions.
+- Registry: Fix an issue when inferring signature or running inference with Snowpark data that has a column whose type
+  is `ARRAY` and contains `NULL` value.
+- Registry: `ModelVersion.run` now accepts fully qualified service name.
+- Monitoring: Fix issue in SDK with creating monitors using fully qualified names.
+- Registry: Fix error in log_model for any sklearn models with only data pre-processing including pre-processing only
+  pipeline models due to default explainability enablement.
+
+### Behavior Changes
+
+### New Features
+
+- Added `user_files` argument to `Registry.log_model` for including images or any extra file with the model.
+- Registry: Added support for handling Hugging Face model configurations with auto-mapping functionality
+- DataConnector: Add new `DataConnector.from_sql()` constructor
+
+## 1.7.2 (2024-11-21)
+
+### Bug Fixes
+
+- Model Explainability: Fix issue that explain is enabled for scikit-learn pipeline
+whose task is UNKNOWN and fails later when invoked.
+
+### Behavior Changes
+
+### New Features
+
+- Registry: Support asynchronous model inference service creation with the `block` option
+  in `ModelVersion.create_service()` set to True by default.
+- Registry: Allow specify `batch_size` when inferencing using sentence-transformers model.
+
+## 1.7.1 (2024-11-05)
 
 ### Bug Fixes
 

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: snowflake-ml-python
-Version: 1.7.1
+Version: 1.7.3
 Summary: The machine learning client library that is used for interacting with Snowflake to build machine learning solutions.
 Author-email: "Snowflake, Inc" <support@snowflake.com>
 License: 
@@ -237,16 +237,17 @@ Requires-Dist: anyio<4,>=3.5.0
 Requires-Dist: cachetools<6,>=3.1.1
 Requires-Dist: cloudpickle>=2.0.0
 Requires-Dist: cryptography
-Requires-Dist: fsspec[http]<2024,>=2022.11
+Requires-Dist: fsspec[http]<2026,>=2024.6.1
 Requires-Dist: importlib_resources<7,>=6.1.1
 Requires-Dist: numpy<2,>=1.23
 Requires-Dist: packaging<25,>=20.9
 Requires-Dist: pandas<3,>=1.0.0
 Requires-Dist: pyarrow
+Requires-Dist: pyjwt<3,>=2.0.0
 Requires-Dist: pytimeparse<2,>=1.1.8
 Requires-Dist: pyyaml<7,>=6.0
 Requires-Dist: retrying<2,>=1.3.3
-Requires-Dist: s3fs<2024,>=2022.11
+Requires-Dist: s3fs<2026,>=2024.6.1
 Requires-Dist: scikit-learn<1.6,>=1.4
 Requires-Dist: scipy<2,>=1.9
 Requires-Dist: snowflake-connector-python[pandas]<4,>=3.5.0
@@ -256,13 +257,14 @@ Requires-Dist: typing-extensions<5,>=4.1.0
 Requires-Dist: xgboost<3,>=1.7.3
 Provides-Extra: all
 Requires-Dist: catboost<2,>=1.2.0; extra == "all"
+Requires-Dist: huggingface_hub<0.26; extra == "all"
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "all"
-Requires-Dist: mlflow<2.4,>=2.1.0; extra == "all"
+Requires-Dist: mlflow<3,>=2.16.0; extra == "all"
 Requires-Dist: peft<1,>=0.5.0; extra == "all"
 Requires-Dist: sentence-transformers<3,>=2.2.2; extra == "all"
 Requires-Dist: sentencepiece<1,>=0.1.95; extra == "all"
 Requires-Dist: shap<1,>=0.46.0; extra == "all"
-Requires-Dist: tensorflow<3,>=2.10; extra == "all"
+Requires-Dist: tensorflow<3,>=2.12.0; extra == "all"
 Requires-Dist: tokenizers<1,>=0.10; extra == "all"
 Requires-Dist: torch<2.3.0,>=2.0.1; extra == "all"
 Requires-Dist: torchdata<1,>=0.4; extra == "all"
@@ -274,15 +276,16 @@ Requires-Dist: lightgbm<5,>=4.1.0; extra == "lightgbm"
 Provides-Extra: llm
 Requires-Dist: peft<1,>=0.5.0; extra == "llm"
 Provides-Extra: mlflow
-Requires-Dist: mlflow<2.4,>=2.1.0; extra == "mlflow"
+Requires-Dist: mlflow<3,>=2.16.0; extra == "mlflow"
 Provides-Extra: shap
 Requires-Dist: shap<1,>=0.46.0; extra == "shap"
 Provides-Extra: tensorflow
-Requires-Dist: tensorflow<3,>=2.10; extra == "tensorflow"
+Requires-Dist: tensorflow<3,>=2.12.0; extra == "tensorflow"
 Provides-Extra: torch
 Requires-Dist: torch<2.3.0,>=2.0.1; extra == "torch"
 Requires-Dist: torchdata<1,>=0.4; extra == "torch"
 Provides-Extra: transformers
+Requires-Dist: huggingface_hub<0.26; extra == "transformers"
 Requires-Dist: sentence-transformers<3,>=2.2.2; extra == "transformers"
 Requires-Dist: sentencepiece<1,>=0.1.95; extra == "transformers"
 Requires-Dist: tokenizers<1,>=0.10; extra == "transformers"
@@ -302,7 +305,7 @@ and deployment process, and includes two key components.
 
 ### Snowpark ML Development
 
-[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowpark-ml-development)
+[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#ml-modeling)
 provides a collection of python APIs enabling efficient ML model development directly in Snowflake:
 
 1. Modeling API (`snowflake.ml.modeling`) for data preprocessing, feature engineering and model training in Snowflake.
@@ -316,14 +319,21 @@ their native data loader formats.
 1. FileSet API: FileSet provides a Python fsspec-compliant API for materializing data into a Snowflake internal stage
 from a query or Snowpark Dataframe along with a number of convenience APIs.
 
-### Snowpark Model Management [Public Preview]
+### Snowflake MLOps
 
-[Snowpark Model Management](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowpark-ml-ops) complements
-the Snowpark ML Development API, and provides model management capabilities along with integrated deployment into Snowflake.
+Snowflake MLOps contains suit of tools and objects to make ML development cycle. It complements
+the Snowpark ML Development API, and provides end to end development to deployment within Snowflake.
 Currently, the API consists of:
 
-1. Registry: A python API for managing models within Snowflake which also supports deployment of ML models into Snowflake
-as native MODEL object running with Snowflake Warehouse.
+1. [Registry](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-model-registry): A python API
+  allows secure deployment and management of models in Snowflake, supporting models trained both inside and outside of
+  Snowflake.
+2. [Feature Store](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-feature-store): A fully
+  integrated solution for defining, managing, storing and discovering ML features derived from your data. The
+  Snowflake Feature Store supports automated, incremental refresh from batch and streaming data sources, so that
+  feature pipelines need be defined only once to be continuously updated with new data.
+3. [Datasets](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#snowflake-datasets): Dataset provide an
+  immutable, versioned snapshot of your data suitable for ingestion by your machine learning models.
 
 ## Getting started
 
@@ -371,9 +381,66 @@ conda install \
 Note that until a `snowflake-ml-python` package version is available in the official Snowflake conda channel, there may
 be compatibility issues. Server-side functionality that `snowflake-ml-python` depends on may not yet be released.
 
+### Verifying the package
+
+1. Install cosign.
+  This example is using golang installation: [installing-cosign-with-go](https://edu.chainguard.dev/open-source/sigstore/cosign/how-to-install-cosign/#installing-cosign-with-go).
+1. Download the file from the repository like [pypi](https://pypi.org/project/snowflake-ml-python/#files).
+1. Download the signature files from the [release tag](https://github.com/snowflakedb/snowflake-ml-python/releases/tag/1.7.0).
+1. Verify signature on projects signed using Jenkins job:
+
+   ```sh
+   cosign verify-blob snowflake_ml_python-1.7.0.tar.gz --key snowflake-ml-python-1.7.0.pub --signature resources.linux.snowflake_ml_python-1.7.0.tar.gz.sig
+
+   cosign verify-blob snowflake_ml_python-1.7.0.tar.gz --key snowflake-ml-python-1.7.0.pub --signature resources.linux.snowflake_ml_python-1.7.0
+   ```
+
+NOTE: Version 1.7.0 is used as example here. Please choose the the latest version.
+
 # Release History
 
-## 1.7.1
+## 1.7.3
+
+- Added lowercase versions of Cortex functions, added deprecation warning to Capitalized versions.
+- Bumped the requirements of `fsspec` and `s3fs` to `>=2024.6.1,<2026`
+- Bumped the requirement of `mlflow` to `>=2.16.0, <3`
+- Registry: Support 500+ features for model registry
+
+### Bug Fixes
+
+- Registry: Fixed a bug when providing non-range index pandas DataFrame as the input to a `ModelVersion.run`.
+- Registry: Improved random model version name generation to prevent collisions.
+- Registry: Fix an issue when inferring signature or running inference with Snowpark data that has a column whose type
+  is `ARRAY` and contains `NULL` value.
+- Registry: `ModelVersion.run` now accepts fully qualified service name.
+- Monitoring: Fix issue in SDK with creating monitors using fully qualified names.
+- Registry: Fix error in log_model for any sklearn models with only data pre-processing including pre-processing only
+  pipeline models due to default explainability enablement.
+
+### Behavior Changes
+
+### New Features
+
+- Added `user_files` argument to `Registry.log_model` for including images or any extra file with the model.
+- Registry: Added support for handling Hugging Face model configurations with auto-mapping functionality
+- DataConnector: Add new `DataConnector.from_sql()` constructor
+
+## 1.7.2 (2024-11-21)
+
+### Bug Fixes
+
+- Model Explainability: Fix issue that explain is enabled for scikit-learn pipeline
+whose task is UNKNOWN and fails later when invoked.
+
+### Behavior Changes
+
+### New Features
+
+- Registry: Support asynchronous model inference service creation with the `block` option
+  in `ModelVersion.create_service()` set to True by default.
+- Registry: Allow specify `batch_size` when inferencing using sentence-transformers model.
+
+## 1.7.1 (2024-11-05)
 
 ### Bug Fixes
 

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/README.md

@@ -12,7 +12,7 @@ and deployment process, and includes two key components.
 
 ### Snowpark ML Development
 
-[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowpark-ml-development)
+[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#ml-modeling)
 provides a collection of python APIs enabling efficient ML model development directly in Snowflake:
 
 1. Modeling API (`snowflake.ml.modeling`) for data preprocessing, feature engineering and model training in Snowflake.
@@ -26,14 +26,21 @@ their native data loader formats.
 1. FileSet API: FileSet provides a Python fsspec-compliant API for materializing data into a Snowflake internal stage
 from a query or Snowpark Dataframe along with a number of convenience APIs.
 
-### Snowpark Model Management [Public Preview]
+### Snowflake MLOps
 
-[Snowpark Model Management](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowpark-ml-ops) complements
-the Snowpark ML Development API, and provides model management capabilities along with integrated deployment into Snowflake.
+Snowflake MLOps contains suit of tools and objects to make ML development cycle. It complements
+the Snowpark ML Development API, and provides end to end development to deployment within Snowflake.
 Currently, the API consists of:
 
-1. Registry: A python API for managing models within Snowflake which also supports deployment of ML models into Snowflake
-as native MODEL object running with Snowflake Warehouse.
+1. [Registry](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-model-registry): A python API
+  allows secure deployment and management of models in Snowflake, supporting models trained both inside and outside of
+  Snowflake.
+2. [Feature Store](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-feature-store): A fully
+  integrated solution for defining, managing, storing and discovering ML features derived from your data. The
+  Snowflake Feature Store supports automated, incremental refresh from batch and streaming data sources, so that
+  feature pipelines need be defined only once to be continuously updated with new data.
+3. [Datasets](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#snowflake-datasets): Dataset provide an
+  immutable, versioned snapshot of your data suitable for ingestion by your machine learning models.
 
 ## Getting started
 
@@ -80,3 +87,19 @@ conda install \
 
 Note that until a `snowflake-ml-python` package version is available in the official Snowflake conda channel, there may
 be compatibility issues. Server-side functionality that `snowflake-ml-python` depends on may not yet be released.
+
+### Verifying the package
+
+1. Install cosign.
+  This example is using golang installation: [installing-cosign-with-go](https://edu.chainguard.dev/open-source/sigstore/cosign/how-to-install-cosign/#installing-cosign-with-go).
+1. Download the file from the repository like [pypi](https://pypi.org/project/snowflake-ml-python/#files).
+1. Download the signature files from the [release tag](https://github.com/snowflakedb/snowflake-ml-python/releases/tag/1.7.0).
+1. Verify signature on projects signed using Jenkins job:
+
+   ```sh
+   cosign verify-blob snowflake_ml_python-1.7.0.tar.gz --key snowflake-ml-python-1.7.0.pub --signature resources.linux.snowflake_ml_python-1.7.0.tar.gz.sig
+
+   cosign verify-blob snowflake_ml_python-1.7.0.tar.gz --key snowflake-ml-python-1.7.0.pub --signature resources.linux.snowflake_ml_python-1.7.0
+   ```
+
+NOTE: Version 1.7.0 is used as example here. Please choose the the latest version.

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/pyproject.toml

@@ -8,7 +8,7 @@ description = "The machine learning client library that is used for interacting
 classifiers = [ "Development Status :: 5 - Production/Stable", "Environment :: Console", "Environment :: Other Environment", "Intended Audience :: Developers", "Intended Audience :: Education", "Intended Audience :: Information Technology", "Intended Audience :: System Administrators", "License :: OSI Approved :: Apache Software License", "Operating System :: OS Independent", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", "Topic :: Database", "Topic :: Software Development", "Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries :: Application Frameworks", "Topic :: Software Development :: Libraries :: Python Modules", "Topic :: Scientific/Engineering :: Information Analysis",]
 requires-python = ">=3.9, <3.12"
 dynamic = [ "version", "readme",]
-dependencies = [ "absl-py>=0.15,<2", "anyio>=3.5.0,<4", "cachetools>=3.1.1,<6", "cloudpickle>=2.0.0", "cryptography", "fsspec[http]>=2022.11,<2024", "importlib_resources>=6.1.1, <7", "numpy>=1.23,<2", "packaging>=20.9,<25", "pandas>=1.0.0,<3", "pyarrow", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "s3fs>=2022.11,<2024", "scikit-learn>=1.4,<1.6", "scipy>=1.9,<2", "snowflake-connector-python[pandas]>=3.5.0,<4", "snowflake-snowpark-python>=1.17.0,<2", "sqlparse>=0.4,<1", "typing-extensions>=4.1.0,<5", "xgboost>=1.7.3,<3",]
+dependencies = [ "absl-py>=0.15,<2", "anyio>=3.5.0,<4", "cachetools>=3.1.1,<6", "cloudpickle>=2.0.0", "cryptography", "fsspec[http]>=2024.6.1,<2026", "importlib_resources>=6.1.1, <7", "numpy>=1.23,<2", "packaging>=20.9,<25", "pandas>=1.0.0,<3", "pyarrow", "pyjwt>=2.0.0, <3", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "s3fs>=2024.6.1,<2026", "scikit-learn>=1.4,<1.6", "scipy>=1.9,<2", "snowflake-connector-python[pandas]>=3.5.0,<4", "snowflake-snowpark-python>=1.17.0,<2", "sqlparse>=0.4,<1", "typing-extensions>=4.1.0,<5", "xgboost>=1.7.3,<3",]
 [[project.authors]]
 name = "Snowflake, Inc"
 email = "support@snowflake.com"
@@ -24,15 +24,15 @@ Issues = "https://github.com/snowflakedb/snowflake-ml-python/issues"
 Changelog = "https://github.com/snowflakedb/snowflake-ml-python/blob/master/CHANGELOG.md"
 
 [project.optional-dependencies]
-all = [ "catboost>=1.2.0, <2", "lightgbm>=4.1.0, <5", "mlflow>=2.1.0,<2.4", "peft>=0.5.0,<1", "sentence-transformers>=2.2.2,<3", "sentencepiece>=0.1.95,<1", "shap>=0.46.0,<1", "tensorflow>=2.10,<3", "tokenizers>=0.10,<1", "torch>=2.0.1,<2.3.0", "torchdata>=0.4,<1", "transformers>=4.32.1,<5",]
+all = [ "catboost>=1.2.0, <2", "huggingface_hub<0.26", "lightgbm>=4.1.0, <5", "mlflow>=2.16.0, <3", "peft>=0.5.0,<1", "sentence-transformers>=2.2.2,<3", "sentencepiece>=0.1.95,<1", "shap>=0.46.0,<1", "tensorflow>=2.12.0,<3", "tokenizers>=0.10,<1", "torch>=2.0.1,<2.3.0", "torchdata>=0.4,<1", "transformers>=4.32.1,<5",]
 catboost = [ "catboost>=1.2.0, <2",]
 lightgbm = [ "lightgbm>=4.1.0, <5",]
 llm = [ "peft>=0.5.0,<1",]
-mlflow = [ "mlflow>=2.1.0,<2.4",]
+mlflow = [ "mlflow>=2.16.0, <3",]
 shap = [ "shap>=0.46.0,<1",]
-tensorflow = [ "tensorflow>=2.10,<3",]
+tensorflow = [ "tensorflow>=2.12.0,<3",]
 torch = [ "torch>=2.0.1,<2.3.0", "torchdata>=0.4,<1",]
-transformers = [ "sentence-transformers>=2.2.2,<3", "sentencepiece>=0.1.95,<1", "tokenizers>=0.10,<1", "transformers>=4.32.1,<5",]
+transformers = [ "huggingface_hub<0.26", "sentence-transformers>=2.2.2,<3", "sentencepiece>=0.1.95,<1", "tokenizers>=0.10,<1", "transformers>=4.32.1,<5",]
 
 [tool.setuptools.package-data]
 "*" = [ "*",]

snowflake_ml_python-1.7.3/snowflake/cortex/__init__.py

@@ -0,0 +1,32 @@
+from snowflake.cortex._classify_text import ClassifyText, classify_text
+from snowflake.cortex._complete import Complete, CompleteOptions, complete
+from snowflake.cortex._embed_text_768 import EmbedText768, embed_text_768
+from snowflake.cortex._embed_text_1024 import EmbedText1024, embed_text_1024
+from snowflake.cortex._extract_answer import ExtractAnswer, extract_answer
+from snowflake.cortex._finetune import Finetune, FinetuneJob, FinetuneStatus
+from snowflake.cortex._sentiment import Sentiment, sentiment
+from snowflake.cortex._summarize import Summarize, summarize
+from snowflake.cortex._translate import Translate, translate
+
+__all__ = [
+    "ClassifyText",
+    "classify_text",
+    "Complete",
+    "complete",
+    "CompleteOptions",
+    "EmbedText768",
+    "embed_text_768",
+    "EmbedText1024",
+    "embed_text_1024",
+    "ExtractAnswer",
+    "extract_answer",
+    "Finetune",
+    "FinetuneJob",
+    "FinetuneStatus",
+    "Sentiment",
+    "sentiment",
+    "Summarize",
+    "summarize",
+    "Translate",
+    "translate",
+]

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/snowflake/cortex/_classify_text.py

@@ -1,5 +1,7 @@
 from typing import List, Optional, Union, cast
 
+from typing_extensions import deprecated
+
 from snowflake import snowpark
 from snowflake.cortex._util import CORTEX_FUNCTIONS_TELEMETRY_PROJECT, call_sql_function
 from snowflake.ml._internal import telemetry
@@ -8,7 +10,7 @@ from snowflake.ml._internal import telemetry
 @telemetry.send_api_usage_telemetry(
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
-def ClassifyText(
+def classify_text(
     str_input: Union[str, snowpark.Column],
     categories: Union[List[str], snowpark.Column],
     session: Optional[snowpark.Session] = None,
@@ -34,3 +36,12 @@ def _classify_text_impl(
     session: Optional[snowpark.Session] = None,
 ) -> Union[str, snowpark.Column]:
     return cast(Union[str, snowpark.Column], call_sql_function(function, session, str_input, categories))
+
+
+ClassifyText = deprecated(
+    "ClassifyText() is deprecated and will be removed in a future release. Please use classify_text() instead."
+)(
+    telemetry.send_api_usage_telemetry(
+        project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
+    )(classify_text)
+)

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/snowflake/cortex/_complete.py

@@ -6,7 +6,7 @@ from typing import Any, Callable, Dict, Iterator, List, Optional, TypedDict, Uni
 from urllib.parse import urlunparse
 
 import requests
-from typing_extensions import NotRequired
+from typing_extensions import NotRequired, deprecated
 
 from snowflake import snowpark
 from snowflake.cortex._sse_client import SSEClient
@@ -127,8 +127,26 @@ def _xp_dict_to_response(raw_resp: Dict[str, Any]) -> requests.Response:
     response.status_code = int(raw_resp["status"])
     response.headers = raw_resp["headers"]
 
+    request_id = None
+    for key, value in raw_resp["headers"].items():
+        # Note: there is some whitespace in the headers making it not possible
+        # to directly index the header reliably.
+        if key.strip().lower() == "x-snowflake-request-id":
+            request_id = value
+            break
+
     data = raw_resp["content"]
-    data = json.loads(data)
+    try:
+        data = json.loads(data)
+    except json.JSONDecodeError:
+        raise ValueError(f"Request failed (request id: {request_id})")
+
+    if response.status_code < 200 or response.status_code >= 300:
+        if "message" not in data:
+            raise ValueError(f"Request failed (request id: {request_id})")
+        message = data["message"]
+        raise ValueError(f"Request failed: {message} (request id: {request_id})")
+
     # Convert the dictionary to a string format that resembles the SSE event format
     # For example, if the dict is {'event': 'message', 'data': 'your data'}, it should be formatted like this:
     sse_format_data = ""
@@ -144,6 +162,7 @@ def _xp_dict_to_response(raw_resp: Dict[str, Any]) -> requests.Response:
 
 @retry
 def _call_complete_xp(
+    snow_api_xp_request_handler: Optional[Callable[..., Dict[str, Any]]],
     model: str,
     prompt: Union[str, List[ConversationMessage]],
     options: Optional[CompleteOptions] = None,
@@ -151,9 +170,8 @@ def _call_complete_xp(
 ) -> requests.Response:
     headers = _make_common_request_headers()
     body = _make_request_body(model, prompt, options)
-    import _snowflake
-
-    raw_resp = _snowflake.send_snow_api_request("POST", _REST_COMPLETE_URL, {}, headers, body, {}, deadline)
+    assert snow_api_xp_request_handler is not None
+    raw_resp = snow_api_xp_request_handler("POST", _REST_COMPLETE_URL, {}, headers, body, {}, deadline)
     return _xp_dict_to_response(raw_resp)
 
 
@@ -218,17 +236,26 @@ def _complete_call_sql_function_snowpark(
 
 
 def _complete_non_streaming_immediate(
+    snow_api_xp_request_handler: Optional[Callable[..., Dict[str, Any]]],
     model: str,
     prompt: Union[str, List[ConversationMessage]],
     options: Optional[CompleteOptions],
     session: Optional[snowpark.Session] = None,
     deadline: Optional[float] = None,
 ) -> str:
-    response = _complete_rest(model=model, prompt=prompt, options=options, session=session, deadline=deadline)
+    response = _complete_rest(
+        snow_api_xp_request_handler=snow_api_xp_request_handler,
+        model=model,
+        prompt=prompt,
+        options=options,
+        session=session,
+        deadline=deadline,
+    )
     return "".join(response)
 
 
 def _complete_non_streaming_impl(
+    snow_api_xp_request_handler: Optional[Callable[..., Dict[str, Any]]],
     function: str,
     model: Union[str, snowpark.Column],
     prompt: Union[str, List[ConversationMessage], snowpark.Column],
@@ -246,19 +273,31 @@ def _complete_non_streaming_impl(
     if isinstance(options, snowpark.Column):
         raise ValueError("'options' cannot be a snowpark.Column when 'prompt' is a string.")
     return _complete_non_streaming_immediate(
-        model=model, prompt=prompt, options=options, session=session, deadline=deadline
+        snow_api_xp_request_handler=snow_api_xp_request_handler,
+        model=model,
+        prompt=prompt,
+        options=options,
+        session=session,
+        deadline=deadline,
     )
 
 
 def _complete_rest(
+    snow_api_xp_request_handler: Optional[Callable[..., Dict[str, Any]]],
     model: str,
     prompt: Union[str, List[ConversationMessage]],
     options: Optional[CompleteOptions] = None,
     session: Optional[snowpark.Session] = None,
     deadline: Optional[float] = None,
 ) -> Iterator[str]:
-    if is_in_stored_procedure():  # type: ignore[no-untyped-call]
-        response = _call_complete_xp(model=model, prompt=prompt, options=options, deadline=deadline)
+    if snow_api_xp_request_handler is not None:
+        response = _call_complete_xp(
+            snow_api_xp_request_handler=snow_api_xp_request_handler,
+            model=model,
+            prompt=prompt,
+            options=options,
+            deadline=deadline,
+        )
     else:
         response = _call_complete_rest(model=model, prompt=prompt, options=options, session=session, deadline=deadline)
     assert response.status_code >= 200 and response.status_code < 300
@@ -268,10 +307,11 @@ def _complete_rest(
 def _complete_impl(
     model: Union[str, snowpark.Column],
     prompt: Union[str, List[ConversationMessage], snowpark.Column],
+    snow_api_xp_request_handler: Optional[Callable[..., Dict[str, Any]]] = None,
+    function: str = "snowflake.cortex.complete",
     options: Optional[CompleteOptions] = None,
     session: Optional[snowpark.Session] = None,
     stream: bool = False,
-    function: str = "snowflake.cortex.complete",
     timeout: Optional[float] = None,
     deadline: Optional[float] = None,
 ) -> Union[str, Iterator[str], snowpark.Column]:
@@ -284,14 +324,29 @@ def _complete_impl(
             raise ValueError("in REST mode, 'model' must be a string")
         if not isinstance(prompt, str) and not isinstance(prompt, List):
             raise ValueError("in REST mode, 'prompt' must be a string or a list of ConversationMessage")
-        return _complete_rest(model=model, prompt=prompt, options=options, session=session, deadline=deadline)
-    return _complete_non_streaming_impl(function, model, prompt, options, session, deadline)
+        return _complete_rest(
+            snow_api_xp_request_handler=snow_api_xp_request_handler,
+            model=model,
+            prompt=prompt,
+            options=options,
+            session=session,
+            deadline=deadline,
+        )
+    return _complete_non_streaming_impl(
+        snow_api_xp_request_handler=snow_api_xp_request_handler,
+        function=function,
+        model=model,
+        prompt=prompt,
+        options=options,
+        session=session,
+        deadline=deadline,
+    )
 
 
 @telemetry.send_api_usage_telemetry(
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
-def Complete(
+def complete(
     model: Union[str, snowpark.Column],
     prompt: Union[str, List[ConversationMessage], snowpark.Column],
     *,
@@ -319,10 +374,19 @@ def Complete(
     Returns:
         A column of string responses.
     """
+
+    # Set the XP snow api function, if available.
+    snow_api_xp_request_handler = None
+    if is_in_stored_procedure():  # type: ignore[no-untyped-call]
+        import _snowflake
+
+        snow_api_xp_request_handler = _snowflake.send_snow_api_request
+
     try:
         return _complete_impl(
             model,
             prompt,
+            snow_api_xp_request_handler=snow_api_xp_request_handler,
             options=options,
             session=session,
             stream=stream,
@@ -331,3 +395,8 @@ def Complete(
         )
     except ValueError as err:
         raise err
+
+
+Complete = deprecated("Complete() is deprecated and will be removed in a future release. Use complete() instead")(
+    telemetry.send_api_usage_telemetry(project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT)(complete)
+)

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/snowflake/cortex/_embed_text_1024.py

@@ -1,5 +1,7 @@
 from typing import List, Optional, Union, cast
 
+from typing_extensions import deprecated
+
 from snowflake import snowpark
 from snowflake.cortex._util import CORTEX_FUNCTIONS_TELEMETRY_PROJECT, call_sql_function
 from snowflake.ml._internal import telemetry
@@ -8,12 +10,12 @@ from snowflake.ml._internal import telemetry
 @telemetry.send_api_usage_telemetry(
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
-def EmbedText1024(
+def embed_text_1024(
     model: Union[str, snowpark.Column],
     text: Union[str, snowpark.Column],
     session: Optional[snowpark.Session] = None,
 ) -> Union[List[float], snowpark.Column]:
-    """TextEmbed calls into the LLM inference service to embed the text.
+    """Calls into the LLM inference service to embed the text.
 
     Args:
         model: A Column of strings representing the model to use for embedding. The value
@@ -35,3 +37,8 @@ def _embed_text_1024_impl(
     session: Optional[snowpark.Session] = None,
 ) -> Union[List[float], snowpark.Column]:
     return cast(Union[List[float], snowpark.Column], call_sql_function(function, session, model, text))
+
+
+EmbedText1024 = deprecated(
+    "EmbedText1024() is deprecated and will be removed in a future release. Use embed_text_1024() instead"
+)(telemetry.send_api_usage_telemetry(project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT)(embed_text_1024))

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/snowflake/cortex/_embed_text_768.py

@@ -1,5 +1,7 @@
 from typing import List, Optional, Union, cast
 
+from typing_extensions import deprecated
+
 from snowflake import snowpark
 from snowflake.cortex._util import CORTEX_FUNCTIONS_TELEMETRY_PROJECT, call_sql_function
 from snowflake.ml._internal import telemetry
@@ -8,12 +10,12 @@ from snowflake.ml._internal import telemetry
 @telemetry.send_api_usage_telemetry(
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
-def EmbedText768(
+def embed_text_768(
     model: Union[str, snowpark.Column],
     text: Union[str, snowpark.Column],
     session: Optional[snowpark.Session] = None,
 ) -> Union[List[float], snowpark.Column]:
-    """TextEmbed calls into the LLM inference service to embed the text.
+    """Calls into the LLM inference service to embed the text.
 
     Args:
         model: A Column of strings representing the model to use for embedding. The value
@@ -35,3 +37,8 @@ def _embed_text_768_impl(
     session: Optional[snowpark.Session] = None,
 ) -> Union[List[float], snowpark.Column]:
     return cast(Union[List[float], snowpark.Column], call_sql_function(function, session, model, text))
+
+
+EmbedText768 = deprecated(
+    "EmbedText768() is deprecated and will be removed in a future release. Use embed_text_768() instead"
+)(telemetry.send_api_usage_telemetry(project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT)(embed_text_768))

{snowflake_ml_python-1.7.1 → snowflake_ml_python-1.7.3}/snowflake/cortex/_extract_answer.py

@@ -1,5 +1,7 @@
 from typing import Optional, Union, cast
 
+from typing_extensions import deprecated
+
 from snowflake import snowpark
 from snowflake.cortex._util import CORTEX_FUNCTIONS_TELEMETRY_PROJECT, call_sql_function
 from snowflake.ml._internal import telemetry
@@ -8,12 +10,12 @@ from snowflake.ml._internal import telemetry
 @telemetry.send_api_usage_telemetry(
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
-def ExtractAnswer(
+def extract_answer(
     from_text: Union[str, snowpark.Column],
     question: Union[str, snowpark.Column],
     session: Optional[snowpark.Session] = None,
 ) -> Union[str, snowpark.Column]:
-    """ExtractAnswer calls into the LLM inference service to extract an answer from within specified text.
+    """Calls into the LLM inference service to extract an answer from within specified text.
 
     Args:
         from_text: A Column of strings representing input text.
@@ -34,3 +36,8 @@ def _extract_answer_impl(
     session: Optional[snowpark.Session] = None,
 ) -> Union[str, snowpark.Column]:
     return cast(Union[str, snowpark.Column], call_sql_function(function, session, from_text, question))
+
+
+ExtractAnswer = deprecated(
+    "ExtractAnswer() is deprecated and will be removed in a future release. Use extract_answer() instead"
+)(telemetry.send_api_usage_telemetry(project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT)(extract_answer))