mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
5e3e6b0bd8
- Replaced SubscriptionSchema with SubscriptionConstructor in various parts of the trigger provider implementation to streamline subscription management. - Enhanced the PluginTriggerProviderController to utilize the new subscription constructor for retrieving default properties and credential schemas. - Removed the deprecated get_provider_subscription_schema method from TriggerManager. - Updated TriggerSubscriptionBuilderService to reflect changes in subscription handling, ensuring compatibility with the new structure. These changes improve the clarity and maintainability of the subscription handling within the trigger provider architecture.
78 lines
4.0 KiB
Python
78 lines
4.0 KiB
Python
from collections.abc import Mapping
|
|
from typing import Any, Optional
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from core.entities.provider_entities import ProviderConfig
|
|
from core.plugin.entities.plugin_daemon import CredentialType
|
|
from core.tools.entities.common_entities import I18nObject
|
|
from core.trigger.entities.entities import (
|
|
SubscriptionConstructor,
|
|
TriggerCreationMethod,
|
|
TriggerDescription,
|
|
TriggerIdentity,
|
|
TriggerParameter,
|
|
)
|
|
|
|
|
|
class TriggerProviderSubscriptionApiEntity(BaseModel):
|
|
id: str = Field(description="The unique id of the subscription")
|
|
name: str = Field(description="The name of the subscription")
|
|
provider: str = Field(description="The provider id of the subscription")
|
|
credential_type: CredentialType = Field(description="The type of the credential")
|
|
credentials: dict = Field(description="The credentials of the subscription")
|
|
endpoint: str = Field(description="The endpoint of the subscription")
|
|
parameters: dict = Field(description="The parameters of the subscription")
|
|
properties: dict = Field(description="The properties of the subscription")
|
|
workflows_in_use: int = Field(description="The number of workflows using this subscription")
|
|
|
|
|
|
class TriggerApiEntity(BaseModel):
|
|
name: str = Field(description="The name of the trigger")
|
|
identity: TriggerIdentity = Field(description="The identity of the trigger")
|
|
description: TriggerDescription = Field(description="The description of the trigger")
|
|
parameters: list[TriggerParameter] = Field(description="The parameters of the trigger")
|
|
output_schema: Optional[Mapping[str, Any]] = Field(description="The output schema of the trigger")
|
|
|
|
|
|
class TriggerProviderApiEntity(BaseModel):
|
|
author: str = Field(..., description="The author of the trigger provider")
|
|
name: str = Field(..., description="The name of the trigger provider")
|
|
label: I18nObject = Field(..., description="The label of the trigger provider")
|
|
description: I18nObject = Field(..., description="The description of the trigger provider")
|
|
icon: Optional[str] = Field(default=None, description="The icon of the trigger provider")
|
|
icon_dark: Optional[str] = Field(default=None, description="The dark icon of the trigger provider")
|
|
tags: list[str] = Field(default_factory=list, description="The tags of the trigger provider")
|
|
|
|
plugin_id: Optional[str] = Field(default="", description="The plugin id of the tool")
|
|
plugin_unique_identifier: Optional[str] = Field(default="", description="The unique identifier of the tool")
|
|
|
|
supported_creation_methods: list[TriggerCreationMethod] = Field(
|
|
default_factory=list,
|
|
description="Supported creation methods for the trigger provider. like 'OAUTH', 'APIKEY', 'MANUAL'.",
|
|
)
|
|
|
|
subscription_constructor: Optional[SubscriptionConstructor] = Field(
|
|
default=None, description="The subscription constructor of the trigger provider"
|
|
)
|
|
|
|
subscription_schema: list[ProviderConfig] = Field(
|
|
default_factory=list,
|
|
description="The subscription schema of the trigger provider",
|
|
)
|
|
triggers: list[TriggerApiEntity] = Field(description="The triggers of the trigger provider")
|
|
|
|
|
|
class SubscriptionBuilderApiEntity(BaseModel):
|
|
id: str = Field(description="The id of the subscription builder")
|
|
name: str = Field(description="The name of the subscription builder")
|
|
provider: str = Field(description="The provider id of the subscription builder")
|
|
endpoint: str = Field(description="The endpoint id of the subscription builder")
|
|
parameters: Mapping[str, Any] = Field(description="The parameters of the subscription builder")
|
|
properties: Mapping[str, Any] = Field(description="The properties of the subscription builder")
|
|
credentials: Mapping[str, str] = Field(description="The credentials of the subscription builder")
|
|
credential_type: CredentialType = Field(description="The credential type of the subscription builder")
|
|
|
|
|
|
__all__ = ["TriggerApiEntity", "TriggerProviderApiEntity", "TriggerProviderSubscriptionApiEntity"]
|