github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-14 13:40:54 +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

5
dbgpt/client/datasource.py
View File

@@ -1,6 +1,7 @@
"""this module contains the datasource client functions."""
from typing import List
from dbgpt._private.pydantic import model_to_dict
from dbgpt.core.schema.api import Result
from .client import Client, ClientException
@@ -17,7 +18,7 @@ async def create_datasource(
datasource (DatasourceModel): The datasource model.
"""
try:
res = await client.get("/datasources", datasource.dict())
res = await client.get("/datasources", model_to_dict(datasource))
result: Result = res.json()
if result["success"]:
return DatasourceModel(**result["data"])
@@ -41,7 +42,7 @@ async def update_datasource(
ClientException: If the request failed.
"""
try:
res = await client.put("/datasources", datasource.dict())
res = await client.put("/datasources", model_to_dict(datasource))
result: Result = res.json()
if result["success"]:
return DatasourceModel(**result["data"])