fix(api): switch dataset query created_by_role to CreatorUserRole enums

Note: `CreatorUserRole.END_USER` is `"end_user"` (underscore), matching the prior value.

Tests not run (not requested).
This commit is contained in:
-LAN-
2026-01-06 19:14:52 +08:00
parent a015cad8b8
commit 7e33faecfe
3 changed files with 12 additions and 8 deletions

View File

@ -12,7 +12,7 @@ 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 UserFrom
from models.enums import CreatorUserRole
_logger = logging.getLogger(__name__)
@ -39,9 +39,9 @@ class DatasetIndexToolCallbackHandler:
source="app",
source_app_id=self._app_id,
created_by_role=(
UserFrom.ACCOUNT
CreatorUserRole.ACCOUNT
if self._invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER}
else UserFrom.END_USER
else CreatorUserRole.END_USER
),
created_by=self._user_id,
)