atomicguard 0.1.0__py3-none-any.whl → 1.2.0__py3-none-any.whl

atomicguard/__init__.py

@@ -25,7 +25,7 @@ from atomicguard.application.agent import DualStateAgent
 from atomicguard.application.workflow import Workflow, WorkflowStep
 
 # Domain exceptions
-from atomicguard.domain.exceptions import RmaxExhausted
+from atomicguard.domain.exceptions import EscalationRequired, RmaxExhausted
 
 # Domain interfaces (for type hints and custom implementations)
 from atomicguard.domain.interfaces import (
@@ -43,6 +43,7 @@ from atomicguard.domain.models import (
     GuardResult,
     WorkflowResult,
     WorkflowState,
+    WorkflowStatus,
 )
 
 # Prompts and tasks (structures only - content defined by calling applications)
@@ -57,6 +58,7 @@ from atomicguard.guards import (
     CompositeGuard,
     DynamicTestGuard,
     HumanReviewGuard,
+    ImportGuard,
     SyntaxGuard,
     TestGuard,
 )
@@ -71,7 +73,7 @@ from atomicguard.infrastructure.persistence import (
     InMemoryArtifactDAG,
 )
 
-__version__ = "0.1.0"
+__version__ = "1.2.0"
 
 __all__ = [
     # Version
@@ -86,6 +88,7 @@ __all__ = [
     "GuardResult",
     "WorkflowState",
     "WorkflowResult",
+    "WorkflowStatus",
     # Prompts and tasks (structures only)
     "PromptTemplate",
     "StepDefinition",
@@ -96,6 +99,7 @@ __all__ = [
     "ArtifactDAGInterface",
     # Domain exceptions
     "RmaxExhausted",
+    "EscalationRequired",
     # Application layer
     "ActionPair",
     "DualStateAgent",
@@ -105,11 +109,12 @@ __all__ = [
     "InMemoryArtifactDAG",
     "FilesystemArtifactDAG",
     # Infrastructure - LLM
-    "OllamaGenerator",
     "MockGenerator",
+    "OllamaGenerator",
     # Guards
     "CompositeGuard",
     "SyntaxGuard",
+    "ImportGuard",
     "TestGuard",
     "DynamicTestGuard",
     "HumanReviewGuard",

atomicguard/application/action_pair.py

@@ -48,6 +48,8 @@ class ActionPair:
         self,
         context: Context,
         dependencies: dict[str, Artifact] | None = None,
+        action_pair_id: str = "unknown",
+        workflow_id: str = "unknown",
     ) -> tuple[Artifact, GuardResult]:
         """
         Execute the atomic generate-then-validate transaction.
@@ -55,11 +57,15 @@ class ActionPair:
         Args:
             context: Generation context
             dependencies: Artifacts from prior workflow steps
+            action_pair_id: Identifier for this action pair
+            workflow_id: UUID of the workflow execution instance
 
         Returns:
             Tuple of (generated artifact, guard result)
         """
         dependencies = dependencies or {}
-        artifact = self._generator.generate(context, self._prompt_template)
+        artifact = self._generator.generate(
+            context, self._prompt_template, action_pair_id, workflow_id
+        )
         result = self._guard.validate(artifact, **dependencies)
         return artifact, result

atomicguard/application/agent.py

@@ -5,13 +5,17 @@ Manages only EnvironmentState (the retry loop).
 WorkflowState is managed by Workflow.
 """
 
+from dataclasses import replace
+
 from atomicguard.application.action_pair import ActionPair
-from atomicguard.domain.exceptions import RmaxExhausted
+from atomicguard.domain.exceptions import EscalationRequired, RmaxExhausted
 from atomicguard.domain.interfaces import ArtifactDAGInterface
 from atomicguard.domain.models import (
     AmbientEnvironment,
     Artifact,
+    ArtifactStatus,
     Context,
+    FeedbackEntry,
 )
 
 
@@ -33,6 +37,8 @@ class DualStateAgent:
         artifact_dag: ArtifactDAGInterface,
         rmax: int = 3,
         constraints: str = "",
+        action_pair_id: str = "unknown",
+        workflow_id: str = "unknown",
     ):
         """
         Args:
@@ -40,11 +46,15 @@ class DualStateAgent:
             artifact_dag: Repository for storing artifacts
             rmax: Maximum retry attempts (default: 3)
             constraints: Global constraints for the ambient environment
+            action_pair_id: Identifier for this action pair (e.g., 'g_test')
+            workflow_id: UUID of the workflow execution instance
         """
         self._action_pair = action_pair
         self._artifact_dag = artifact_dag
         self._rmax = rmax
         self._constraints = constraints
+        self._action_pair_id = action_pair_id
+        self._workflow_id = workflow_id
 
     def execute(
         self,
@@ -69,18 +79,44 @@ class DualStateAgent:
         context = self._compose_context(specification, dependencies)
         feedback_history: list[tuple[Artifact, str]] = []
         retry_count = 0
+        previous_id: str | None = None  # Track chain linkage
 
         while retry_count <= self._rmax:
-            artifact, result = self._action_pair.execute(context, dependencies)
+            artifact, result = self._action_pair.execute(
+                context, dependencies, self._action_pair_id, self._workflow_id
+            )
+
+            # Build feedback history for context snapshot
+            fb_entries = tuple(
+                FeedbackEntry(artifact_id=a.artifact_id, feedback=f)
+                for a, f in feedback_history
+            )
 
-            self._artifact_dag.store(
-                artifact, metadata="" if result.passed else result.feedback
+            # Update artifact with guard result AND provenance metadata
+            artifact = replace(
+                artifact,
+                previous_attempt_id=previous_id,
+                status=ArtifactStatus.ACCEPTED
+                if result.passed
+                else ArtifactStatus.REJECTED,
+                guard_result=result.passed,
+                feedback=result.feedback,
+                context=replace(
+                    artifact.context,
+                    feedback_history=fb_entries,
+                ),
             )
+            self._artifact_dag.store(artifact)
 
             if result.passed:
                 return artifact
+            elif result.fatal:
+                # Non-recoverable failure - escalate immediately
+                raise EscalationRequired(artifact, result.feedback)
             else:
+                # Recoverable failure - retry
                 feedback_history.append((artifact, result.feedback))
+                previous_id = artifact.artifact_id  # Track for next iteration
                 retry_count += 1
                 context = self._refine_context(
                     specification, artifact, feedback_history, dependencies
@@ -105,7 +141,9 @@ class DualStateAgent:
             specification=specification,
             current_artifact=None,
             feedback_history=(),
-            dependencies=tuple(dependencies.items()),  # Pass to generator
+            dependency_artifacts=tuple(
+                (k, v.artifact_id) for k, v in dependencies.items()
+            ),  # Store IDs, not full artifacts
         )
 
     def _refine_context(
@@ -125,5 +163,7 @@ class DualStateAgent:
             specification=specification,
             current_artifact=artifact.content,
             feedback_history=tuple((a.content, f) for a, f in feedback_history),
-            dependencies=tuple(dependencies.items()),  # Preserve dependencies on retry
+            dependency_artifacts=tuple(
+                (k, v.artifact_id) for k, v in dependencies.items()
+            ),  # Preserve dependencies on retry
         )