d

Author: sfc-gh-jdu

src/snowflake/snowpark/functions.py

@@ -13871,3 +13871,65 @@ def ai_sentiment(
         return _call_function(
             sql_func_name, text_col, cat_col, _ast=ast, _emit_ast=_emit_ast
         )
+
+
+@publicapi
+def ai_translate(
+    text: ColumnOrLiteralStr,
+    source_language: ColumnOrLiteralStr,
+    target_language: ColumnOrLiteralStr,
+    _emit_ast: bool = True,
+) -> Column:
+    """
+    Translates the given input text from one supported language to another.
+
+    Args:
+        text: A string or Column containing the text to be translated.
+        source_language: A string or Column specifying the language code for the source language.
+            Specify an empty string ``''`` to automatically detect the source language.
+        target_language: A string or Column specifying the language code for the target language.
+
+    Returns:
+        A string containing a translation of the original text into the target language.
+
+    See details in `AI_TRANSLATE <https://docs.snowflake.com/en/sql-reference/functions/ai_translate>`_.
+
+    Examples::
+
+        >>> # Translate literal text from English to German
+        >>> df = session.range(1).select(
+        ...     ai_translate('Hello world', 'en', 'de').alias('translation')
+        ... )
+        >>> df.collect()[0][0].lower()
+        'hallo welt'
+
+        >>> # Auto-detect source language and translate to English
+        >>> df = session.range(1).select(
+        ...     ai_translate('Hola mundo', '', 'en').alias('translation')
+        ... )
+        >>> df.collect()[0][0].lower()
+        'hi world'
+    """
+    sql_func_name = "ai_translate"
+
+    # Build AST
+    ast = (
+        build_function_expr(sql_func_name, [text, source_language, target_language])
+        if _emit_ast
+        else None
+    )
+
+    # Convert arguments to columns
+    text_col = _to_col_if_lit(text, sql_func_name)
+    source_lang_col = _to_col_if_lit(source_language, sql_func_name)
+    target_lang_col = _to_col_if_lit(target_language, sql_func_name)
+
+    # Call the function
+    return _call_function(
+        sql_func_name,
+        text_col,
+        source_lang_col,
+        target_lang_col,
+        _ast=ast,
+        _emit_ast=_emit_ast,
+    )