github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-26 20:29:34 +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

6
dbgpt/core/interface/message.py
View File

@@ -6,7 +6,7 @@ from abc import ABC, abstractmethod
from datetime import datetime
from typing import Callable, Dict, List, Optional, Tuple, Union, cast
from dbgpt._private.pydantic import BaseModel, Field
from dbgpt._private.pydantic import BaseModel, Field, model_to_dict
from dbgpt.core.interface.storage import (
InMemoryStorage,
ResourceIdentifier,
@@ -42,7 +42,7 @@ class BaseMessage(BaseModel, ABC):
"""
return {
"type": self.type,
"data": self.dict(),
"data": model_to_dict(self),
"index": self.index,
"round_index": self.round_index,
}
@@ -264,7 +264,7 @@ class ModelMessage(BaseModel):
Returns:
List[Dict[str, str]]: The dict list
"""
return list(map(lambda m: m.dict(), messages))
return list(map(lambda m: model_to_dict(m), messages))
@staticmethod
def build_human_message(content: str) -> "ModelMessage":