fix potential bad request issue

This commit is contained in:
Aiden Dai
2025-03-13 07:16:42 +08:00
parent 5ff18c0acd
commit 33e8fcfd3b

View File

@@ -1,8 +1,10 @@
import time
from typing import Literal, Iterable
from api.setting import DEFAULT_MODEL
from pydantic import BaseModel, Field
from api.setting import DEFAULT_MODEL
class Model(BaseModel):
id: str
@@ -58,7 +60,7 @@ class UserMessage(BaseModel):
class AssistantMessage(BaseModel):
name: str | None = None
role: Literal["assistant"] = "assistant"
content: str | list[TextContent | ImageContent] | None
content: str | list[TextContent | ImageContent] | None = None
tool_calls: list[ToolCall] | None = None