Kaynağa Gözat

Chore: PromptMessage is not an abstract base class (#15965)

Yongtao Huang 1 ay önce
ebeveyn
işleme
963b6f628a

+ 1 - 1
api/core/model_runtime/docs/en_US/interfaces.md

@@ -493,7 +493,7 @@ If inputting a combination of text and images, the images need to be constructed
 The base class for all Role message bodies, used only for parameter declaration and cannot be initialized.
 The base class for all Role message bodies, used only for parameter declaration and cannot be initialized.
 
 
 ```python
 ```python
-class PromptMessage(ABC, BaseModel):
+class PromptMessage(BaseModel):
     """
     """
     Model class for prompt message.
     Model class for prompt message.
     """
     """

+ 1 - 1
api/core/model_runtime/docs/zh_Hans/interfaces.md

@@ -533,7 +533,7 @@ class ImagePromptMessageContent(PromptMessageContent):
 所有 Role 消息体的基类,仅作为参数声明用,不可初始化。
 所有 Role 消息体的基类,仅作为参数声明用,不可初始化。
 
 
 ```python
 ```python
-class PromptMessage(ABC, BaseModel):
+class PromptMessage(BaseModel):
     """
     """
     Model class for prompt message.
     Model class for prompt message.
     """
     """

+ 0 - 2
api/core/model_runtime/entities/__init__.py

@@ -31,11 +31,9 @@ __all__ = [
     "ModelPropertyKey",
     "ModelPropertyKey",
     "MultiModalPromptMessageContent",
     "MultiModalPromptMessageContent",
     "PromptMessage",
     "PromptMessage",
-    "PromptMessage",
     "PromptMessageContent",
     "PromptMessageContent",
     "PromptMessageContentType",
     "PromptMessageContentType",
     "PromptMessageRole",
     "PromptMessageRole",
-    "PromptMessageRole",
     "PromptMessageTool",
     "PromptMessageTool",
     "SystemPromptMessage",
     "SystemPromptMessage",
     "TextPromptMessageContent",
     "TextPromptMessageContent",

+ 1 - 2
api/core/model_runtime/entities/message_entities.py

@@ -1,4 +1,3 @@
-from abc import ABC
 from collections.abc import Sequence
 from collections.abc import Sequence
 from enum import Enum, StrEnum
 from enum import Enum, StrEnum
 from typing import Optional
 from typing import Optional
@@ -119,7 +118,7 @@ class DocumentPromptMessageContent(MultiModalPromptMessageContent):
     type: PromptMessageContentType = PromptMessageContentType.DOCUMENT
     type: PromptMessageContentType = PromptMessageContentType.DOCUMENT
 
 
 
 
-class PromptMessage(ABC, BaseModel):
+class PromptMessage(BaseModel):
     """
     """
     Model class for prompt message.
     Model class for prompt message.
     """
     """