[Chore/Refactor] Switch from MyPy to Basedpyright for type checking (#25047)
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
parent
1fff4620e6
commit
9d5956cef8
84 changed files with 2380 additions and 2351 deletions
|
|
@ -1,12 +1,30 @@
|
|||
from .clean_when_dataset_deleted import handle
|
||||
from .clean_when_document_deleted import handle
|
||||
from .create_document_index import handle
|
||||
from .create_installed_app_when_app_created import handle
|
||||
from .create_site_record_when_app_created import handle
|
||||
from .delete_tool_parameters_cache_when_sync_draft_workflow import handle
|
||||
from .update_app_dataset_join_when_app_model_config_updated import handle
|
||||
from .update_app_dataset_join_when_app_published_workflow_updated import handle
|
||||
from .clean_when_dataset_deleted import handle as handle_clean_when_dataset_deleted
|
||||
from .clean_when_document_deleted import handle as handle_clean_when_document_deleted
|
||||
from .create_document_index import handle as handle_create_document_index
|
||||
from .create_installed_app_when_app_created import handle as handle_create_installed_app_when_app_created
|
||||
from .create_site_record_when_app_created import handle as handle_create_site_record_when_app_created
|
||||
from .delete_tool_parameters_cache_when_sync_draft_workflow import (
|
||||
handle as handle_delete_tool_parameters_cache_when_sync_draft_workflow,
|
||||
)
|
||||
from .update_app_dataset_join_when_app_model_config_updated import (
|
||||
handle as handle_update_app_dataset_join_when_app_model_config_updated,
|
||||
)
|
||||
from .update_app_dataset_join_when_app_published_workflow_updated import (
|
||||
handle as handle_update_app_dataset_join_when_app_published_workflow_updated,
|
||||
)
|
||||
|
||||
# Consolidated handler replaces both deduct_quota_when_message_created and
|
||||
# update_provider_last_used_at_when_message_created
|
||||
from .update_provider_when_message_created import handle
|
||||
from .update_provider_when_message_created import handle as handle_update_provider_when_message_created
|
||||
|
||||
__all__ = [
|
||||
"handle_clean_when_dataset_deleted",
|
||||
"handle_clean_when_document_deleted",
|
||||
"handle_create_document_index",
|
||||
"handle_create_installed_app_when_app_created",
|
||||
"handle_create_site_record_when_app_created",
|
||||
"handle_delete_tool_parameters_cache_when_sync_draft_workflow",
|
||||
"handle_update_app_dataset_join_when_app_model_config_updated",
|
||||
"handle_update_app_dataset_join_when_app_published_workflow_updated",
|
||||
"handle_update_provider_when_message_created",
|
||||
]
|
||||
|
|
|
|||
|
|
@ -61,7 +61,7 @@ def get_dataset_ids_from_workflow(published_workflow: Workflow) -> set[str]:
|
|||
try:
|
||||
node_data = KnowledgeRetrievalNodeData(**node.get("data", {}))
|
||||
dataset_ids.update(dataset_id for dataset_id in node_data.dataset_ids)
|
||||
except Exception as e:
|
||||
except Exception:
|
||||
continue
|
||||
|
||||
return dataset_ids
|
||||
|
|
|
|||
|
|
@ -204,7 +204,7 @@ def _calculate_quota_usage(
|
|||
elif quota_unit == QuotaUnit.TIMES:
|
||||
return 1
|
||||
return None
|
||||
except Exception as e:
|
||||
except Exception:
|
||||
logger.exception("Failed to calculate quota usage")
|
||||
return None
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue