删除了add_file_content

This commit is contained in:
Windpicker-owo
2025-08-01 12:49:09 +08:00
parent 28ea2e7a0c
commit fcaa78f9a5
4 changed files with 14 additions and 30 deletions

View File

@@ -1,6 +1,5 @@
import base64
from enum import Enum
from io import BytesIO
# 设计这系列类的目的是为未来可能的扩展做准备
@@ -34,7 +33,7 @@ class Message:
class MessageBuilder:
def __init__(self):
self.__role: RoleType = RoleType.User
self.__content: list[tuple[str, str] | str | tuple[str, BytesIO]] = []
self.__content: list[tuple[str, str] | str] = []
self.__tool_call_id: str | None = None
def set_role(self, role: RoleType = RoleType.User) -> "MessageBuilder":
@@ -54,20 +53,6 @@ class MessageBuilder:
"""
self.__content.append(text)
return self
def add_file_content(
self, file_name: str, file_base64: str
) -> "MessageBuilder":
"""
添加文件内容
:param file_name: 文件名(包含类型后缀)
:param file_base64: 文件的base64编码
:return: MessageBuilder对象
"""
if not file_name or not file_base64:
raise ValueError("文件名和base64编码不能为空")
self.__content.append((file_name, BytesIO(base64.b64decode(file_base64))))
return self
def add_image_content(
self, image_format: str, image_base64: str