Skip to content

Commit 9817089

Browse files
committed
fix: fix schema not found issue by enforce pydantic serialize_by_alias
1 parent 3a9b7ab commit 9817089

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

lmdeploy/serve/openai/protocol.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import Any, Dict, List, Literal, Optional, Union
66

77
import shortuuid
8-
from pydantic import BaseModel, Field
8+
from pydantic import BaseModel, ConfigDict, Field
99

1010

1111
class ErrorResponse(BaseModel):
@@ -90,10 +90,12 @@ class JsonSchema(BaseModel):
9090
name: str
9191
# description is not used since it depends on model
9292
description: Optional[str] = None
93+
# `schema` is a reserved field in Pydantic BaseModel
9394
# use alias since pydantic does not support the OpenAI key `schema`
9495
json_schema: Optional[Dict[str, Any]] = Field(default=None, alias='schema', examples=[None])
9596
# strict is not used
9697
strict: Optional[bool] = False
98+
model_config = ConfigDict(serialize_by_alias=True)
9799

98100

99101
class ResponseFormat(BaseModel):

0 commit comments

Comments
 (0)