github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-05 02:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): Upgrade pydantic to 2.x (#1428)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-04-20 09:41:16 +08:00
committed by GitHub
parent baa1e3f9f6
commit 57be1ece18
103 changed files with 1146 additions and 534 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/serve/prompt/api/schemas.py
View File

@@ -1,16 +1,15 @@
# Define your Pydantic schemas here
from typing import Optional
from dbgpt._private.pydantic import BaseModel, Field
from dbgpt._private.pydantic import BaseModel, ConfigDict, Field
from ..config import SERVE_APP_NAME_HUMP
class ServeRequest(BaseModel):
"""Prompt request model"""
"""Prompt request model."""
class Config:
title = f"ServeRequest for {SERVE_APP_NAME_HUMP}"
model_config = ConfigDict(title=f"ServeRequest for {SERVE_APP_NAME_HUMP}")
chat_scene: Optional[str] = Field(
None,
@@ -69,8 +68,7 @@ class ServeRequest(BaseModel):
class ServerResponse(ServeRequest):
"""Prompt response model"""
class Config:
title = f"ServerResponse for {SERVE_APP_NAME_HUMP}"
model_config = ConfigDict(title=f"ServerResponse for {SERVE_APP_NAME_HUMP}")
id: Optional[int] = Field(
None,