snowflake-ml-python 1.7.2__tar.gz → 1.7.4__tar.gz

{snowflake_ml_python-1.7.2 → snowflake_ml_python-1.7.4}/CHANGELOG.md

@@ -1,18 +1,66 @@
 # Release History
 
-## 1.7.2
+## 1.7.4
+
+- FileSet: The `snowflake.ml.fileset.FileSet` has been deprecated and will be removed in a future version.
+  Use [snowflake.ml.dataset.Dataset](https://docs.snowflake.com/en/developer-guide/snowflake-ml/dataset) and
+  [snowflake.ml.data.DataConnector](https://docs.snowflake.com/en/developer-guide/snowpark-ml/reference/latest/api/data/snowflake.ml.data.data_connector.DataConnector)
+  instead.
 
 ### Bug Fixes
 
-- Model Explainability: Fix issue that explain is enabled for scikit-learn pipeline
-whose task is UNKNOWN and fails later when invoked.
+- Registry: Fixed an issue that the hugging face pipeline is loaded using incorrect dtype.
+- Registry: Fixed an issue that only 1 row is used when infer the model signature in the modeling model.
 
 ### Behavior Changes
 
+- Registry: `ModelVersion.run` on a service would require redeploying the service once account opts into nested function.
+
+### New Features
+
+- Add new `snowflake.ml.jobs` preview API for running headless workloads on SPCS using
+  [Container Runtime for ML](https://docs.snowflake.com/en/developer-guide/snowflake-ml/container-runtime-ml)
+- Added `guardrails` option to Cortex `complete` function, enabling
+  [Cortex Guard](https://docs.snowflake.com/en/user-guide/snowflake-cortex/llm-functions#cortex-guard) support
+
+## 1.7.3 (2025-01-08)
+
+- 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.
+
+### 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
+- Registry: Provided new arguments to `snowflake.ml.model.model_signature.infer_signature` method to specify rows limit
+  to be used when inferring the signature.
+
+## 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.
+
 ### 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)
 
@@ -27,8 +75,6 @@ signature inference.
 - Dataset: Fix missing `snowflake.ml.dataset.*` module exports in wheel.
 - Registry: Fix the issue that `tf_keras.Model` is not recognized as keras model when logging.
 
-### Behavior Changes
-
 ### New Features
 
 - Registry: Option to `enable_monitoring` set to False by default.  This will gate access to preview features of Model Monitoring.
@@ -277,10 +323,6 @@ data from explainablity and data lineage.
 - Modeling: Quick fix `import snowflake.ml.modeling.parameters.enable_anonymous_sproc` cannot be imported due to package
   dependency error.
 
-### Behavior Changes
-
-### New Features
-
 ## 1.5.1 (05-22-2024)
 
 ### Bug Fixes

{snowflake_ml_python-1.7.2 → snowflake_ml_python-1.7.4}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: snowflake-ml-python
-Version: 1.7.2
+Version: 1.7.4
 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: 
@@ -233,11 +233,11 @@ Requires-Python: <3.12,>=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: absl-py<2,>=0.15
-Requires-Dist: anyio<4,>=3.5.0
+Requires-Dist: anyio<5,>=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
@@ -247,23 +247,24 @@ 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
-Requires-Dist: snowflake-snowpark-python<2,>=1.17.0
+Requires-Dist: snowflake-snowpark-python!=1.26.0,<2,>=1.17.0
 Requires-Dist: sqlparse<1,>=0.4
 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"
@@ -275,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"
@@ -397,19 +399,67 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.7.2
+## 1.7.4
+
+- FileSet: The `snowflake.ml.fileset.FileSet` has been deprecated and will be removed in a future version.
+  Use [snowflake.ml.dataset.Dataset](https://docs.snowflake.com/en/developer-guide/snowflake-ml/dataset) and
+  [snowflake.ml.data.DataConnector](https://docs.snowflake.com/en/developer-guide/snowpark-ml/reference/latest/api/data/snowflake.ml.data.data_connector.DataConnector)
+  instead.
 
 ### Bug Fixes
 
-- Model Explainability: Fix issue that explain is enabled for scikit-learn pipeline
-whose task is UNKNOWN and fails later when invoked.
+- Registry: Fixed an issue that the hugging face pipeline is loaded using incorrect dtype.
+- Registry: Fixed an issue that only 1 row is used when infer the model signature in the modeling model.
 
 ### Behavior Changes
 
+- Registry: `ModelVersion.run` on a service would require redeploying the service once account opts into nested function.
+
+### New Features
+
+- Add new `snowflake.ml.jobs` preview API for running headless workloads on SPCS using
+  [Container Runtime for ML](https://docs.snowflake.com/en/developer-guide/snowflake-ml/container-runtime-ml)
+- Added `guardrails` option to Cortex `complete` function, enabling
+  [Cortex Guard](https://docs.snowflake.com/en/user-guide/snowflake-cortex/llm-functions#cortex-guard) support
+
+## 1.7.3 (2025-01-08)
+
+- 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.
+
+### 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
+- Registry: Provided new arguments to `snowflake.ml.model.model_signature.infer_signature` method to specify rows limit
+  to be used when inferring the signature.
+
+## 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.
+
 ### 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)
 
@@ -424,8 +474,6 @@ signature inference.
 - Dataset: Fix missing `snowflake.ml.dataset.*` module exports in wheel.
 - Registry: Fix the issue that `tf_keras.Model` is not recognized as keras model when logging.
 
-### Behavior Changes
-
 ### New Features
 
 - Registry: Option to `enable_monitoring` set to False by default.  This will gate access to preview features of Model Monitoring.
@@ -674,10 +722,6 @@ data from explainablity and data lineage.
 - Modeling: Quick fix `import snowflake.ml.modeling.parameters.enable_anonymous_sproc` cannot be imported due to package
   dependency error.
 
-### Behavior Changes
-
-### New Features
-
 ## 1.5.1 (05-22-2024)
 
 ### Bug Fixes

{snowflake_ml_python-1.7.2 → snowflake_ml_python-1.7.4}/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", "pyjwt>=2.0.0, <3", "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,<5", "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,!=1.26.0", "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.4/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.2 → snowflake_ml_python-1.7.4}/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.2 → snowflake_ml_python-1.7.4}/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
@@ -49,6 +49,10 @@ class CompleteOptions(TypedDict):
     generally used as an alternative to temperature. The difference is that top_p restricts the set of possible tokens
     that the model outputs, while temperature influences which tokens are chosen at each step. """
 
+    guardrails: NotRequired[bool]
+    """ A boolean value that controls whether Cortex Guard filters unsafe or harmful responses
+    from the language model. """
+
 
 class ResponseParseException(Exception):
     """This exception is raised when the server response cannot be parsed."""
@@ -56,6 +60,15 @@ class ResponseParseException(Exception):
     pass
 
 
+class GuardrailsOptions(TypedDict):
+    enabled: bool
+    """A boolean value that controls whether Cortex Guard filters unsafe or harmful responses
+    from the language model."""
+
+    response_when_unsafe: str
+    """The response to return when the language model generates unsafe or harmful content."""
+
+
 _MAX_RETRY_SECONDS = 30
 
 
@@ -117,6 +130,12 @@ def _make_request_body(
             data["temperature"] = options["temperature"]
         if "top_p" in options:
             data["top_p"] = options["top_p"]
+        if "guardrails" in options and options["guardrails"]:
+            guardrails_options: GuardrailsOptions = {
+                "enabled": True,
+                "response_when_unsafe": "Response filtered by Cortex Guard",
+            }
+            data["guardrails"] = guardrails_options
     return data
 
 
@@ -127,8 +146,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 +181,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 +189,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 +255,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 +292,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 +326,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 +343,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 +393,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 +414,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.2 → snowflake_ml_python-1.7.4}/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.2 → snowflake_ml_python-1.7.4}/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))