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: Run AWEL flow in CLI (#1341)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-27 12:50:05 +08:00
committed by GitHub
parent 340a9fbc35
commit 3a7a2cbbb8
42 changed files with 1454 additions and 422 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
dbgpt/client/flow.py
View File

@@ -93,19 +93,23 @@ async def get_flow(client: Client, flow_id: str) -> FlowPanel:
raise ClientException(f"Failed to get flow: {e}")
async def list_flow(client: Client) -> List[FlowPanel]:
async def list_flow(
client: Client, name: str | None = None, uid: str | None = None
) -> List[FlowPanel]:
"""
List flows.
Args:
client (Client): The dbgpt client.
name (str): The name of the flow.
uid (str): The uid of the flow.
Returns:
List[FlowPanel]: The list of flow panels.
Raises:
ClientException: If the request failed.
"""
try:
res = await client.get("/awel/flows")
res = await client.get("/awel/flows", **{"name": name, "uid": uid})
result: Result = res.json()
if result["success"]:
return [FlowPanel(**flow) for flow in result["data"]["items"]]