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

feat(agent): Release agent SDK (#1396)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-04-10 22:44:53 +08:00
committed by GitHub
parent 37e7c0151b
commit df80bc2079
152 changed files with 5680 additions and 6114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
dbgpt/client/schema.py
View File

@@ -75,8 +75,8 @@ class ChatMode(Enum):
CHAT_DATA = "chat_data"
class AwelTeamModel(BaseModel):
"""Awel team model."""
class AWELTeamModel(BaseModel):
"""AWEL team model."""
dag_id: str = Field(
...,
@@ -137,7 +137,7 @@ class AgentResourceType(Enum):
TextFile = "text_file"
ExcelFile = "excel_file"
ImageFile = "image_file"
AwelFlow = "awel_flow"
AWELFlow = "awel_flow"
class AgentResourceModel(BaseModel):
@@ -206,7 +206,7 @@ class AppModel(BaseModel):
app_describe: Optional[str] = Field(None, title="app describe")
team_mode: Optional[str] = Field(None, title="team mode")
language: Optional[str] = Field("en", title="language")
team_context: Optional[Union[str, AwelTeamModel]] = Field(
team_context: Optional[Union[str, AWELTeamModel]] = Field(
None, title="team context"
)
user_code: Optional[str] = Field(None, title="user code")