Parcourir la source

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

Yongtao Huang il y a 1 mois
Parent
commit
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.
 
 ```python
-class PromptMessage(ABC, BaseModel):
+class PromptMessage(BaseModel):
     """
     Model class for prompt message.
     """

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

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

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

@@ -31,11 +31,9 @@ __all__ = [
     "ModelPropertyKey",
     "MultiModalPromptMessageContent",
     "PromptMessage",
-    "PromptMessage",
     "PromptMessageContent",
     "PromptMessageContentType",
     "PromptMessageRole",
-    "PromptMessageRole",
     "PromptMessageTool",
     "SystemPromptMessage",
     "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 enum import Enum, StrEnum
 from typing import Optional
@@ -119,7 +118,7 @@ class DocumentPromptMessageContent(MultiModalPromptMessageContent):
     type: PromptMessageContentType = PromptMessageContentType.DOCUMENT
 
 
-class PromptMessage(ABC, BaseModel):
+class PromptMessage(BaseModel):
     """
     Model class for prompt message.
     """