删除ERNIE-Bot-8K模型名 (#238)

* 删除ERNIE-Bot-8K模型名 * 删除使用ERNIE-Bot-8K模型的代码 * 修改模型名字
baidubce · Apr 12, 2024 · e3b651d · e3b651d
1 parent e4f1597
commit e3b651d
Show file tree

Hide file tree

Showing 10 changed files with 8 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -90,7 +90,6 @@ ERNIE-Bot 4.0, ERNIE-Bot, ERNIE-3.5-4K-0205, ERNIE-3.5-8K-0205, ERNIE-3.5-8K-122
 | 千帆模型名                   | AppBuilder-SDK短名 |
 |----------------------------|------------------|
 | ERNIE-Bot 4.0              |       eb-4       |
-| ERNIE-Bot-8K               |       eb-8k      |
 | ERNIE-Bot                  |       eb         |
 | ERNIE-Bot-turbo            |       eb-turbo   |
 | EB-turbo-AppBuilder专用版   |       ernie_speed_appbuilder           |

diff --git a/appbuilder/core/components/gbi/basic.py b/appbuilder/core/components/gbi/basic.py
@@ -50,5 +50,5 @@ class ColumnItem(BaseModel):
 
 
 SUPPORTED_MODEL_NAME = {
-    "ERNIE-Bot 4.0", "ERNIE-Bot-8K", "ERNIE-Bot", "ERNIE-Bot-turbo"
+    "ERNIE-Bot 4.0", "ERNIE-Bot", "ERNIE-Bot-turbo"
 }
diff --git a/appbuilder/core/components/gbi/nl2sql/README.md b/appbuilder/core/components/gbi/nl2sql/README.md
@@ -59,7 +59,7 @@ print(f"llm result: {nl2sql_result_message.content.llm_result}")
 ## 参数说明
 
 ### 初始化参数
-- model_name:  支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot-8K, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
+- model_name:  支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
 - table_schemas: 表的 schema，例如:
 
 ```

diff --git a/appbuilder/core/components/gbi/nl2sql/component.py b/appbuilder/core/components/gbi/nl2sql/component.py
@@ -45,7 +45,7 @@ def __init__(self, model_name: str, table_schemas: List[str], knowledge: Dict =
         """
         创建 gbi nl2sql 对象
         Args:
-            model_name:  支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot-8K, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
+            model_name:  支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
             table_schemas: 表的 schema 列表，例如: ```
                             CREATE TABLE `mytable` (
                             `d_year` COMMENT '年度,2019,2020..2022..',

diff --git a/appbuilder/core/components/gbi/select_table/README.md b/appbuilder/core/components/gbi/select_table/README.md
@@ -51,7 +51,7 @@ print(f"选的表是: {select_table_result_message.content}")
 
 ## 参数说明
 ### 初始化参数
-- model_name: 支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot-8K, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
+- model_name: 支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
 - table_descriptions: 表的描述是个字典，key: 是表的名字, value: 是表的描述，例如:
 
 ```

diff --git a/appbuilder/core/components/gbi/select_table/component.py b/appbuilder/core/components/gbi/select_table/component.py
@@ -41,7 +41,7 @@ def __init__(self, model_name: str, table_descriptions: Dict[str, str],
         """
         创建 GBI 选表对象
         Args:
-            model_name: 支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot-8K, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
+            model_name: 支持的模型名字 ERNIE-Bot 4.0, ERNIE-Bot, ERNIE-Bot-turbo, ERNIE Speed-AppBuilder
             table_descriptions: 表的描述是个字典，key: 是表的名字, value: 是表的描述，例如:
                                 {
                                     "超市营收明细表": "超市营收明细表，包含超市各种信息等",

diff --git a/appbuilder/tests/test_gbi_nl2sql.py b/appbuilder/tests/test_gbi_nl2sql.py
@@ -76,7 +76,7 @@ def setUp(self):
         """
         设置环境变量及必要数据。
         """
-        model_name = "ERNIE-Bot-8K"
+        model_name = "ERNIE-Bot 4.0"
         table_schemas = [SUPER_MARKET_SCHEMA]
         self.nl2sql_node = appbuilder.NL2Sql(model_name=model_name,
                                              table_schemas=table_schemas)

diff --git a/appbuilder/tests/test_gbi_select_table.py b/appbuilder/tests/test_gbi_select_table.py
@@ -75,7 +75,7 @@ def setUp(self):
         """
         设置环境变量及必要数据。
         """
-        model_name = "ERNIE-Bot-8K"
+        model_name = "ERNIE-Bot 4.0"
 
         self.select_table_node = \
             appbuilder.SelectTable(model_name=model_name,

diff --git a/appbuilder/utils/model_util.py b/appbuilder/utils/model_util.py
@@ -25,7 +25,6 @@
 # 实际上，模型名称和简称之间存在多对一的关系，因此这里不能仅使用一个字典来存储名称映射信息
 model_name_mapping = [
     ("ERNIE-Bot 4.0", "eb-4"),
-    ("ERNIE-Bot-8K", "eb-8k"),
     ("ERNIE-Bot", "eb"),
     ("ERNIE-Bot-turbo", "eb-turbo"),
     ("EB-turbo-AppBuilder专用版", "eb-turbo-appbuilder"),

diff --git a/docs/get_model_list.md b/docs/get_model_list.md
@@ -38,7 +38,7 @@ os.environ["APPBUILDER_TOKEN"] = "bce-YOURTOKEN"
 
 
 ### 返回示例
-['ERNIE-Bot 4.0', 'ERNIE-Bot-8K', 'ERNIE-Bot', 'ERNIE-Bot-turbo']
+['ERNIE-Bot 4.0', 'ERNIE-Bot', 'ERNIE-Bot-turbo']
 
 
 ## 高级用法