diff --git a/src/backend/base/langflow/services/telemetry/schema.py b/src/backend/base/langflow/services/telemetry/schema.py new file mode 100644 index 000000000..d9ac221b0 --- /dev/null +++ b/src/backend/base/langflow/services/telemetry/schema.py @@ -0,0 +1,35 @@ +from pydantic import BaseModel + + +class RunPayload(BaseModel): + IsWebhook: bool = False + seconds: int + success: bool + errorMessage: str = "" + + +class ShutdownPayload(BaseModel): + timeRunning: int + + +class VersionPayload(BaseModel): + version: str + platform: str + python: str + arch: str + autoLogin: bool + cacheType: str + backendOnly: bool + + +class PlaygroundPayload(BaseModel): + seconds: int + componentCount: int + success: bool + + +class ComponentPayload(BaseModel): + name: str + seconds: int + success: bool + errorMessage: str diff --git a/src/backend/base/langflow/services/telemetry/service.py b/src/backend/base/langflow/services/telemetry/service.py index 45d3fa353..15ccacf69 100644 --- a/src/backend/base/langflow/services/telemetry/service.py +++ b/src/backend/base/langflow/services/telemetry/service.py @@ -9,47 +9,19 @@ from loguru import logger from pydantic import BaseModel from langflow.services.base import Service +from langflow.services.telemetry.schema import ( + ComponentPayload, + PlaygroundPayload, + RunPayload, + ShutdownPayload, + VersionPayload, +) from langflow.utils.version import get_version_info if TYPE_CHECKING: from langflow.services.settings.service import SettingsService -class RunPayload(BaseModel): - isEndpointName: str - IsWebhook: bool - seconds: int - success: bool - errorMessage: str - - -class ShutdownPayload(BaseModel): - timeRunning: int - - -class VersionPayload(BaseModel): - version: str - platform: str - python: str - arch: str - autoLogin: bool - cacheType: str - backendOnly: bool - - -class PlaygroundPayload(BaseModel): - seconds: int - componentCount: int - success: bool - - -class ComponentPayload(BaseModel): - name: str - seconds: int - success: bool - errorMessage: str - - class TelemetryService(Service): name = "telemetry_service"