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

feat(model): Support yi proxy LLM (#1303)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-15 22:15:37 +08:00
committed by GitHub
parent 6974329116
commit 130ffb08c9
144 changed files with 1286 additions and 1088 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dbgpt/core/awel/dag/base.py
View File

@@ -529,7 +529,7 @@ class DAGContext:
raise ValueError("task_name can't be None")
node_id = self._node_name_to_ids.get(task_name)
if not node_id:
raise ValueError(f"Task name {task_name} not exists in DAG")
raise ValueError(f"Task name {task_name} not in DAG")
task_output = self._task_outputs.get(node_id)
if not task_output:
raise ValueError(f"Task output for task {task_name} not exists")
@@ -845,9 +845,7 @@ def _visualize_dag(
# Check if the edge has been added
if (str(node), str(downstream_node)) not in added_edges:
dot.edge(str(node), str(downstream_node))
mermaid_str += (
f" {node.graph_str} --> {downstream_node.graph_str};\n"
)
mermaid_str += f" {node.graph_str} --> {downstream_node.graph_str};\n" # noqa
added_edges.add((str(node), str(downstream_node)))
add_edges(downstream_node)