diff --git a/appbuilder/core/assistant/type/__init__.py b/appbuilder/core/assistant/type/__init__.py index 133ff4c9b..b9141c83d 100644 --- 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 index 5866e66c0..fecc77b75 100644 --- 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 index ac02d4dca..97814779f 100644 --- 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 index 7d67c2ea7..86852a1c9 100644 --- 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)