Skip to content

Commit f01620e

Browse files
authored
fix: error when function or function score fetch empty params (#3119)
Signed-off-by: aoiasd <zhicheng.yue@zilliz.com>
1 parent df125d2 commit f01620e

File tree

1 file changed

+16
-8
lines changed

1 file changed

+16
-8
lines changed

pymilvus/client/prepare.py

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1634,20 +1634,28 @@ def function_score_schema(function_score: FunctionScore) -> schema_types.Functio
16341634
type=ranker.type,
16351635
description=ranker.description,
16361636
input_field_names=ranker.input_field_names,
1637-
params=[
1638-
common_types.KeyValuePair(key=str(k), value=Prepare.common_kv_value(v))
1639-
for k, v in ranker.params.items()
1640-
],
1637+
params=(
1638+
[
1639+
common_types.KeyValuePair(key=str(k), value=Prepare.common_kv_value(v))
1640+
for k, v in ranker.params.items()
1641+
]
1642+
if ranker.params
1643+
else []
1644+
),
16411645
)
16421646
for ranker in function_score.functions
16431647
]
16441648

16451649
return schema_types.FunctionScore(
16461650
functions=functions,
1647-
params=[
1648-
common_types.KeyValuePair(key=str(k), value=Prepare.common_kv_value(v))
1649-
for k, v in function_score.params.items()
1650-
],
1651+
params=(
1652+
[
1653+
common_types.KeyValuePair(key=str(k), value=Prepare.common_kv_value(v))
1654+
for k, v in function_score.params.items()
1655+
]
1656+
if function_score.params
1657+
else []
1658+
),
16511659
)
16521660

16531661
@staticmethod

0 commit comments

Comments
 (0)