From a015cad8b82510f4689651278d8beed984ebbde0 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Tue, 6 Jan 2026 19:02:54 +0800 Subject: [PATCH] chore: run make lint (2 files reformatted, all checks passed) chore: run make type-check (0 errors, 0 warnings, 0 notes) --- api/core/callback_handler/index_tool_callback_handler.py | 4 +++- api/core/rag/retrieval/dataset_retrieval.py | 4 +--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/api/core/callback_handler/index_tool_callback_handler.py b/api/core/callback_handler/index_tool_callback_handler.py index 1387967e9b..4e6d4c2ced 100644 --- a/api/core/callback_handler/index_tool_callback_handler.py +++ b/api/core/callback_handler/index_tool_callback_handler.py @@ -39,7 +39,9 @@ class DatasetIndexToolCallbackHandler: source="app", source_app_id=self._app_id, created_by_role=( - UserFrom.ACCOUNT if self._invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER} else UserFrom.END_USER + UserFrom.ACCOUNT + if self._invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER} + else UserFrom.END_USER ), created_by=self._user_id, ) diff --git a/api/core/rag/retrieval/dataset_retrieval.py b/api/core/rag/retrieval/dataset_retrieval.py index 0142b1e77e..3ebad0c521 100644 --- a/api/core/rag/retrieval/dataset_retrieval.py +++ b/api/core/rag/retrieval/dataset_retrieval.py @@ -177,9 +177,7 @@ class DatasetRetrieval: ) all_documents = [] - user_from = ( - UserFrom.ACCOUNT if invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER} else UserFrom.END_USER - ) + user_from = UserFrom.ACCOUNT if invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER} else UserFrom.END_USER if retrieve_config.retrieve_strategy == DatasetRetrieveConfigEntity.RetrieveStrategy.SINGLE: all_documents = self.single_retrieve( app_id,