github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-15 05:59:59 +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

4
dbgpt/serve/flow/service/service.py
View File

@@ -157,7 +157,7 @@ class Service(BaseService[ServeEntity, ServeRequest, ServerResponse]):
try:
if state == State.DEPLOYED:
# Register the DAG
self.dag_manager.register_dag(dag)
self.dag_manager.register_dag(dag, request.uid)
# Update state to RUNNING
request.state = State.RUNNING
request.error_message = ""
@@ -199,7 +199,7 @@ class Service(BaseService[ServeEntity, ServeRequest, ServerResponse]):
entity.version == "0.1.0" and entity.state == State.INITIALIZING
):
# Register the DAG
self.dag_manager.register_dag(dag)
self.dag_manager.register_dag(dag, entity.uid)
# Update state to RUNNING
entity.state = State.RUNNING
entity.error_message = ""