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).
This commit is contained in:
-LAN-
2026-01-06 19:41:48 +08:00
parent df8f762159
commit b64725b733

View File

@ -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,
)