From 7f6523e0014360ad2179db218b65229cf344fa07 Mon Sep 17 00:00:00 2001 From: Dillon DuPont Date: Fri, 11 Jul 2025 13:25:18 -0400 Subject: [PATCH] Replaced pkg_resources --- libs/python/computer-server/computer_server/main.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/libs/python/computer-server/computer_server/main.py b/libs/python/computer-server/computer_server/main.py index 306f3cd2..ab1a9f30 100644 --- a/libs/python/computer-server/computer_server/main.py +++ b/libs/python/computer-server/computer_server/main.py @@ -32,10 +32,15 @@ app = FastAPI( protocol_version = 1 try: - import pkg_resources - package_version = pkg_resources.get_distribution("cua-computer-server").version -except pkg_resources.DistributionNotFound: - package_version = "unknown" + from importlib.metadata import version + package_version = version("cua-computer-server") +except Exception: + # Fallback for cases where package is not installed or importlib.metadata is not available + try: + import pkg_resources + package_version = pkg_resources.get_distribution("cua-computer-server").version + except Exception: + package_version = "unknown" accessibility_handler, automation_handler, diorama_handler, file_handler = HandlerFactory.create_handlers() handlers = {