Skip to content

Commit dde8d62

Browse files
committed
Update databend-udf to 0.2.11 and use VECTOR type
1 parent 600d217 commit dde8d62

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

databend_aiserver/udfs/embeddings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def _resolve_model(model: str) -> Tuple[str, int]:
159159

160160
@udf(
161161
input_types=["STRING"],
162-
result_type="ARRAY(NULLABLE(FLOAT))",
162+
result_type="VECTOR(1024)",
163163
name="ai_embed_1024",
164164
io_threads=4,
165165
batch_mode=True,

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ readme = "README.md"
1010
requires-python = ">=3.10"
1111
authors = [{ name = "Databend Labs" }]
1212
dependencies = [
13-
"databend-udf==0.2.9",
13+
"databend-udf==0.2.11",
1414
"opendal==0.45.1",
1515
"pypdf==4.0.0",
1616
"python-docx==1.2.0",

0 commit comments

Comments
 (0)