[bugf][bootup][WORKSPACE_DIR is discarded at import, so it never takes effect]

bootup() runs at 'import swarms' and assigned WORKSPACE_DIR unconditionally,
then disable_logging() assigned it again with the relative string
'agent_workspace'. Either write alone discards whatever the caller exported.

Default it only when unset, which is what bootup's own comment already claims
('Setup workspace dir only if needed' guarded the mkdir, not the assignment),
and drop the stray assignment from disable_logging, which only reads the value.

Author: ayaangazali

swarms/telemetry/bootup.py

@@ -21,10 +21,13 @@ def bootup():
         # Silence wandb
         os.environ["WANDB_SILENT"] = "true"
 
-        # Setup workspace dir only if needed
-        if not workspace_path.exists():
-            workspace_path.mkdir(parents=True, exist_ok=True)
-        os.environ["WORKSPACE_DIR"] = str(workspace_path)
+        # Only default it. Assigning unconditionally discarded whatever the
+        # caller had exported, so WORKSPACE_DIR never survived `import swarms`.
+        if not os.getenv("WORKSPACE_DIR"):
+            os.environ["WORKSPACE_DIR"] = str(workspace_path)
+        Path(os.environ["WORKSPACE_DIR"]).mkdir(
+            parents=True, exist_ok=True
+        )
 
         # Suppress deprecation warnings
         warnings.filterwarnings("ignore", category=DeprecationWarning)

swarms/utils/disable_logging.py

@@ -9,8 +9,6 @@ def disable_logging():
     Disables logging for specific modules and sets up file and stream handlers.
     Runs in a separate thread to avoid blocking the main thread.
     """
-    os.environ["WORKSPACE_DIR"] = "agent_workspace"
-
     warnings.filterwarnings("ignore", category=UserWarning)
 
     # disable tensorflow warnings
@@ -46,7 +44,7 @@ def disable_logging():
     logging.getLogger().handlers = []
 
     # Get the workspace directory from the environment variables
-    workspace_dir = os.environ["WORKSPACE_DIR"]
+    workspace_dir = os.getenv("WORKSPACE_DIR") or "agent_workspace"
 
     # Check if the workspace directory exists, if not, create it
     if not os.path.exists(workspace_dir):