snowflake-ml-python 1.37.0__tar.gz → 1.38.0__tar.gz

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/CHANGELOG.md

@@ -1,6 +1,29 @@
 # Release History
 
-## 1.37.0
+## 1.38.0
+
+### New Features
+
+### Bug Fixes
+
+### Behavior Changes
+
+* Registry: For HuggingFace `text-generation` pipelines whose tokenizer defines a chat template, the
+  auto-inferred signature now matches the OpenAI Chat Completions API
+  (`_OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC`). Inputs are a single `messages` column and inference
+  controls (`temperature`, `max_completion_tokens`, `stop`, `n`, `stream`, `top_p`, `frequency_penalty`,
+  `presence_penalty`) move from `inputs` to `params` with default values. Predictions return the OpenAI
+  response shape (`id`, `object`, `created`, `model`, `choices`, `usage`); the generated text is at
+  `choices[0].message.content` instead of `outputs[0].generated_text`.
+
+* Registry: For HuggingFace `image-text-to-text`, `video-text-to-text`, and `audio-text-to-text`
+  pipelines, the auto-inferred signature now uses `_OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC` instead of
+  `_OPENAI_CHAT_SIGNATURE_SPEC`. The input column set narrows to just `messages`, and the inference
+  controls move to `params` with default values; the output schema is unchanged.
+
+### Deprecations
+
+## 1.37.0 (2026-04-29)
 
 ### New Features
 

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.37.0
+Version: 1.38.0
 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: 
@@ -250,7 +250,6 @@ 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<2026,>=2024.6.1
 Requires-Dist: scikit-learn<1.8
 Requires-Dist: scipy<2,>=1.9
 Requires-Dist: shap<1,>=0.46.0
@@ -417,7 +416,30 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.37.0
+## 1.38.0
+
+### New Features
+
+### Bug Fixes
+
+### Behavior Changes
+
+* Registry: For HuggingFace `text-generation` pipelines whose tokenizer defines a chat template, the
+  auto-inferred signature now matches the OpenAI Chat Completions API
+  (`_OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC`). Inputs are a single `messages` column and inference
+  controls (`temperature`, `max_completion_tokens`, `stop`, `n`, `stream`, `top_p`, `frequency_penalty`,
+  `presence_penalty`) move from `inputs` to `params` with default values. Predictions return the OpenAI
+  response shape (`id`, `object`, `created`, `model`, `choices`, `usage`); the generated text is at
+  `choices[0].message.content` instead of `outputs[0].generated_text`.
+
+* Registry: For HuggingFace `image-text-to-text`, `video-text-to-text`, and `audio-text-to-text`
+  pipelines, the auto-inferred signature now uses `_OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC` instead of
+  `_OPENAI_CHAT_SIGNATURE_SPEC`. The input column set narrows to just `messages`, and the inference
+  controls move to `params` with default values; the output schema is unchanged.
+
+### Deprecations
+
+## 1.37.0 (2026-04-29)
 
 ### New Features
 

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/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", "Programming Language :: Python :: 3.12", "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.14"
 dynamic = [ "version", "readme",]
