github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: gaoyuan <gaoyuan.20001218@bytedance.com>
This commit is contained in:
高远 2024-04-30 08:06:00 +08:00 committed by GitHub
parent c172611647
commit a7a4630bf4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
libs/community/langchain_community/vectorstores/vikingdb.py
View File

@ -115,19 +115,27 @@ class VikingDB(VectorStore):
# print(key, value) # print(key, value)
if isinstance(value, str): if isinstance(value, str):
fields.append(Field(key, FieldType.String)) fields.append(Field(key, FieldType.String))
if isinstance(value, int): elif isinstance(value, int):
fields.append(Field(key, FieldType.Int64)) fields.append(Field(key, FieldType.Int64))
if isinstance(value, bool): elif isinstance(value, bool):
fields.append(Field(key, FieldType.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 isinstance(item, str) for item in value
): ):
fields.append(Field(key, FieldType.List_String)) 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 isinstance(item, int) for item in value
): ):
fields.append(Field(key, FieldType.List_Int64)) 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)) fields.append(Field("primary_key", FieldType.String, is_primary_key=True))