chore: refurish python code by applying Pylint linter rules (#8322)

2024-09-13 22:42:08 +08:00
parent 1ab81b4972
commit a1104ab97e
126 changed files with 253 additions and 272 deletions
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@@ -115,7 +115,7 @@ class DatasetRetrieval:

            available_datasets.append(dataset)
        all_documents = []
-        user_from = "account" if invoke_from in [InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER] else "end_user"
+        user_from = "account" if invoke_from in {InvokeFrom.EXPLORE, InvokeFrom.DEBUGGER} else "end_user"
        if retrieve_config.retrieve_strategy == DatasetRetrieveConfigEntity.RetrieveStrategy.SINGLE:
            all_documents = self.single_retrieve(
                app_id,