mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
896f3252b8
- Replaced direct imports of `TriggerProviderID` and `ToolProviderID` from `core.plugin.entities.plugin` with imports from `models.provider_ids` for better organization. - Refactored workflow node classes to inherit from a unified `Node` class, improving consistency and maintainability. - Removed unused code and comments to clean up the implementation, particularly in the `workflow_trigger.py` and `builtin_tools_manage_service.py` files. These changes enhance the clarity and structure of the codebase, facilitating easier future modifications.
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import logging
|
|
import re
|
|
from collections.abc import Sequence
|
|
from typing import Any
|
|
|
|
from core.plugin.entities.plugin_daemon import CredentialType
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def generate_provider_name(
|
|
providers: Sequence[Any], credential_type: CredentialType, fallback_context: str = "provider"
|
|
) -> str:
|
|
try:
|
|
return generate_incremental_name(
|
|
[provider.name for provider in providers],
|
|
f"{credential_type.get_name()}",
|
|
)
|
|
except Exception as e:
|
|
logger.warning("Error generating next provider name for %r: %r", fallback_context, e)
|
|
return f"{credential_type.get_name()} 1"
|
|
|
|
|
|
def generate_incremental_name(
|
|
names: Sequence[str],
|
|
default_pattern: str,
|
|
) -> str:
|
|
pattern = rf"^{re.escape(default_pattern)}\s+(\d+)$"
|
|
numbers = []
|
|
|
|
for name in names:
|
|
if not name:
|
|
continue
|
|
match = re.match(pattern, name.strip())
|
|
if match:
|
|
numbers.append(int(match.group(1)))
|
|
|
|
if not numbers:
|
|
return f"{default_pattern} 1"
|
|
|
|
max_number = max(numbers)
|
|
return f"{default_pattern} {max_number + 1}"
|