Files
dify-plugin-sdks/python/dify_plugin/config/config.py
T
-LAN- b3e0612e53 chore: Apply format
Signed-off-by: -LAN- <laipz8200@outlook.com>
2024-12-02 21:18:36 +08:00

43 lines
1.5 KiB
Python

from enum import Enum
from typing import Optional
from pydantic import Field
from pydantic_settings import BaseSettings, SettingsConfigDict
class InstallMethod(Enum):
Local = "local"
Remote = "remote"
AWSLambda = "aws_lambda"
class DifyPluginEnv(BaseSettings):
MAX_REQUEST_TIMEOUT: int = Field(default=300, description="Maximum request timeout in seconds")
MAX_WORKER: int = Field(
default=1000,
description="Maximum worker count, gevent will be used for async IO"
"and you dont need to worry about the thread count",
)
HEARTBEAT_INTERVAL: float = Field(default=10, description="Heartbeat interval in seconds")
INSTALL_METHOD: InstallMethod = Field(
default=InstallMethod.Local,
description="Installation method, local or network",
)
REMOTE_INSTALL_HOST: str = Field(default="localhost", description="Remote installation host")
REMOTE_INSTALL_PORT: int = Field(default=5003, description="Remote installation port")
REMOTE_INSTALL_KEY: Optional[str] = Field(default=None, description="Remote installation key")
AWS_LAMBDA_PORT: int = Field(default=8080, description="AWS Lambda port")
DIFY_PLUGIN_DAEMON_URL: str = Field(default="http://localhost:5002", description="backwards invocation address")
model_config = SettingsConfigDict(
# read from dotenv format config file
env_file=".env",
env_file_encoding="utf-8",
frozen=True,
# ignore extra attributes
extra="ignore",
)