update assistant model setting (baidubce#336)

hermitgreen · Jun 3, 2024 · ed25ca9 · ed25ca9
1 parent 751af79
commit ed25ca9
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 16 deletions.
diff --git a/appbuilder/core/assistant/type/__init__.py b/appbuilder/core/assistant/type/__init__.py
@@ -19,7 +19,6 @@
     AssistantFunction,
     AssistantTool,
     ResponseFormat,
-    AssistantModel,
     AssistantText,
     AssistantContent
 )

diff --git a/appbuilder/core/assistant/type/assistant_type.py b/appbuilder/core/assistant/type/assistant_type.py
@@ -18,8 +18,7 @@
 from enum import Enum
 from appbuilder.core.assistant.type import (
     AssistantTool,
-    ResponseFormat,
-    AssistantModel
+    ResponseFormat
 )
 
 # AssistantFilesCreateResponse类，用于描述创建助理文件后的响应信息
@@ -77,7 +76,7 @@ class AssistantFilesDeleteResponse(BaseModel):
 
 # AssistantCreateRequest类，用于描述创建助理的请求参数
 class AssistantCreateRequest(BaseModel):
-    model: AssistantModel = Field(default="ERNIE-4.0-8K")  # 使用的模型
+    model: str = Field(default="ERNIE-4.0-8K")  # 使用的模型
     name: str = Field(default="", min_length=1, max_length=64, pattern="^[0-9a-zA-Z_-]+$")  # 助理名称
     description: str = Field(default="", max_length=512)  # 助理描述
     response_format: ResponseFormat = Field(default=ResponseFormat.TEXT)  # 响应格式
@@ -106,7 +105,7 @@ class AssistantCreateResponse(BaseModel):
 
 class AssistantUpdateRequest(BaseModel):
     assistant_id: Optional[str] = ""  # 助理ID
-    model: AssistantModel = Field(default="ERNIE-4.0-8K")  # 使用的模型
+    model: str = Field(default="ERNIE-4.0-8K")  # 使用的模型
     name: str = Field(default="", min_length=1, max_length=64, pattern="^[0-9a-zA-Z_-]+$")  # 助理名称
     description: str = Field(default="", max_length=512)  # 助理描述
     response_format: ResponseFormat = Field(default=ResponseFormat.TEXT)  # 响应格式

diff --git a/appbuilder/core/assistant/type/public_type.py b/appbuilder/core/assistant/type/public_type.py
@@ -99,15 +99,6 @@ class ResponseFormat(str, Enum):
     JSON_OBJECT = 'json_object'
 
 
-class AssistantModel(str, Enum):
-    """
-    表示助理模型的枚举类型。
-
-    Values:
-        ERNIE_408K: ERNIE-4.0-8K模型。
-    """
-    ERNIE_408K = 'ERNIE-4.0-8K'
-
 class AssistantText(BaseModel):
     """
     表示助理文本内容的模型。

diff --git a/appbuilder/core/assistant/type/thread_type.py b/appbuilder/core/assistant/type/thread_type.py
@@ -20,7 +20,6 @@
 from appbuilder.core.assistant.type import (
     AssistantTool,
     AssistantContent,
-    AssistantModel,
     ResponseFormat
 )
 
@@ -329,7 +328,7 @@ class ToolOutput(BaseModel):
 
 class AssistantRunRequest(BaseModel):
     thread_id: Optional[str] = Field(default="")
-    model: AssistantModel = Field(default="ERNIE-4.0-8K")
+    model: str = Field(default="ERNIE-4.0-8K")
     assistant_id: Optional[str] = Field(default="")
     metadata: Optional[dict] =  Field(default={}, max_length=16)
     response_format: ResponseFormat = Field(default=ResponseFormat.TEXT)