From b64725b73376efb13449b280a8bab3a7eaf42198 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Tue, 6 Jan 2026 19:41:48 +0800 Subject: [PATCH] refactor(api): use InvokeFrom.to_creator_user_role in dataset query callback CreatorUserRole.END_USER maps to `end_user` (underscore), and the callback now uses the shared `to_creator_user_role` conversion. Tests not run (not requested). --- api/core/callback_handler/index_tool_callback_handler.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/api/core/callback_handler/index_tool_callback_handler.py b/api/core/callback_handler/index_tool_callback_handler.py index b054409681..0a690d08a8 100644 --- a/api/core/callback_handler/index_tool_callback_handler.py +++ b/api/core/callback_handler/index_tool_callback_handler.py @@ -12,7 +12,6 @@ from core.rag.models.document import Document from extensions.ext_database import db from models.dataset import ChildChunk, DatasetQuery, DocumentSegment from models.dataset import Document as DatasetDocument -from models.enums import CreatorUserRole _logger = logging.getLogger(__name__) @@ -38,11 +37,7 @@ class DatasetIndexToolCallbackHandler: content=query, source="app", source_app_id=self._app_id, - created_by_role=( - CreatorUserRole.ACCOUNT - if self._invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER} - else CreatorUserRole.END_USER - ), + created_by_role=self._invoke_from.to_creator_user_role(), created_by=self._user_id, )