zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-06-29 23:41:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: bootstrap agent replica on demand (#14588)

Browse Source 
### What problem does this PR solve?

bootstrap agent replica on demand

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
buua436
2026-05-06 19:07:50 +08:00
committed by GitHub
parent f32034e83e
commit c9513e5ecb
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
api/apps/restful_apis/agent_api.py
View File

@@ -941,11 +941,27 @@ async def agent_chat_completion(tenant_id, agent_id=None):
code=RetCode.OPERATING_ERROR,
)
_, cvs = await thread_pool_exec(UserCanvasService.get_by_id, agent_id)
if not cvs:
return get_data_error_result(message="canvas not found.")
replica_payload = CanvasReplicaService.load_for_run(
canvas_id=agent_id,
tenant_id=str(tenant_id),
runtime_user_id=user_id,
)
if not replica_payload:
try:
replica_payload = CanvasReplicaService.bootstrap(
canvas_id=agent_id,
tenant_id=str(tenant_id),
runtime_user_id=user_id,
dsl=cvs.dsl,
canvas_category=getattr(cvs, "canvas_category", CanvasCategory.Agent),
title=getattr(cvs, "title", ""),
)
except ValueError as exc:
return get_data_error_result(message=str(exc))
if not replica_payload:
return get_data_error_result(message="canvas replica not found, please fetch the agent first.")
@@ -954,7 +970,6 @@ async def agent_chat_completion(tenant_id, agent_id=None):
canvas_category = replica_payload.get("canvas_category", CanvasCategory.Agent)
dsl_str = json.dumps(replica_dsl, ensure_ascii=False)
_, cvs = await thread_pool_exec(UserCanvasService.get_by_id, agent_id)
if cvs.canvas_category == CanvasCategory.DataFlow:
task_id = get_uuid()
Pipeline(