Snowflake Native App中Python UDF返回数据异常与格式调整问题

一、返回Variant仅得单个元素的原因 #

出现这种情况，核心原因是UDF逻辑里错误返回了列表的单个元素而非完整列表：

• 可能是循环处理时提前执行return，只返回了第一个元素的结果；
• 或是解析JSON数组后，误取了数组的第一个元素（比如ids[0]）进行处理，而非遍历整个数组。
  另外，Snowflake Python UDF返回Variant时，需确保返回的是完整Python列表（会自动映射为Snowflake Variant数组），如果返回单个字符串，就会出现['John']这种单元素数组。

二、返回String格式不符合需求的解决思路 #

你需要的是每行一个元素的输出，本质是要将UDF的输出拆分为多行结果。普通UDF是单值返回，无法直接实现，必须用**表值函数（Table Function）**来生成多行数据，而非返回JSON格式的字符串。

三、正确实现代码示例 #

1. 修复Variant返回完整数组的问题 #

如果只是需要返回完整的Variant数组，确保代码返回完整的结果列表：

import requests
import json

def get_names(json_array_str):
    ids = json.loads(json_array_str)
    names = []
    for id_val in ids:
        # 替换为实际的API请求逻辑
        resp = requests.get(f"https://your-api-url/{id_val}")
        names.append(resp.json().get("name"))
    # 返回完整列表，Snowflake自动转为Variant数组
    return names

创建UDF：

CREATE OR REPLACE FUNCTION GET_NAMES(json_array_str STRING)
RETURNS VARIANT
LANGUAGE PYTHON
RUNTIME_VERSION = '3.8'
HANDLER = 'get_names'
EXTERNAL_ACCESS_INTEGRATIONS = (your_external_access_integration)
PACKAGES = ('requests');

2. 实现每行一个元素的输出（表值函数） #

要生成每行一个元素的结果，必须创建表值函数，通过yield逐个返回结果：

import requests
import json

def get_names_table(json_array_str):
    ids = json.loads(json_array_str)
    for id_val in ids:
        resp = requests.get(f"https://your-api-url/{id_val}")
        name = resp.json().get("name")
        # 用yield返回单个元组，对应一行数据
        yield (name,)

创建表值函数：

CREATE OR REPLACE FUNCTION GET_NAMES_TABLE(json_array_str STRING)
RETURNS TABLE(name STRING)
LANGUAGE PYTHON
RUNTIME_VERSION = '3.8'
HANDLER = 'get_names_table'
EXTERNAL_ACCESS_INTEGRATIONS = (your_external_access_integration)
PACKAGES = ('requests');

调用方式：

SELECT name FROM TABLE(GET_NAMES_TABLE('[1,2,3,4]'));

执行后得到的结果格式如下：

NAME
----
John
Tom
Jimmy
Harry

四、关键注意事项 #

• 必须配置EXTERNAL_ACCESS_INTEGRATIONS，授权UDF访问外部API；
• 若API请求耗时较长，建议批量请求优化性能，避免循环内逐个请求拖慢效率；
• 表值函数的handler必须通过yield返回元组，每个元组对应一行输出。

内容的提问来源于stack exchange，提问作者Navidk