mirror of
https://github.com/langgenius/dify.git
synced 2026-03-11 02:07:50 +08:00
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:
@ -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,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user