jack 1 месяц назад
Родитель
Сommit
d1eea098a6
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      app/models/message.py

+ 2 - 2
app/models/message.py

@@ -12,7 +12,7 @@ class MessageBase(BaseModel):
     role: str = Field(sa_column=Column(Enum("assistant", "user", "system", "function", "tool"), nullable=False))
     thread_id: str = Field(nullable=False)
     object: str = Field(nullable=False, default="thread.message")
-    content: Optional[list] = Field(default=None, sa_column=Column(JSON))
+    content:  Union[str, List[dict]] = Field(default=None, sa_column=Column(JSON))
     file_ids: Optional[list] = Field(default=None, sa_column=Column(JSON))
     attachments: Optional[list] = Field(default=None, sa_column=Column(JSON))  # 附件
     metadata_: Optional[dict] = Field(default=None, sa_column=Column("metadata", JSON), schema_extra={"validation_alias": "metadata"})
@@ -33,7 +33,7 @@ class MessageCreate(BaseModel):
 
 
 class MessageUpdate(BaseModel):
-    content: Optional[str] = Field(default=None)
+    content:  Union[str, List[dict]] = Field(default=None)
     metadata_: Optional[dict] = Field(default=None, schema_extra={"validation_alias": "metadata"})