fix(query): Fix cast array to vector failed (#18375)

* fix(query): Fix cast array to vector failed

* add tests

Author: b41sh

src/query/expression/src/evaluator.rs

@@ -941,6 +941,7 @@ impl<'a> Evaluator<'a> {
                             .set_span(span));
                         }
                         let mut vals = Vec::with_capacity(dimension);
+                        let col = col.remove_nullable();
                         match col {
                             Column::Number(num_col) => {
                                 for i in 0..dimension {

tests/sqllogictests/suites/udf_server/udf_server_test.test

@@ -578,6 +578,36 @@ EvalScalar
 ├── estimated rows: 1.00
 └── DummyTableScan
 
+
+statement ok
+CREATE OR REPLACE FUNCTION embedding_4 (VARCHAR) RETURNS ARRAY(FLOAT NULL) LANGUAGE python IMMUTABLE HANDLER = 'embedding_4' ADDRESS = 'http://0.0.0.0:8815';
+
+query T
+SELECT embedding_4('databend.com')::vector(4);
+----
+[1.1,1.2,1.3,1.4]
+
+statement ok
+CREATE OR REPLACE TABLE test(url STRING, length INT64);
+
+statement ok
+INSERT INTO test (url) VALUES('databend.com'),('databend.cn');
+
+query T
+SELECT embedding_4('databend.com')::vector(4) fro
+----
+[1.1,1.2,1.3,1.4]
+
+query T
+SELECT embedding_4(url)::vector(4) FROM test;
+----
+[1.1,1.2,1.3,1.4]
+[1.1,1.2,1.3,1.4]
+
+statement ok
+drop FUNCTION embedding_4;
+
+
 statement ok
 remove @udf_stage;
 

tests/udf/udf_server.py

@@ -414,6 +414,12 @@ def ping(s: str) -> str:
 def check_headers() -> str:
     return "success"
 
+@udf(
+    input_types=["VARCHAR"],
+    result_type="ARRAY(FLOAT32 NULL)"
+)
+def embedding_4(s: str):
+    return [1.1, 1.2, 1.3, 1.4]
 
 if __name__ == "__main__":
     udf_server = CheckHeadersServer(
@@ -444,7 +450,9 @@ def check_headers() -> str:
     udf_server.add_function(wait_concurrent)
     udf_server.add_function(url_len)
     udf_server.add_function(check_headers)
+    udf_server.add_function(embedding_4)
 
     # Built-in function
     udf_server.add_function(ping)
     udf_server.serve()
+