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

9
dbgpt/client/knowledge.py
View File

@@ -2,6 +2,7 @@
import json
from typing import List
from dbgpt._private.pydantic import model_to_dict, model_to_json
from dbgpt.core.schema.api import Result
from .client import Client, ClientException
@@ -20,7 +21,7 @@ async def create_space(client: Client, space_model: SpaceModel) -> SpaceModel:
ClientException: If the request failed.
"""
try:
res = await client.post("/knowledge/spaces", space_model.dict())
res = await client.post("/knowledge/spaces", model_to_dict(space_model))
result: Result = res.json()
if result["success"]:
return SpaceModel(**result["data"])
@@ -42,7 +43,7 @@ async def update_space(client: Client, space_model: SpaceModel) -> SpaceModel:
ClientException: If the request failed.
"""
try:
res = await client.put("/knowledge/spaces", space_model.dict())
res = await client.put("/knowledge/spaces", model_to_dict(space_model))
result: Result = res.json()
if result["success"]:
return SpaceModel(**result["data"])
@@ -126,7 +127,7 @@ async def create_document(client: Client, doc_model: DocumentModel) -> DocumentM
"""
try:
res = await client.post_param("/knowledge/documents", doc_model.dict())
res = await client.post_param("/knowledge/documents", model_to_dict(doc_model))
result: Result = res.json()
if result["success"]:
return DocumentModel(**result["data"])
@@ -210,7 +211,7 @@ async def sync_document(client: Client, sync_model: SyncModel) -> List:
"""
try:
res = await client.post(
"/knowledge/documents/sync", [json.loads(sync_model.json())]
"/knowledge/documents/sync", [json.loads(model_to_json(sync_model))]
)
result: Result = res.json()
if result["success"]: