primitive 0.1.55__tar.gz → 0.1.57__tar.gz

{primitive-0.1.55 → primitive-0.1.57}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.55
+Version: 0.1.57
 Project-URL: Documentation, https://github.com//primitivecorp/primitive-cli#readme
 Project-URL: Issues, https://github.com//primitivecorp/primitive-cli/issues
 Project-URL: Source, https://github.com//primitivecorp/primitive-cli

{primitive-0.1.55 → primitive-0.1.57}/src/primitive/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Dylan Stein <dylan@primitive.tech>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.1.55"
+__version__ = "0.1.57"

{primitive-0.1.55 → primitive-0.1.57}/src/primitive/agent/runner.py

@@ -50,6 +50,7 @@ class AgentRunner:
     ) -> None:
         self.primitive = primitive
         self.source_dir = source_dir
+        self.workdir = "."
         self.job_id = job_id
         self.job_slug = job_slug
         self.max_log_size = max_log_size
@@ -111,10 +112,14 @@ class AgentRunner:
         for step in self.steps():
             logger.info(f"Beginning step {step['name']}")
 
+            # Update workdir
+            if "workdir" in step:
+                self.workdir = step["workdir"]
+
             # Define step proc
             proc = Process(
                 cmd=step["cmd"],
-                workdir=Path(self.source_dir / step["workdir"]),
+                workdir=Path(self.source_dir / self.workdir),
                 env=environment,
             )
 

{primitive-0.1.55 → primitive-0.1.57}/src/primitive/agent/uploader.py

@@ -1,5 +1,6 @@
 import typing
 import shutil
+import os
 from loguru import logger
 from pathlib import Path, PurePath
 from ..utils.cache import get_artifacts_cache
@@ -33,14 +34,26 @@ class Uploader:
         for job_cache in subdirs:
             job_run_id = job_cache.name
 
-            files = sorted(
-                [
-                    w_path / file
-                    for w_path, _, w_files in job_cache.walk()
-                    for file in w_files
-                ],
-                key=lambda p: p.stat().st_size,
-            )
+            files = None
+            has_walk = getattr(job_cache, "walk", None)
+            if has_walk:
+                files = sorted(
+                    [
+                        w_path / file
+                        for w_path, _, w_files in job_cache.walk()
+                        for file in w_files
+                    ],
+                    key=lambda p: p.stat().st_size,
+                )
+            else:
+                files = sorted(
+                    [
+                        Path(Path(w_path) / file)
+                        for w_path, _, w_files in os.walk(job_cache)
+                        for file in w_files
+                    ],
+                    key=lambda p: p.stat().st_size,
+                )
 
             file_ids = []
             for file in files: