diff --git a/src/codegate/updates/client.py b/src/codegate/updates/client.py index f899a43b..77dd6ac0 100644 --- a/src/codegate/updates/client.py +++ b/src/codegate/updates/client.py @@ -2,12 +2,15 @@ import requests import structlog +import os logger = structlog.get_logger("codegate") __update_client_singleton = None +is_dev_env = bool(os.environ.get("CODEGATE_DEV_ENV")) + # Enum representing whether the request is coming from the front-end or the back-end. class Origin(Enum): @@ -25,9 +28,13 @@ def get_latest_version(self, origin: Origin) -> str: """ Retrieves the latest version of CodeGate from updates.codegate.ai """ + + user_agent = f"codegate/{self.__current_version} {origin.value}" + if is_dev_env: + user_agent += "-dev" headers = { "X-Instance-ID": self.__instance_id, - "User-Agent": f"codegate/{self.__current_version} {origin.value}", + "User-Agent": user_agent, } try: