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

@ -63,7 +63,7 @@ from libs.json_in_md_parser import parse_and_check_json_markdown
from models import UploadFile
from models.dataset import ChildChunk, Dataset, DatasetMetadata, DatasetQuery, DocumentSegment, SegmentAttachmentBinding
from models.dataset import Document as DatasetDocument
from models.enums import UserFrom
from models.enums import CreatorUserRole
from services.external_knowledge_service import ExternalDatasetService
default_retrieval_model: dict[str, Any] = {
@ -177,7 +177,11 @@ class DatasetRetrieval:
)
all_documents = []
user_from = UserFrom.ACCOUNT if invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER} else UserFrom.END_USER
user_from = (
CreatorUserRole.ACCOUNT
if invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER}
else CreatorUserRole.END_USER
)
if retrieve_config.retrieve_strategy == DatasetRetrieveConfigEntity.RetrieveStrategy.SINGLE:
all_documents = self.single_retrieve(
app_id,