primitive 0.1.67__tar.gz → 0.1.68__tar.gz

{primitive-0.1.67 → primitive-0.1.68}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: primitive
-Version: 0.1.67
+Version: 0.1.68
 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.67 → primitive-0.1.68}/src/primitive/__about__.py

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

{primitive-0.1.67 → primitive-0.1.68}/src/primitive/agent/runner.py

@@ -47,6 +47,7 @@ class AgentRunner:
         job_id: str,
         job_slug: str,
         max_log_size: int = 10 * 1024 * 1024,
+        job: Optional[JobDescription] = None,
     ) -> None:
         self.primitive = primitive
         self.source_dir = source_dir
@@ -61,7 +62,13 @@ class AgentRunner:
         logger.enable("primitive")
         self.swap_logs(label="init")
 
-        logger.info(f"Scanning directory for job {self.job_slug}")
+        if job:
+            self.job = job
+        else:
+            self.load_job_from_file()
+    
+    def load_job_from_file(self) -> JobDescription:
+        logger.info(f"Scanning directory for job file {self.job_slug}")
 
         # Look for job based on slug
         yaml_file = Path(self.source_dir / ".primitive" / f"{self.job_slug}.yaml")