-dependencies = [ "anyio>=3.5.0,<5", "cachetools>=3.1.1,<6", "cloudpickle>=2.0.0", "cryptography", "fsspec[http]>=2024.6.1,<2026", "h2>=4.3.0,<5", "importlib_resources>=6.1.1, <7", "numpy>=1.23,<3", "packaging>=20.9,<26", "pandas>=2.1.4,<3", "platformdirs<5", "pyarrow<19.0.0", "pydantic>=2.8.2, <3", "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.8", "scipy>=1.9,<2", "shap>=0.46.0,<1", "snowflake-connector-python[pandas]>=3.17.3,<5", "snowflake-snowpark-python>=1.17.0,<2,!=1.26.0", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5", "xgboost<4",]
+dependencies = [ "anyio>=3.5.0,<5", "cachetools>=3.1.1,<6", "cloudpickle>=2.0.0", "cryptography", "fsspec[http]>=2024.6.1,<2026", "h2>=4.3.0,<5", "importlib_resources>=6.1.1, <7", "numpy>=1.23,<3", "packaging>=20.9,<26", "pandas>=2.1.4,<3", "platformdirs<5", "pyarrow<19.0.0", "pydantic>=2.8.2, <3", "pyjwt>=2.0.0, <3", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "scikit-learn<1.8", "scipy>=1.9,<2", "shap>=0.46.0,<1", "snowflake-connector-python[pandas]>=3.17.3,<5", "snowflake-snowpark-python>=1.17.0,<2,!=1.26.0", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5", "xgboost<4",]
 [[project.authors]]
 name = "Snowflake, Inc"
 email = "support@snowflake.com"

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/_internal/env_utils.py

@@ -1,7 +1,9 @@
 import collections
 import copy
+import os
 import pathlib
 import re
+import sys
 import textwrap
 import warnings
 from enum import Enum
@@ -54,6 +56,24 @@ def get_execution_context() -> str:
         return "EXTERNAL"
 
 
+def is_local_conda_environment() -> bool:
+    """Return True if the active Python interpreter appears to run inside a conda env
+    or under conda session context (e.g. a venv created inside an activated conda env).
+
+    Uses the presence of a ``conda-meta`` directory under :py:data:`sys.prefix`, which is
+    standard for conda-created environments. A nested venv typically has ``sys.prefix``
+    pointing at the venv without ``conda-meta``; conda still sets ``CONDA_PREFIX`` when
+    an environment is activated, so a non-empty ``CONDA_PREFIX`` is treated as conda context.
+
+    Returns:
+        ``True`` if ``sys.prefix`` contains ``conda-meta`` or ``CONDA_PREFIX`` is set and
+        non-empty, else ``False``.
+    """
+    if os.path.isdir(os.path.join(sys.prefix, "conda-meta")):
+        return True
+    return bool(os.environ.get("CONDA_PREFIX", "").strip())
+
+
 def _validate_pip_requirement_string(req_str: str) -> requirements.Requirement:
     """Validate the input pip requirement string according to PEP 508.
 

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/feature_store/feature.py

@@ -12,9 +12,12 @@ from typing import Any, Optional
 from snowflake.ml.feature_store.aggregation import (
     AggregationSpec,
     AggregationType,
+    interval_to_seconds,
     is_lifetime_window,
 )
 
+_NO_OFFSET = "0"
+
 
 class Feature:
     """Fluent builder for defining features in a FeatureView.
@@ -96,6 +99,11 @@ class Feature:
         else:
             window_suffix = self._window.replace(" ", "").lower()
         base_name = f"{self._column}_{self._function.value}_{window_suffix}"
+
+        offset_seconds = interval_to_seconds(self._offset) if self._offset != _NO_OFFSET else 0
+        if offset_seconds > 0:
+            offset_suffix = self._offset.replace(" ", "").lower()
+            base_name = f"{base_name}_offset_{offset_suffix}"
         return base_name.upper()
 
     # Factory methods for creating features

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/feature_store/online_service.py

@@ -190,11 +190,6 @@ def create_online_service(
             error_code=error_codes.INVALID_ARGUMENT,
             original_exception=ValueError("producer_role and consumer_role must be non-empty."),
         )
-    if producer_role.strip() == consumer_role.strip():
-        raise snowml_exceptions.SnowflakeMLException(
-            error_code=error_codes.INVALID_ARGUMENT,
-            original_exception=ValueError("producer_role and consumer_role must be different roles."),
-        )
     payload = json.dumps(
         {
             "roles": {

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/jobs/_utils/payload_utils.py

@@ -7,7 +7,14 @@ import logging
 import pickle
 import sys
 import textwrap
-from importlib.abc import Traversable
+
+try:
+    from importlib.resources.abc import (
+        Traversable,  # Python 3.12+; importlib.abc.Traversable is deprecated
+    )
+except ImportError:
+    from importlib.abc import Traversable
+
 from pathlib import Path, PurePath
 from types import ModuleType
 from typing import IO, Any, Callable, Optional, Union, cast, get_args, get_origin

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/fill_mask.py

@@ -24,7 +24,7 @@ class FillMaskTaskHandler(_task_handler.HuggingFaceTaskHandler):
         **kwargs: Any,
     ) -> Any:
         input_data = X[signature.inputs[0].name].to_list()
-        return getattr(raw_model, target_method)(input_data)
+        return getattr(raw_model, target_method)(input_data, **kwargs)
 
     @override
     def _needs_list_wrapping(

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/image_classification.py

@@ -27,7 +27,7 @@ class ImageClassificationTaskHandler(_task_handler.HuggingFaceTaskHandler):
 
         input_col = signature.inputs[0].name
         images = [Image.open(io.BytesIO(img_bytes)) for img_bytes in X[input_col].to_list()]
-        return getattr(raw_model, target_method)(images)
+        return getattr(raw_model, target_method)(images, **kwargs)
 
     @override
     def _needs_list_wrapping(

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/image_feature_extraction.py

@@ -27,4 +27,4 @@ class ImageFeatureExtractionTaskHandler(_task_handler.HuggingFaceTaskHandler):
 
         input_col = signature.inputs[0].name
         images = [Image.open(io.BytesIO(img_bytes)) for img_bytes in X[input_col].to_list()]
-        return getattr(raw_model, target_method)(images)
+        return getattr(raw_model, target_method)(images, **kwargs)

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/object_detection.py

@@ -28,7 +28,7 @@ class ObjectDetectionTaskHandler(_task_handler.HuggingFaceTaskHandler):
 
         input_col = signature.inputs[0].name
         images = [Image.open(io.BytesIO(img_bytes)) for img_bytes in X[input_col].to_list()]
-        return getattr(raw_model, target_method)(images)
+        return getattr(raw_model, target_method)(images, **kwargs)
 
     @override
     def _needs_list_wrapping(

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/question_answering.py

@@ -24,7 +24,7 @@ class QuestionAnsweringTaskHandler(_task_handler.HuggingFaceTaskHandler):
         **kwargs: Any,
     ) -> Any:
         input_data = X.to_dict(orient="records")
-        return getattr(raw_model, target_method)(input_data)
+        return getattr(raw_model, target_method)(input_data, **kwargs)
 
     @override
     def _needs_list_wrapping(

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/table_question_answering.py

@@ -25,7 +25,7 @@ class TableQuestionAnsweringTaskHandler(_task_handler.HuggingFaceTaskHandler):
     ) -> Any:
         X["table"] = X["table"].apply(json.loads)
         input_data = X.to_dict(orient="records")
-        return getattr(raw_model, target_method)(input_data)
+        return getattr(raw_model, target_method)(input_data, **kwargs)
 
     @override
     def _needs_list_wrapping(

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/text_classification.py

@@ -1,5 +1,6 @@
 from typing import TYPE_CHECKING, Any
 
+import numpy as np
 import pandas as pd
 from typing_extensions import override
 
@@ -24,7 +25,13 @@ class TextClassificationTaskHandler(_task_handler.HuggingFaceTaskHandler):
         **kwargs: Any,
     ) -> Any:
         input_data = X[signature.inputs[0].name].to_list()
-        return getattr(raw_model, target_method)(input_data)
+        # TextClassificationPipeline._sanitize_parameters uses isinstance(top_k, int)
+        # which rejects np.int64 values arriving from Snowflake UDF DataFrames.
+        native_kwargs = {
+            k: int(v) if isinstance(v, np.integer) else float(v) if isinstance(v, np.floating) else v
+            for k, v in kwargs.items()
+        }
+        return getattr(raw_model, target_method)(input_data, **native_kwargs)
 
     @override
     def _needs_list_wrapping(

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/video_classification.py

@@ -39,7 +39,7 @@ class VideoClassificationTaskHandler(_task_handler.HuggingFaceTaskHandler):
                 temp_file.flush()
                 temp_file_paths.append(temp_file.name)
                 temp_files.append(temp_file)
-            return getattr(raw_model, target_method)(temp_file_paths)
+            return getattr(raw_model, target_method)(temp_file_paths, **kwargs)
         finally:
             for f in temp_files:
                 f.close()

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/visual_question_answering.py

@@ -37,7 +37,7 @@ class VisualQuestionAnsweringTaskHandler(_task_handler.HuggingFaceTaskHandler):
         def process_row(row: pd.Series) -> Any:
             pil_image = Image.open(io.BytesIO(row[input_col]))
             extra_kwargs = {k: row[k] for k in row.index if k != input_col}
-            return getattr(raw_model, target_method)(pil_image, **extra_kwargs)
+            return getattr(raw_model, target_method)(pil_image, **extra_kwargs, **kwargs)
 
         return X.apply(process_row, axis=1).to_list()
 

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/zero_shot_audio_classification.py

@@ -24,7 +24,7 @@ class ZeroShotAudioClassificationTaskHandler(_task_handler.HuggingFaceTaskHandle
     ) -> Any:
         def process_row(row: pd.Series) -> Any:
             return getattr(raw_model, target_method)(
-                row[signature.inputs[0].name], candidate_labels=row["candidate_labels"]
+                row[signature.inputs[0].name], candidate_labels=row["candidate_labels"], **kwargs
             )
 
         return X.apply(process_row, axis=1).to_list()

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/zero_shot_image_classification.py

@@ -27,6 +27,6 @@ class ZeroShotImageClassificationTaskHandler(_task_handler.HuggingFaceTaskHandle
 
         def process_row(row: pd.Series) -> Any:
             pil_image = Image.open(io.BytesIO(row[signature.inputs[0].name]))
-            return getattr(raw_model, target_method)(pil_image, candidate_labels=row["candidate_labels"])
+            return getattr(raw_model, target_method)(pil_image, candidate_labels=row["candidate_labels"], **kwargs)
 
         return X.apply(process_row, axis=1).to_list()

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/zero_shot_object_detection.py

@@ -27,6 +27,6 @@ class ZeroShotObjectDetectionTaskHandler(_task_handler.HuggingFaceTaskHandler):
 
         def process_row(row: pd.Series) -> Any:
             pil_image = Image.open(io.BytesIO(row[signature.inputs[0].name]))
-            return getattr(raw_model, target_method)(pil_image, candidate_labels=row["candidate_labels"])
+            return getattr(raw_model, target_method)(pil_image, candidate_labels=row["candidate_labels"], **kwargs)
 
         return X.apply(process_row, axis=1).to_list()

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_handlers/huggingface/zero_shot_text_classification.py

@@ -24,7 +24,7 @@ class ZeroShotTextClassificationTaskHandler(_task_handler.HuggingFaceTaskHandler
     ) -> Any:
         def process_row(row: pd.Series) -> Any:
             return getattr(raw_model, target_method)(
-                row[signature.inputs[0].name], candidate_labels=row["candidate_labels"]
+                row[signature.inputs[0].name], candidate_labels=row["candidate_labels"], **kwargs
             )
 
         return X.apply(process_row, axis=1).to_list()

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_meta/model_meta.py

@@ -31,6 +31,32 @@ _SNOWFLAKE_PKG_NAME = "snowflake"
 _SNOWFLAKE_ML_PKG_NAME = f"{_SNOWFLAKE_PKG_NAME}.ml"
 
 
+def _resolve_prefer_pip(
+    *,
+    target_platforms: Optional[list[model_types.TargetPlatform]],
+    conda_dependencies: Optional[list[str]],
+) -> bool:
+    """Return whether automatic packaging deps should go on pip (vs conda).
+
+    True for SPCS-only ``target_platforms``, or when ``model_env._ENABLE_PIP_ONLY_PACKAGING``
+    is set (client import of ``enable_pip_only_packaging``), the user gave no conda
+    dependencies, and :func:`~snowflake.ml._internal.env_utils.is_local_conda_environment`
+    is false.
+
+    Args:
+        target_platforms: Deployment targets for the model, or ``None`` for defaults.
+        conda_dependencies: User-specified conda dependency strings, or ``None`` / empty if none.
+
+    Returns:
+        Whether ``ModelEnv`` should prefer pip when adding automatic packaging dependencies.
+    """
+    if target_platforms == [model_types.TargetPlatform.SNOWPARK_CONTAINER_SERVICES]:
+        return True
+    return bool(
+        model_env._ENABLE_PIP_ONLY_PACKAGING and not conda_dependencies and not env_utils.is_local_conda_environment()
+    )
+
+
 @contextmanager
 def create_model_metadata(
     *,
@@ -103,7 +129,10 @@ def create_model_metadata(
         else:
             raise ValueError("`snowflake.ml` is imported via a way that embedding local ML library is not supported.")
 
-    prefer_pip = target_platforms == [model_types.TargetPlatform.SNOWPARK_CONTAINER_SERVICES]
+    prefer_pip = _resolve_prefer_pip(
+        target_platforms=target_platforms,
+        conda_dependencies=conda_dependencies,
+    )
     env = _create_env_for_model_metadata(
         conda_dependencies=conda_dependencies,
         pip_requirements=pip_requirements,

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py

@@ -21,7 +21,6 @@ REQUIREMENTS = [
     "pyyaml>=6.0,<7",
     "requests",
     "retrying>=1.3.3,<2",
-    "s3fs>=2024.6.1,<2026",
     "scikit-learn<1.8",
     "scipy>=1.9,<2",
     "shap>=0.46.0,<1",

{snowflake_ml_python-1.37.0 → snowflake_ml_python-1.38.0}/snowflake/ml/model/_signatures/utils.py

@@ -139,6 +139,10 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=[
+                core.ParamSpec(name="targets", dtype=core.DataType.STRING, default_value=None, shape=(-1,)),
+                core.ParamSpec(name="top_k", dtype=core.DataType.INT64, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.TokenClassificationPipeline
@@ -163,6 +167,15 @@ def huggingface_pipeline_signature_auto_infer(
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.QuestionAnsweringPipeline
     if task == "question-answering":
+        qa_params = [
+            core.ParamSpec(name="top_k", dtype=core.DataType.INT64, default_value=1),
+            core.ParamSpec(name="doc_stride", dtype=core.DataType.INT64, default_value=128),
+            core.ParamSpec(name="max_answer_len", dtype=core.DataType.INT64, default_value=15),
+            core.ParamSpec(name="max_seq_len", dtype=core.DataType.INT64, default_value=384),
+            core.ParamSpec(name="max_question_len", dtype=core.DataType.INT64, default_value=64),
+            core.ParamSpec(name="handle_impossible_answer", dtype=core.DataType.BOOL, default_value=False),
+            core.ParamSpec(name="align_to_words", dtype=core.DataType.BOOL, default_value=True),
+        ]
         # If top_k and topk is not set or set to 1, then the output is a dict per input, thus we could expand.
         if params.get("top_k", 1) == 1 and params.get("topk", 1) == 1:
             return core.ModelSignature(
@@ -176,6 +189,7 @@ def huggingface_pipeline_signature_auto_infer(
                     core.FeatureSpec(name="end", dtype=core.DataType.INT64),
                     core.FeatureSpec(name="answer", dtype=core.DataType.STRING),
                 ],
+                params=qa_params,
             )
         # Else it is a list of dict per input.
         return core.ModelSignature(
@@ -195,6 +209,7 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=qa_params,
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.SummarizationPipeline
@@ -229,10 +244,19 @@ def huggingface_pipeline_signature_auto_infer(
                 core.FeatureSpec(name="cells", dtype=core.DataType.STRING, shape=(-1,)),
                 core.FeatureSpec(name="aggregator", dtype=core.DataType.STRING),
             ],
+            params=[
+                core.ParamSpec(name="sequential", dtype=core.DataType.BOOL, default_value=False),
+                core.ParamSpec(name="padding", dtype=core.DataType.STRING, default_value=None),
+                core.ParamSpec(name="truncation", dtype=core.DataType.STRING, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.TextClassificationPipeline
     if task == "text-classification" or task == "sentiment-analysis":
+        text_cls_params = [
+            core.ParamSpec(name="top_k", dtype=core.DataType.INT64, default_value=1),
+            core.ParamSpec(name="function_to_apply", dtype=core.DataType.STRING, default_value=None),
+        ]
         # If top_k is set, return a list of dict per input
         if params.get("top_k", None) is not None:
             return core.ModelSignature(
@@ -249,6 +273,7 @@ def huggingface_pipeline_signature_auto_infer(
                         shape=(-1,),
                     ),
                 ],
+                params=text_cls_params,
             )
         # Else, return a dict per input
         return core.ModelSignature(
@@ -259,6 +284,7 @@ def huggingface_pipeline_signature_auto_infer(
                 core.FeatureSpec(name="label", dtype=core.DataType.STRING),
                 core.FeatureSpec(name="score", dtype=core.DataType.DOUBLE),
             ],
+            params=text_cls_params,
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.ImageClassificationPipeline
@@ -277,6 +303,11 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=[
+                core.ParamSpec(name="top_k", dtype=core.DataType.INT64, default_value=5),
+                core.ParamSpec(name="function_to_apply", dtype=core.DataType.STRING, default_value=None),
+                core.ParamSpec(name="timeout", dtype=core.DataType.DOUBLE, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.AutomaticSpeechRecognitionPipeline
@@ -319,6 +350,12 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=[
+                core.ParamSpec(name="top_k", dtype=core.DataType.INT64, default_value=5),
+                core.ParamSpec(name="num_frames", dtype=core.DataType.INT64, default_value=None),
+                core.ParamSpec(name="frame_sampling_rate", dtype=core.DataType.INT64, default_value=1),
+                core.ParamSpec(name="function_to_apply", dtype=core.DataType.STRING, default_value="softmax"),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.DocumentQuestionAnsweringPipeline
@@ -359,6 +396,10 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=[
+                core.ParamSpec(name="top_k", dtype=core.DataType.INT64, default_value=5),
+                core.ParamSpec(name="timeout", dtype=core.DataType.DOUBLE, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.ImageFeatureExtractionPipeline
@@ -370,6 +411,9 @@ def huggingface_pipeline_signature_auto_infer(
             outputs=[
                 core.FeatureSpec(name="feature_extraction", dtype=core.DataType.DOUBLE, shape=(-1,)),
             ],
+            params=[
+                core.ParamSpec(name="timeout", dtype=core.DataType.DOUBLE, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.ImageToTextPipeline
@@ -414,6 +458,10 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=[
+                core.ParamSpec(name="threshold", dtype=core.DataType.DOUBLE, default_value=0.5),
+                core.ParamSpec(name="timeout", dtype=core.DataType.DOUBLE, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.ZeroShotImageClassificationPipeline
@@ -433,6 +481,12 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=[
+                core.ParamSpec(
+                    name="hypothesis_template", dtype=core.DataType.STRING, default_value="This is a photo of {}."
+                ),
+                core.ParamSpec(name="timeout", dtype=core.DataType.DOUBLE, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.ZeroShotObjectDetectionPipeline
@@ -461,6 +515,11 @@ def huggingface_pipeline_signature_auto_infer(
                     shape=(-1,),
                 ),
             ],
+            params=[
+                core.ParamSpec(name="threshold", dtype=core.DataType.DOUBLE, default_value=0.1),
+                core.ParamSpec(name="top_k", dtype=core.DataType.INT64, default_value=None),
+                core.ParamSpec(name="timeout", dtype=core.DataType.DOUBLE, default_value=None),
+            ],
         )
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.TextGenerationPipeline
@@ -470,28 +529,7 @@ def huggingface_pipeline_signature_auto_infer(
                 f"Auto deployment for HuggingFace pipeline {task} "
                 "when `return_tensors` set to `True` has not been supported yet."
             )
-        # Always generate a list of dict per input
-        return core.ModelSignature(
-            inputs=[
-                core.FeatureGroupSpec(
-                    name="inputs",
-                    specs=[
-                        core.FeatureSpec(name="role", dtype=core.DataType.STRING),
-                        core.FeatureSpec(name="content", dtype=core.DataType.STRING),
-                    ],
-                    shape=(-1,),
-                ),
-            ],
-            outputs=[
-                core.FeatureGroupSpec(
-                    name="outputs",
-                    specs=[
-                        core.FeatureSpec(name="generated_text", dtype=core.DataType.STRING),
-                    ],
-                    shape=(-1,),
-                )
-            ],
-        )
+        return openai_signatures._OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC
 
     if task == "text-generation" and not has_chat_template:
         if params.get("return_tensors", False):
@@ -523,8 +561,11 @@ def huggingface_pipeline_signature_auto_infer(
                 "when `return_tensors` set to `True` has not been supported yet."
             )
         # Always generate a dict per input
-        # defaulting to OPENAI_CHAT_SIGNATURE_SPEC for image-text-to-text pipeline
-        return openai_signatures._OPENAI_CHAT_SIGNATURE_SPEC
+        # defaulting to OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC for image-text-to-text,
+        # video-text-to-text, and audio-text-to-text pipelines so inference controls
+        # (temperature, max_completion_tokens, etc.) are exposed as params with defaults
+        # rather than required input columns.
+        return openai_signatures._OPENAI_CHAT_SIGNATURE_WITH_PARAMS_SPEC
 
     # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.Text2TextGenerationPipeline
     if task == "text2text-generation":
@@ -570,6 +611,36 @@ def huggingface_pipeline_signature_auto_infer(
                 core.FeatureSpec(name="labels", dtype=core.DataType.STRING, shape=(-1,)),
                 core.FeatureSpec(name="scores", dtype=core.DataType.DOUBLE, shape=(-1,)),
             ],
+            params=[
+                core.ParamSpec(
+                    name="hypothesis_template", dtype=core.DataType.STRING, default_value="This example is {}."
+                ),
+                core.ParamSpec(name="multi_label", dtype=core.DataType.BOOL, default_value=False),
+            ],
+        )
+
+    # https://huggingface.co/docs/transformers/en/main_classes/pipelines#transformers.ZeroShotAudioClassificationPipeline
+    if task == "zero-shot-audio-classification":
+        return core.ModelSignature(
+            inputs=[
+                core.FeatureSpec(name="audio", dtype=core.DataType.BYTES),
+                core.FeatureSpec(name="candidate_labels", dtype=core.DataType.STRING, shape=(-1,)),
+            ],
+            outputs=[
+                core.FeatureGroupSpec(
+                    name="labels",
+                    specs=[
+                        core.FeatureSpec(name="label", dtype=core.DataType.STRING),
+                        core.FeatureSpec(name="score", dtype=core.DataType.DOUBLE),
+                    ],
+                    shape=(-1,),
+                ),
+            ],
+            params=[
+                core.ParamSpec(
+                    name="hypothesis_template", dtype=core.DataType.STRING, default_value="This is a sound of {}."
+                ),
+            ],
         )
 
     return None