snowflake-ml-python 2.1.0__py3-none-any.whl → 2.2.0__py3-none-any.whl

snowflake/cortex/_classify_text.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, cast
+from typing import Union, cast
 
 from typing_extensions import deprecated
 
@@ -11,10 +11,10 @@ from snowflake.ml._internal import telemetry
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
 def classify_text(
-    str_input: Union[str, snowpark.Column],
-    categories: Union[list[str], snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    str_input: str | snowpark.Column,
+    categories: list[str] | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     """Use the LLM inference service to classify the INPUT text into one of the target CATEGORIES.
 
     Args:
@@ -31,10 +31,10 @@ def classify_text(
 
 def _classify_text_impl(
     function: str,
-    str_input: Union[str, snowpark.Column],
-    categories: Union[list[str], snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    str_input: str | snowpark.Column,
+    categories: list[str] | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     return cast(Union[str, snowpark.Column], call_sql_function(function, session, str_input, categories))
 
 

snowflake/cortex/_complete.py

@@ -1,9 +1,8 @@
 import json
 import logging
 import time
-import typing
 from io import BytesIO
-from typing import Any, Callable, Iterator, Optional, TypedDict, Union, cast
+from typing import Any, Callable, Iterator, TypedDict, cast
 from urllib.parse import urlunparse
 
 import requests
@@ -78,9 +77,9 @@ class MidStreamException(Exception):
 
     def __init__(
         self,
-        reason: typing.Optional[str] = None,
-        http_resp: typing.Optional[Any] = None,
-        request_id: typing.Optional[str] = None,
+        reason: str | None = None,
+        http_resp: Any | None = None,
+        request_id: str | None = None,
     ) -> None:
         message = ""
         if reason is not None:
@@ -107,7 +106,7 @@ _MAX_RETRY_SECONDS = 30
 
 def retry(func: Callable[..., requests.Response]) -> Callable[..., requests.Response]:
     def inner(*args: Any, **kwargs: Any) -> requests.Response:
-        deadline = cast(Optional[float], kwargs["deadline"])
+        deadline = cast(float | None, kwargs["deadline"])
         kwargs = {key: value for key, value in kwargs.items() if key != "deadline"}
         expRetrySeconds = 0.5
         while True:
@@ -141,7 +140,7 @@ def _make_common_request_headers() -> dict[str, str]:
     return headers
 
 
-def _get_request_id(resp: dict[str, Any]) -> Optional[Any]:
+def _get_request_id(resp: dict[str, Any]) -> Any | None:
     request_id = None
     if "headers" in resp:
         for key, value in resp["headers"].items():
@@ -181,8 +180,8 @@ def _validate_response_format_object(options: CompleteOptions) -> None:
 
 def _make_request_body(
     model: str,
-    prompt: Union[str, list[ConversationMessage]],
-    options: Optional[CompleteOptions] = None,
+    prompt: str | list[ConversationMessage],
+    options: CompleteOptions | None = None,
 ) -> dict[str, Any]:
     data = {
         "model": model,
@@ -249,11 +248,11 @@ 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]]],
+    snow_api_xp_request_handler: Callable[..., dict[str, Any]] | None,
     model: str,
-    prompt: Union[str, list[ConversationMessage]],
-    options: Optional[CompleteOptions] = None,
-    deadline: Optional[float] = None,
+    prompt: str | list[ConversationMessage],
+    options: CompleteOptions | None = None,
+    deadline: float | None = None,
 ) -> requests.Response:
     headers = _make_common_request_headers()
     body = _make_request_body(model, prompt, options)
@@ -265,9 +264,9 @@ def _call_complete_xp(
 @retry
 def _call_complete_rest(
     model: str,
-    prompt: Union[str, list[ConversationMessage]],
-    options: Optional[CompleteOptions] = None,
-    session: Optional[snowpark.Session] = None,
+    prompt: str | list[ConversationMessage],
+    options: CompleteOptions | None = None,
+    session: snowpark.Session | None = None,
 ) -> requests.Response:
     session = session or context.get_active_session()
     if session is None:
@@ -305,8 +304,8 @@ def _call_complete_rest(
 
 def _return_stream_response(
     response: requests.Response,
-    deadline: Optional[float],
-    session: Optional[snowpark.Session] = None,
+    deadline: float | None,
+    session: snowpark.Session | None = None,
 ) -> Iterator[str]:
     request_id = _get_request_id(dict(response.headers))
     client = SSEClient(response)
@@ -333,18 +332,18 @@ def _return_stream_response(
 
 
 def _complete_call_sql_function_snowpark(
-    function: str, *args: Union[str, snowpark.Column, CompleteOptions]
+    function: str, *args: str | snowpark.Column | CompleteOptions
 ) -> snowpark.Column:
     return cast(snowpark.Column, functions.builtin(function)(*args))
 
 
 def _complete_non_streaming_immediate(
-    snow_api_xp_request_handler: Optional[Callable[..., dict[str, Any]]],
+    snow_api_xp_request_handler: Callable[..., dict[str, Any]] | None,
     model: str,
-    prompt: Union[str, list[ConversationMessage]],
-    options: Optional[CompleteOptions],
-    session: Optional[snowpark.Session] = None,
-    deadline: Optional[float] = None,
+    prompt: str | list[ConversationMessage],
+    options: CompleteOptions | None,
+    session: snowpark.Session | None = None,
+    deadline: float | None = None,
 ) -> str:
     response = _complete_rest(
         snow_api_xp_request_handler=snow_api_xp_request_handler,
@@ -358,14 +357,14 @@ def _complete_non_streaming_immediate(
 
 
 def _complete_non_streaming_impl(
-    snow_api_xp_request_handler: Optional[Callable[..., dict[str, Any]]],
+    snow_api_xp_request_handler: Callable[..., dict[str, Any]] | None,
     function: str,
-    model: Union[str, snowpark.Column],
-    prompt: Union[str, list[ConversationMessage], snowpark.Column],
-    options: Optional[Union[CompleteOptions, snowpark.Column]],
-    session: Optional[snowpark.Session] = None,
-    deadline: Optional[float] = None,
-) -> Union[str, snowpark.Column]:
+    model: str | snowpark.Column,
+    prompt: str | list[ConversationMessage] | snowpark.Column,
+    options: CompleteOptions | snowpark.Column | None,
+    session: snowpark.Session | None = None,
+    deadline: float | None = None,
+) -> str | snowpark.Column:
     if isinstance(prompt, snowpark.Column):
         if options is not None:
             return _complete_call_sql_function_snowpark(function, model, prompt, options)
@@ -388,12 +387,12 @@ def _complete_non_streaming_impl(
 
 
 def _complete_rest(
-    snow_api_xp_request_handler: Optional[Callable[..., dict[str, Any]]],
+    snow_api_xp_request_handler: Callable[..., dict[str, Any]] | None,
     model: str,
-    prompt: Union[str, list[ConversationMessage]],
-    options: Optional[CompleteOptions] = None,
-    session: Optional[snowpark.Session] = None,
-    deadline: Optional[float] = None,
+    prompt: str | list[ConversationMessage],
+    options: CompleteOptions | None = None,
+    session: snowpark.Session | None = None,
+    deadline: float | None = None,
 ) -> Iterator[str]:
     if options:
         _validate_response_format_object(options)
@@ -412,16 +411,16 @@ 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,
+    model: str | snowpark.Column,
+    prompt: str | list[ConversationMessage] | snowpark.Column,
+    snow_api_xp_request_handler: Callable[..., dict[str, Any]] | None = None,
     function: str = "snowflake.cortex.complete",
-    options: Optional[CompleteOptions] = None,
-    session: Optional[snowpark.Session] = None,
+    options: CompleteOptions | None = None,
+    session: snowpark.Session | None = None,
     stream: bool = False,
-    timeout: Optional[float] = None,
-    deadline: Optional[float] = None,
-) -> Union[str, Iterator[str], snowpark.Column]:
+    timeout: float | None = None,
+    deadline: float | None = None,
+) -> str | Iterator[str] | snowpark.Column:
     if timeout is not None and deadline is not None:
         raise ValueError('only one of "timeout" and "deadline" must be set')
     if timeout is not None:
@@ -454,15 +453,15 @@ def _complete_impl(
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
 def complete(
-    model: Union[str, snowpark.Column],
-    prompt: Union[str, list[ConversationMessage], snowpark.Column],
+    model: str | snowpark.Column,
+    prompt: str | list[ConversationMessage] | snowpark.Column,
     *,
-    options: Optional[CompleteOptions] = None,
-    session: Optional[snowpark.Session] = None,
+    options: CompleteOptions | None = None,
+    session: snowpark.Session | None = None,
     stream: bool = False,
-    timeout: Optional[float] = None,
-    deadline: Optional[float] = None,
-) -> Union[str, Iterator[str], snowpark.Column]:
+    timeout: float | None = None,
+    deadline: float | None = None,
+) -> str | Iterator[str] | snowpark.Column:
     """Complete calls into the LLM inference service to perform completion.
 
     Args:

snowflake/cortex/_embed_text_1024.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, cast
+from typing import Union, cast
 
 from typing_extensions import deprecated
 
@@ -11,10 +11,10 @@ from snowflake.ml._internal import telemetry
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
 def embed_text_1024(
-    model: Union[str, snowpark.Column],
-    text: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[list[float], snowpark.Column]:
+    model: str | snowpark.Column,
+    text: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> list[float] | snowpark.Column:
     """Calls into the LLM inference service to embed the text.
 
     Args:
@@ -32,10 +32,10 @@ def embed_text_1024(
 
 def _embed_text_1024_impl(
     function: str,
-    model: Union[str, snowpark.Column],
-    text: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[list[float], snowpark.Column]:
+    model: str | snowpark.Column,
+    text: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> list[float] | snowpark.Column:
     return cast(Union[list[float], snowpark.Column], call_sql_function(function, session, model, text))
 
 

snowflake/cortex/_embed_text_768.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, cast
+from typing import Union, cast
 
 from typing_extensions import deprecated
 
@@ -11,10 +11,10 @@ from snowflake.ml._internal import telemetry
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
 def embed_text_768(
-    model: Union[str, snowpark.Column],
-    text: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[list[float], snowpark.Column]:
+    model: str | snowpark.Column,
+    text: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> list[float] | snowpark.Column:
     """Calls into the LLM inference service to embed the text.
 
     Args:
@@ -32,10 +32,10 @@ def embed_text_768(
 
 def _embed_text_768_impl(
     function: str,
-    model: Union[str, snowpark.Column],
-    text: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[list[float], snowpark.Column]:
+    model: str | snowpark.Column,
+    text: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> list[float] | snowpark.Column:
     return cast(Union[list[float], snowpark.Column], call_sql_function(function, session, model, text))
 
 

snowflake/cortex/_extract_answer.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, cast
+from typing import Union, cast
 
 from typing_extensions import deprecated
 
@@ -11,10 +11,10 @@ from snowflake.ml._internal import telemetry
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
 def extract_answer(
-    from_text: Union[str, snowpark.Column],
-    question: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    from_text: str | snowpark.Column,
+    question: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     """Calls into the LLM inference service to extract an answer from within specified text.
 
     Args:
@@ -31,10 +31,10 @@ def extract_answer(
 
 def _extract_answer_impl(
     function: str,
-    from_text: Union[str, snowpark.Column],
-    question: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    from_text: str | snowpark.Column,
+    question: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     return cast(Union[str, snowpark.Column], call_sql_function(function, session, from_text, question))
 
 

snowflake/cortex/_finetune.py

@@ -1,6 +1,6 @@
 import json
 from dataclasses import dataclass
-from typing import Any, Optional, Union, cast
+from typing import Any, cast
 
 from snowflake import snowpark
 from snowflake.cortex._util import (
@@ -17,7 +17,7 @@ CORTEX_FINETUNE_DOCUMENTATION_URL = "https://docs.snowflake.com/en/user-guide/sn
 
 
 class FinetuneError(Exception):
-    def __init__(self, message: str, original_exception: Optional[Exception] = None) -> None:
+    def __init__(self, message: str, original_exception: Exception | None = None) -> None:
         """Finetuning Exception Class.
 
         Args:
@@ -41,46 +41,46 @@ class FinetuneError(Exception):
 class FinetuneStatus:
     """Fine-tuning job status."""
 
-    id: Optional[str] = None
+    id: str | None = None
     """Workflow ID for the fine-tuning run."""
 
-    status: Optional[str] = None
+    status: str | None = None
     """Status string, e.g. PENDING, RUNNING, SUCCESS, ERROR, CANCELLED."""
 
-    base_model: Optional[str] = None
+    base_model: str | None = None
     """Name of the base model that is being fine-tuned."""
 
-    created_on: Optional[int] = None
+    created_on: int | None = None
     """Creation timestamp of the Fine-tuning job in milliseconds."""
 
-    error: Optional[dict[str, Any]] = None
+    error: dict[str, Any] | None = None
     """Error message propagated from the job."""
 
-    finished_on: Optional[int] = None
+    finished_on: int | None = None
     """Completion timestamp of the Fine-tuning job in milliseconds."""
 
-    progress: Optional[float] = None
+    progress: float | None = None
     """Progress made as a fraction of total [0.0,1.0]."""
 
-    training_result: Optional[list[dict[str, Any]]] = None
+    training_result: list[dict[str, Any]] | None = None
     """Detailed metrics report for a completed training."""
 
-    trained_tokens: Optional[int] = None
+    trained_tokens: int | None = None
     """Number of tokens trained on. If multiple epochs are run, this can be larger than number of tokens in the
     training data."""
 
-    training_data: Optional[str] = None
+    training_data: str | None = None
     """Training data query."""
 
-    validation_data: Optional[str] = None
+    validation_data: str | None = None
     """Validation data query."""
 
-    model: Optional[str] = None
+    model: str | None = None
     """Location of the fine-tuned model."""
 
 
 class FinetuneJob:
-    def __init__(self, session: Optional[snowpark.Session], status: FinetuneStatus) -> None:
+    def __init__(self, session: snowpark.Session | None, status: FinetuneStatus) -> None:
         """Fine-tuning Job.
 
         Args:
@@ -145,7 +145,7 @@ class Finetune:
         project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
         subproject=CORTEX_FINETUNE_TELEMETRY_SUBPROJECT,
     )
-    def __init__(self, session: Optional[snowpark.Session] = None) -> None:
+    def __init__(self, session: snowpark.Session | None = None) -> None:
         """Cortex Fine-Tuning API.
 
         [Documentation](https://docs.snowflake.com/en/user-guide/snowflake-cortex/cortex-finetuning)
@@ -165,9 +165,9 @@ class Finetune:
         self,
         name: str,
         base_model: str,
-        training_data: Union[str, snowpark.DataFrame],
-        validation_data: Optional[Union[str, snowpark.DataFrame]] = None,
-        options: Optional[dict[str, Any]] = None,
+        training_data: str | snowpark.DataFrame,
+        validation_data: str | snowpark.DataFrame | None = None,
+        options: dict[str, Any] | None = None,
     ) -> FinetuneJob:
         """Create a new fine-tuning runs.
 
@@ -212,7 +212,7 @@ class Finetune:
         else:
             training_string = training_data
 
-        validation_string: Optional[str] = None
+        validation_string: str | None = None
         if isinstance(validation_data, snowpark.DataFrame):
             if snowpark_dataframe_utils.is_single_query_snowpark_dataframe(validation_data):
                 validation_string = str(validation_data.queries["queries"][0])
@@ -253,7 +253,7 @@ class Finetune:
         return [FinetuneJob(session=self._session, status=FinetuneStatus(**run_status)) for run_status in result]
 
 
-def _try_load_json(json_string: str) -> Union[dict[Any, Any], list[Any]]:
+def _try_load_json(json_string: str) -> dict[Any, Any] | list[Any]:
     try:
         result = json.loads(str(json_string))
     except json.JSONDecodeError as e:
@@ -269,5 +269,5 @@ def _try_load_json(json_string: str) -> Union[dict[Any, Any], list[Any]]:
     return result
 
 
-def _finetune_impl(operation: str, session: Optional[snowpark.Session], function_args: list[Any]) -> str:
+def _finetune_impl(operation: str, session: snowpark.Session | None, function_args: list[Any]) -> str:
     return call_sql_function_literals(_CORTEX_FINETUNE_SYSTEM_FUNCTION_NAME, session, operation, *function_args)

snowflake/cortex/_sentiment.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, cast
+from typing import cast
 
 from typing_extensions import deprecated
 
@@ -10,9 +10,7 @@ from snowflake.ml._internal import telemetry
 @telemetry.send_api_usage_telemetry(
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
-def sentiment(
-    text: Union[str, snowpark.Column], session: Optional[snowpark.Session] = None
-) -> Union[float, snowpark.Column]:
+def sentiment(text: str | snowpark.Column, session: snowpark.Session | None = None) -> float | snowpark.Column:
     """Calls into the LLM inference service to perform sentiment analysis on the input text.
 
     Args:
@@ -26,8 +24,8 @@ def sentiment(
 
 
 def _sentiment_impl(
-    function: str, text: Union[str, snowpark.Column], session: Optional[snowpark.Session] = None
-) -> Union[float, snowpark.Column]:
+    function: str, text: str | snowpark.Column, session: snowpark.Session | None = None
+) -> float | snowpark.Column:
 
     output = call_sql_function(function, session, text)
     if isinstance(output, snowpark.Column):

snowflake/cortex/_sse_client.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, Iterator, Optional
+from typing import Any, Iterator
 
 _FIELD_SEPARATOR = ":"
 
@@ -9,11 +9,11 @@ class Event:
 
     def __init__(
         self,
-        id: Optional[str] = None,
+        id: str | None = None,
         event: str = "message",
         data: str = "",
-        comment: Optional[str] = None,
-        retry: Optional[int] = None,
+        comment: str | None = None,
+        retry: int | None = None,
     ) -> None:
         self.id = id
         self.event = event

snowflake/cortex/_summarize.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, cast
+from typing import Union, cast
 
 from typing_extensions import deprecated
 
@@ -11,9 +11,9 @@ from snowflake.ml._internal import telemetry
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
 def summarize(
-    text: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    text: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     """Calls into the LLM inference service to summarize the input text.
 
     Args:
@@ -29,9 +29,9 @@ def summarize(
 
 def _summarize_impl(
     function: str,
-    text: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    text: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     return cast(Union[str, snowpark.Column], call_sql_function(function, session, text))
 
 

snowflake/cortex/_translate.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union, cast
+from typing import Union, cast
 
 from typing_extensions import deprecated
 
@@ -11,11 +11,11 @@ from snowflake.ml._internal import telemetry
     project=CORTEX_FUNCTIONS_TELEMETRY_PROJECT,
 )
 def translate(
-    text: Union[str, snowpark.Column],
-    from_language: Union[str, snowpark.Column],
-    to_language: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    text: str | snowpark.Column,
+    from_language: str | snowpark.Column,
+    to_language: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     """Calls into the LLM inference service to perform translation.
 
     Args:
@@ -33,11 +33,11 @@ def translate(
 
 def _translate_impl(
     function: str,
-    text: Union[str, snowpark.Column],
-    from_language: Union[str, snowpark.Column],
-    to_language: Union[str, snowpark.Column],
-    session: Optional[snowpark.Session] = None,
-) -> Union[str, snowpark.Column]:
+    text: str | snowpark.Column,
+    from_language: str | snowpark.Column,
+    to_language: str | snowpark.Column,
+    session: snowpark.Session | None = None,
+) -> str | snowpark.Column:
     return cast(Union[str, snowpark.Column], call_sql_function(function, session, text, from_language, to_language))
 
 

snowflake/cortex/_util.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional, Union, cast
+from typing import Any, Union, cast
 
 from snowflake import snowpark
 from snowflake.ml._internal.exceptions import error_codes, exceptions
@@ -20,9 +20,9 @@ class SnowflakeConfigurationException(Exception):
 # (e.g. snowpark column and literal type modes).
 def call_sql_function(
     function: str,
-    session: Optional[snowpark.Session],
-    *args: Union[str, list[str], snowpark.Column, dict[str, Union[int, float]]],
-) -> Union[str, list[float], snowpark.Column]:
+    session: snowpark.Session | None,
+    *args: str | list[str] | snowpark.Column | dict[str, int | float],
+) -> str | list[float] | snowpark.Column:
     handle_as_column = False
 
     for arg in args:
@@ -38,16 +38,16 @@ def call_sql_function(
 
 
 def _call_sql_function_column(
-    function: str, *args: Union[str, list[str], snowpark.Column, dict[str, Union[int, float]]]
+    function: str, *args: str | list[str] | snowpark.Column | dict[str, int | float]
 ) -> snowpark.Column:
     return cast(snowpark.Column, functions.builtin(function)(*args))
 
 
 def _call_sql_function_immediate(
     function: str,
-    session: Optional[snowpark.Session],
-    *args: Union[str, list[str], snowpark.Column, dict[str, Union[int, float]]],
-) -> Union[str, list[float]]:
+    session: snowpark.Session | None,
+    *args: str | list[str] | snowpark.Column | dict[str, int | float],
+) -> str | list[float]:
     session = session or context.get_active_session()
     if session is None:
         raise SnowflakeAuthenticationException(
@@ -64,7 +64,7 @@ def _call_sql_function_immediate(
     return cast(str, df.collect()[0][0])
 
 
-def call_sql_function_literals(function: str, session: Optional[snowpark.Session], *args: Any) -> str:
+def call_sql_function_literals(function: str, session: snowpark.Session | None, *args: Any) -> str:
     r"""Call a SQL function with only literal arguments.
 
     This is useful for calling system functions.