AVB/bedrock-access-gateway
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix potential bad request issue

Browse Source
This commit is contained in:
Aiden Dai
2025-03-13 07:16:42 +08:00
parent 5ff18c0acd
commit 33e8fcfd3b
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/api/schema.py
+4 -2
View File
@@ -1,8 +1,10 @@
import time import time
from typing import Literal, Iterable from typing import Literal, Iterable
from api.setting import DEFAULT_MODEL
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from api.setting import DEFAULT_MODEL
class Model(BaseModel): class Model(BaseModel):
id: str id: str
@@ -58,7 +60,7 @@ class UserMessage(BaseModel):
class AssistantMessage(BaseModel): class AssistantMessage(BaseModel):
name: str | None = None name: str | None = None
role: Literal["assistant"] = "assistant" role: Literal["assistant"] = "assistant"
content: str | list[TextContent | ImageContent] | None content: str | list[TextContent | ImageContent] | None = None
tool_calls: list[ToolCall] | None = None tool_calls: list[ToolCall] | None = None