fix: setup logging

Author: cristipufu

src/uipath/_cli/__init__.py

@@ -5,6 +5,7 @@
 
 from uipath.functions import register_default_runtime_factory
 
+from .._utils._logs import setup_logging
 from ._utils._common import add_cwd_to_path, load_environment_variables
 from ._utils._context import CliContext
 from .cli_add import add as add
@@ -88,6 +89,8 @@ def cli(
         debug=debug,
     )
 
+    setup_logging(should_debug=debug)
+
     if lv:
         try:
             version = importlib.metadata.version("uipath-langchain")

src/uipath/_utils/_logs.py

@@ -5,9 +5,17 @@
 
 
 def setup_logging(should_debug: Optional[bool] = None) -> None:
-    if not logging.root.handlers and not logger.handlers:
-        logging.basicConfig(
-            format="%(message)s",
-            datefmt="%Y-%m-%d %H:%M:%S",
+    """Configure logging for the CLI."""
+    if not logger.handlers:
+        handler = logging.StreamHandler()
+        handler.setFormatter(
+            logging.Formatter(
+                fmt="%(message)s",
+                datefmt="%Y-%m-%d %H:%M:%S",
+            )
         )
+        logger.addHandler(handler)
         logger.setLevel(logging.DEBUG if should_debug else logging.INFO)
+
+        # Prevent propagation to root logger to avoid duplicate logs
+        logger.propagate = False

src/uipath/platform/_uipath.py

@@ -26,7 +26,6 @@
     UiPathOpenAIService,
 )
 from .._utils._auth import resolve_config
-from .._utils._logs import setup_logging
 from .errors import BaseUrlMissingError, SecretMissingError
 
 
@@ -55,7 +54,6 @@ def __init__(
                     raise BaseUrlMissingError() from e
                 elif error["loc"][0] == "secret":
                     raise SecretMissingError() from e
-        setup_logging(should_debug=debug)
         self._execution_context = ExecutionContext()
 
     @property