From d08b9bfd3dca8621005b9a431c06cec367a43216 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 23 Jun 2024 00:54:46 -0300 Subject: [PATCH] feat: Add arch and autoLogin fields to VersionPayload The VersionPayload class in the TelemetryService module has been updated to include the arch field for architecture information and the autoLogin field to indicate whether auto login is enabled. This change enhances the telemetry data collected by the application. --- .../base/langflow/services/telemetry/service.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/services/telemetry/service.py b/src/backend/base/langflow/services/telemetry/service.py index 822b40b21..e9894a13d 100644 --- a/src/backend/base/langflow/services/telemetry/service.py +++ b/src/backend/base/langflow/services/telemetry/service.py @@ -34,6 +34,8 @@ class VersionPayload(BaseModel): version: str platform: str python: str + arch: str + autoLogin: bool cacheType: str backendOnly: bool @@ -83,7 +85,7 @@ class TelemetryService(Service): logger.debug("Telemetry tracking is disabled.") return - url = f"{self.base_url}/{self.package}" + url = f"{self.base_url}/{self.package.lower()}" if path: url = f"{url}/{path}" try: @@ -104,12 +106,15 @@ class TelemetryService(Service): async def log_package_version(self): python_version = ".".join(platform.python_version().split(".")[:2]) version_info = get_version_info() + architecture = platform.architecture()[0] payload = VersionPayload( version=version_info["version"], platform=platform.platform(), python=python_version, cacheType=self.settings_service.settings.cache_type, backendOnly=self.settings_service.settings.backend_only, + arch=architecture, + autoLogin=self.settings_service.auth_settings.AUTO_LOGIN, ) await self.telemetry_queue.put((self.send_telemetry_data, payload, None)) @@ -125,8 +130,7 @@ class TelemetryService(Service): try: self.running = True self.worker_task = asyncio.create_task(self.telemetry_worker()) - await self.worker_task - await self.log_package_version() + asyncio.create_task(self.log_package_version()) except Exception as e: logger.error(f"Error starting telemetry service: {e}")