Update for vdb readme (#205)

* update for vdb readme * update check
baidubce · Mar 20, 2024 · 5ad6fb0 · 5ad6fb0
1 parent 495bba4
commit 5ad6fb0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/appbuilder/core/components/retriever/baidu_vdb/README.md b/appbuilder/core/components/retriever/baidu_vdb/README.md
@@ -82,7 +82,7 @@ os.environ["APPBUILDER_TOKEN"] = "bce-YOURTOKEN"
 
 | 参数名称    | 参数类型   |是否必须 | 描述               | 示例值           |
 |---------|--------|--------|------------------|---------------|
-| message | String |是 | 需要检索的内容, 类型为Message，content类型为str, 长度要求(0,1000)          | "中国2023人均GDP" |
+| message | String |是 | 需要检索的内容, 类型为Message，content类型为str, 长度要求(0,512)          | "中国2023人均GDP" |
 | top_k   | int    |否 | 返回相似度最高的top_k个内容,top_k的数值范围(1,embedding索引数量] | 1             |
 
 

diff --git a/appbuilder/core/components/retriever/baidu_vdb/baiduvdb_retriever.py b/appbuilder/core/components/retriever/baidu_vdb/baiduvdb_retriever.py
@@ -472,8 +472,8 @@ def run(self, query: Message, top_k: int = 1):
                              .format(type(content)))
         if len(content) == 0:
             raise ValueError("Parameter `query` content is empty")
-        if len(content) > 1000:
-            raise ValueError("Parameter `query` content is too long, max length per batch size is 1000")
+        if len(content) > 512:
+            raise ValueError("Parameter `query` content is too long, max length per batch size is 512")
 
         query_embedding = self.embedding(query)
         anns = AnnSearch(