Snowflake脚本块赋值VECTOR(FLOAT,768)向量嵌入报错求助

报错原因 #

Snowflake的SQL脚本块（Script Block）变量系统暂不支持VECTOR类型作为变量数据类型，尽管普通SQL语句可以正常处理VECTOR，但脚本块的变量类型体系尚未适配该类型，因此直接将嵌入函数返回的向量赋值给变量会触发Unsupported datatype 'VECTOR(FLOAT, 768)'编译错误。

可行解决方案 #

提供两种实用方案满足不同复用需求：

方案1：直接在计算中调用嵌入函数（适合低复用场景） #

如果向量仅需使用1-2次，可跳过变量赋值，直接在相似度计算逻辑中调用EMBED_TEXT_768函数：

begin
    -- 直接调用嵌入函数完成余弦相似度计算
    select cosine_similarity(
        SNOWFLAKE.CORTEX.EMBED_TEXT_768('snowflake-arctic-embed-m-v1.5', 'hello world'),
        target_vector_column
    ) from your_vector_table;

    return TRUE;
end;

方案2：用临时表存储向量实现复用（适合高复用场景） #

若需要多次复用向量结果，可将向量存入临时表，后续通过查询临时表获取向量：

begin
    -- 创建临时表存储生成的向量
    create or replace temporary table temp_query_vector as
    select SNOWFLAKE.CORTEX.EMBED_TEXT_768('snowflake-arctic-embed-m-v1.5', 'hello world') as query_vec;

    -- 第一次复用向量执行相似度搜索
    select cosine_similarity(
        (select query_vec from temp_query_vector),
        target_vector_column
    ) from your_vector_table;

    -- 第二次复用向量筛选高相似度结果（示例）
    select * from your_vector_table
    where cosine_similarity(
        (select query_vec from temp_query_vector),
        target_vector_column
    ) > 0.8;

    return TRUE;
end;

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