primitive 0.1.69__tar.gz → 0.1.70__tar.gz

{primitive-0.1.69 → primitive-0.1.70}/PKG-INFO

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

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

{primitive-0.1.69 → primitive-0.1.70}/src/primitive/agent/runner.py

@@ -49,6 +49,7 @@ class AgentRunner:
         job_slug: str,
         max_log_size: int = 10 * 1024 * 1024,
         log_to_file: bool = True,
+        job: Optional[JobDescription] = None,
     ) -> None:
         self.primitive = primitive
         self.source_dir = source_dir
@@ -70,33 +71,36 @@ class AgentRunner:
             )
 
         # Attempt to load job from file
-        logger.info(f"Scanning directory for job file {self.job_slug}")
-        yaml_file = Path(self.source_dir / ".primitive" / f"{self.job_slug}.yaml")
-        yml_file = Path(self.source_dir / ".primitive" / f"{self.job_slug}.yml")
-
-        if yaml_file.exists() and yml_file.exists():
-            logger.error(
-                f"Found two job descriptions with the same slug: {self.job_slug}"
-            )
-            self.primitive.jobs.job_run_update(
-                self.job_id, status="request_completed", conclusion="failure"
-            )
-            raise FileExistsError
+        if not job:
+            logger.info(f"Scanning directory for job file {self.job_slug}")
+            yaml_file = Path(self.source_dir / ".primitive" / f"{self.job_slug}.yaml")
+            yml_file = Path(self.source_dir / ".primitive" / f"{self.job_slug}.yml")
+
+            if yaml_file.exists() and yml_file.exists():
+                logger.error(
+                    f"Found two job descriptions with the same slug: {self.job_slug}"
+                )
+                self.primitive.jobs.job_run_update(
+                    self.job_id, status="request_completed", conclusion="failure"
+                )
+                raise FileExistsError
+
+            if yaml_file.exists():
+                self.job = yaml.load(open(yaml_file, "r"), Loader=Loader)
+            elif yml_file.exists():
+                self.job = yaml.load(open(yml_file, "r"), Loader=Loader)
+            else:
+                logger.error(
+                    f"No job description with matching slug '{self.job_slug}' found"
+                )
+                self.primitive.jobs.job_run_update(
+                    self.job_id, status="request_completed", conclusion="failure"
+                )
+                raise FileNotFoundError
 
-        if yaml_file.exists():
-            self.job = yaml.load(open(yaml_file, "r"), Loader=Loader)
-        elif yml_file.exists():
-            self.job = yaml.load(open(yml_file, "r"), Loader=Loader)
+            logger.info(f"Found job description for {self.job_slug}")
         else:
-            logger.error(
-                f"No job description with matching slug '{self.job_slug}' found"
-            )
-            self.primitive.jobs.job_run_update(
-                self.job_id, status="request_completed", conclusion="failure"
-            )
-            raise FileNotFoundError
-
-        logger.info(f"Found job description for {self.job_slug}")
+            self.job = job
 
     @staticmethod
     def log_serializer() -> Callable: