mirror of
https://github.com/langgenius/dify.git
synced 2026-01-14 06:07:33 +08:00
c3ebb22a4b
- Introduced a new field `workflows_in_use` to the TriggerProviderSubscriptionApiEntity to track the number of workflows utilizing each subscription. - Enhanced the TriggerProviderService to populate this field by querying the WorkflowPluginTrigger model for usage counts associated with each subscription. This addition improves the visibility of subscription usage within the trigger provider context.
71 lines
3.8 KiB
Python
71 lines
3.8 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 (
|
|
SubscriptionSchema,
|
|
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")
|
|
|
|
credentials_schema: list[ProviderConfig] = Field(description="The credentials schema of the trigger provider")
|
|
oauth_client_schema: list[ProviderConfig] = Field(
|
|
default_factory=list, description="The schema of the OAuth client"
|
|
)
|
|
subscription_schema: Optional[SubscriptionSchema] = Field(
|
|
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"]
|