community[patch]: Modify the text field type and add new exception handling (#20116)

Co-authored-by: gaoyuan <gaoyuan.20001218@bytedance.com>
pull/20114/head
高远 2 months ago committed by GitHub
parent c172611647
commit a7a4630bf4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -115,19 +115,27 @@ class VikingDB(VectorStore):
# print(key, value)
if isinstance(value, str):
fields.append(Field(key, FieldType.String))
if isinstance(value, int):
elif isinstance(value, int):
fields.append(Field(key, FieldType.Int64))
if isinstance(value, bool):
elif isinstance(value, bool):
fields.append(Field(key, FieldType.Bool))
if isinstance(value, list) and all(
elif isinstance(value, list) and all(
isinstance(item, str) for item in value
):
fields.append(Field(key, FieldType.List_String))
if isinstance(value, list) and all(
elif isinstance(value, list) and all(
isinstance(item, int) for item in value
):
fields.append(Field(key, FieldType.List_Int64))
fields.append(Field("text", FieldType.String))
elif isinstance(value, bytes):
fields.append(Field(key, FieldType.Text))
else:
raise ValueError(
"metadatas value is invalid"
"please change the type of metadatas."
)
# fields.append(Field("text", FieldType.String))
fields.append(Field("text", FieldType.Text))
fields.append(Field("primary_key", FieldType.String, is_primary_key=True))

Loading…
Cancel
Save