first commit
This commit is contained in:
@@ -0,0 +1,20 @@
|
||||
from marshmallow import Schema, fields
|
||||
|
||||
|
||||
class MessageSchema(Schema):
|
||||
id = fields.Integer(dump_only=True)
|
||||
conversation_id = fields.Integer()
|
||||
role = fields.String(required=True)
|
||||
content = fields.String(required=True)
|
||||
tokens = fields.Integer(allow_none=True)
|
||||
model = fields.String(allow_none=True)
|
||||
tool_calls = fields.List(fields.Dict(), allow_none=True)
|
||||
metadata = fields.Dict(allow_none=True)
|
||||
created_at = fields.DateTime(dump_only=True)
|
||||
|
||||
|
||||
class MessageCreateSchema(Schema):
|
||||
role = fields.String(required=True)
|
||||
content = fields.String(required=True)
|
||||
tool_calls = fields.List(fields.Dict(), allow_none=True)
|
||||
metadata = fields.Dict(allow_none=True)
|
||||
Reference in New Issue
Block a user