feat: 重构图片消息传递逻辑 (#957, #955)

2026-07-19 10:56:06 +00:00 · 2024-12-24 10:57:17 +08:00
parent 535c4a8a11
commit 12cfce3622
9 changed files with 70 additions and 54 deletions
@@ -38,6 +38,8 @@ class ContentElement(pydantic.BaseModel):

    image_url: typing.Optional[ImageURLContentObject] = None

+    image_base64: typing.Optional[str] = None
+
    def __str__(self):
        if self.type == 'text':
            return self.text
@@ -53,6 +55,10 @@ class ContentElement(pydantic.BaseModel):
    @classmethod
    def from_image_url(cls, image_url: str):
        return cls(type='image_url', image_url=ImageURLContentObject(url=image_url))
+    
+    @classmethod
+    def from_image_base64(cls, image_base64: str):
+        return cls(type='image_base64', image_base64=image_base64)


 class Message(pydantic.BaseModel):