mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-03 10:05:13 +00:00
feat(core): Upgrade pydantic to 2.x (#1428)
This commit is contained in:
@@ -3,6 +3,7 @@ from typing import Any, Dict, List, Union
|
||||
|
||||
from sqlalchemy import Column, DateTime, Integer, String, Text
|
||||
|
||||
from dbgpt._private.pydantic import model_to_dict
|
||||
from dbgpt.serve.rag.api.schemas import SpaceServeRequest, SpaceServeResponse
|
||||
from dbgpt.storage.metadata import BaseDao, Model
|
||||
|
||||
@@ -89,7 +90,7 @@ class KnowledgeSpaceDao(BaseDao):
|
||||
entry = query.first()
|
||||
if entry is None:
|
||||
raise Exception("Invalid request")
|
||||
for key, value in update_request.dict().items(): # type: ignore
|
||||
for key, value in model_to_dict(update_request).items(): # type: ignore
|
||||
if value is not None:
|
||||
setattr(entry, key, value)
|
||||
session.merge(entry)
|
||||
@@ -117,7 +118,9 @@ class KnowledgeSpaceDao(BaseDao):
|
||||
T: The entity
|
||||
"""
|
||||
request_dict = (
|
||||
request.dict() if isinstance(request, SpaceServeRequest) else request
|
||||
model_to_dict(request)
|
||||
if isinstance(request, SpaceServeRequest)
|
||||
else request
|
||||
)
|
||||
entity = KnowledgeSpaceEntity(**request_dict)
|
||||
return entity
|
||||
|
Reference in New Issue
Block a user