mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-18 13:47:21 +08:00
Feat: Modify the style of the release confirmation box. (#13542)
### What problem does this PR solve? Feat: Modify the style of the release confirmation box. ### Type of change - [x] New Feature (non-breaking change which adds functionality) --------- Co-authored-by: Yingfeng <yingfeng.zhang@gmail.com> Co-authored-by: balibabu <assassin_cike@163.com> Co-authored-by: 6ba3i <isbaaoui09@gmail.com>
This commit is contained in:
@@ -24,6 +24,7 @@ from api.db import CanvasCategory
|
||||
from api.db.services.canvas_service import CanvasTemplateService, UserCanvasService, API4ConversationService
|
||||
from api.db.services.document_service import DocumentService
|
||||
from api.db.services.file_service import FileService
|
||||
from api.db.services.knowledgebase_service import KnowledgebaseService
|
||||
from api.db.services.pipeline_operation_log_service import PipelineOperationLogService
|
||||
from api.db.services.task_service import queue_dataflow, CANVAS_DEBUG_DOC_ID, TaskService
|
||||
from api.db.services.user_service import TenantService
|
||||
@@ -153,6 +154,11 @@ def get(canvas_id):
|
||||
c = c.to_dict()
|
||||
c["last_publish_time"] = last_publish_time
|
||||
|
||||
# For pipeline type, get associated datasets
|
||||
if c.get("canvas_category") == CanvasCategory.DataFlow:
|
||||
datasets = list(KnowledgebaseService.query(pipeline_id=canvas_id))
|
||||
c["datasets"] = [{"id": d.id, "name": d.name, "avatar": d.avatar} for d in datasets]
|
||||
|
||||
return get_json_result(data=c)
|
||||
|
||||
|
||||
|
||||
@@ -87,19 +87,20 @@ async def create(tenant_id, chat_id):
|
||||
async def create_agent_session(tenant_id, agent_id):
|
||||
req = await get_request_json()
|
||||
user_id = req.get("user_id") or request.args.get("user_id", tenant_id)
|
||||
release_mode = req.get("release", request.args.get("release", False))
|
||||
e, cvs = UserCanvasService.get_by_id(agent_id)
|
||||
if not e:
|
||||
return get_error_data_result("Agent not found.")
|
||||
release_mode = bool(req.get("release", request.args.get("release", False)))
|
||||
|
||||
if not UserCanvasService.query(user_id=tenant_id, id=agent_id):
|
||||
return get_error_data_result("You cannot access the agent.")
|
||||
if not isinstance(cvs.dsl, str):
|
||||
cvs.dsl = json.dumps(cvs.dsl, ensure_ascii=False)
|
||||
|
||||
if release_mode and not bool(cvs.release):
|
||||
raise PermissionError("No available published version")
|
||||
try:
|
||||
cvs, dsl = UserCanvasService.get_agent_dsl_with_release(agent_id, release_mode, tenant_id)
|
||||
except LookupError:
|
||||
return get_error_data_result("Agent not found.")
|
||||
except PermissionError as e:
|
||||
return get_error_data_result(str(e))
|
||||
|
||||
session_id = get_uuid()
|
||||
canvas = Canvas(cvs.dsl, tenant_id, agent_id, canvas_id=cvs.id)
|
||||
canvas = Canvas(dsl, tenant_id, agent_id, canvas_id=cvs.id)
|
||||
canvas.reset()
|
||||
|
||||
cvs.dsl = json.loads(str(canvas))
|
||||
|
||||
Reference in New Issue
Block a user