atomicguard 0.1.0__tar.gz → 1.1.0__tar.gz

{atomicguard-0.1.0/src/atomicguard.egg-info → atomicguard-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomicguard
-Version: 0.1.0
+Version: 1.1.0
 Summary: A Dual-State Agent Framework for reliable LLM code generation with guard-validated loops
 Author-email: Matthew Thompson <thompsonson@gmail.com>
 Maintainer-email: Matthew Thompson <thompsonson@gmail.com>
@@ -26,17 +26,10 @@ Classifier: Typing :: Typed
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: click>=8.3.1
 Requires-Dist: matplotlib>=3.10.0
-Requires-Dist: openai>=2.12.0
-Requires-Dist: rich>=14.0.0
-Provides-Extra: dev
-Requires-Dist: mypy>=1.13.0; extra == "dev"
-Requires-Dist: pre-commit>=4.5.0; extra == "dev"
-Requires-Dist: ruff>=0.14.0; extra == "dev"
-Provides-Extra: test
-Requires-Dist: pytest>=8.0.0; extra == "test"
-Requires-Dist: pytest-cov>=6.0.0; extra == "test"
+Requires-Dist: openhands-ai>=0.27.0
+Requires-Dist: pydantic-ai>=1.0.0
+Requires-Dist: pytestarch>=4.0.1
 Dynamic: license-file
 
 # AtomicGuard
@@ -124,13 +117,20 @@ atomicguard/
 
 ## Citation
 
+If you use this framework in your research, please cite the paper:
+
+> Thompson, M. (2025). Managing the Stochastic: Foundations of Learning in Neuro-Symbolic Systems for Software Engineering. arXiv preprint arXiv:2512.20660.
+
 ```bibtex
 @article{thompson2025managing,
   title={Managing the Stochastic: Foundations of Learning in Neuro-Symbolic Systems for Software Engineering},
   author={Thompson, Matthew},
-  year={2025}
+  journal={arXiv preprint arXiv:2512.20660},
+  year={2025},
+  url={[https://arxiv.org/abs/2512.20660](https://arxiv.org/abs/2512.20660)}
 }
-```
+
+``` 
 
 ## License
 

{atomicguard-0.1.0 → atomicguard-1.1.0}/README.md

@@ -83,13 +83,20 @@ atomicguard/
 
 ## Citation
 
+If you use this framework in your research, please cite the paper:
+
+> Thompson, M. (2025). Managing the Stochastic: Foundations of Learning in Neuro-Symbolic Systems for Software Engineering. arXiv preprint arXiv:2512.20660.
+
 ```bibtex
 @article{thompson2025managing,
   title={Managing the Stochastic: Foundations of Learning in Neuro-Symbolic Systems for Software Engineering},
   author={Thompson, Matthew},
-  year={2025}
+  journal={arXiv preprint arXiv:2512.20660},
+  year={2025},
+  url={[https://arxiv.org/abs/2512.20660](https://arxiv.org/abs/2512.20660)}
 }
-```
+
+``` 
 
 ## License
 

{atomicguard-0.1.0 → atomicguard-1.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "atomicguard"
-version = "0.1.0"
+version = "1.1.0"
 description = "A Dual-State Agent Framework for reliable LLM code generation with guard-validated loops"
 readme = "README.md"
 license = { text = "MIT" }
@@ -35,10 +35,10 @@ classifiers = [
     "Typing :: Typed",
 ]
 dependencies = [
-    "click>=8.3.1",
     "matplotlib>=3.10.0",
-    "openai>=2.12.0",
-    "rich>=14.0.0",
+    "openhands-ai>=0.27.0",
+    "pydantic-ai>=1.0.0",
+    "pytestarch>=4.0.1",
 ]
 
 [project.urls]
@@ -48,16 +48,13 @@ Documentation = "https://github.com/thompsonson/atomicguard#readme"
 Issues = "https://github.com/thompsonson/atomicguard/issues"
 Changelog = "https://github.com/thompsonson/atomicguard/blob/main/CHANGELOG.md"
 
-[project.optional-dependencies]
-dev = [
-    "mypy>=1.13.0",
-    "pre-commit>=4.5.0",
-    "ruff>=0.14.0",
-]
-test = [
-    "pytest>=8.0.0",
-    "pytest-cov>=6.0.0",
-]
+[project.entry-points."atomicguard.generators"]
+# Core generators
+OllamaGenerator = "atomicguard.infrastructure.llm:OllamaGenerator"
+MockGenerator = "atomicguard.infrastructure.llm:MockGenerator"
+
+# Note: Dev dependencies are in [dependency-groups] below, not here.
+# [project.optional-dependencies] is for end-user extras only.
 
 [build-system]
 requires = ["setuptools>=75.0.0", "wheel"]
@@ -91,7 +88,6 @@ prerelease = false
 
 [tool.semantic_release.remote]
 type = "github"
-token = { env = "GH_TOKEN" }
 
 # =============================================================================
 # Testing
@@ -184,3 +180,23 @@ mypy_path = "src"
 [[tool.mypy.overrides]]
 module = "tests.*"
 disallow_untyped_defs = false
+
+[dependency-groups]
+dev = [
+    "mypy>=1.19.1",
+    "pre-commit>=4.5.0",
+    "python-semantic-release>=9.21.1",
+    "types-jsonschema>=4.25.1.20251009",
+    "ruff>=0.14.9",
+    "types-jsonschema>=4.25.1.20251009",
+]
+test = [
+    "pytest>=8.0.0",
+    "pytest-cov>=6.0.0",
+]
+examples = [
+    "click>=8.1.0",
+    "gradio>=4.0.0",
+    "rich>=13.0.0",
+    "typer>=0.9.0",
+]

{atomicguard-0.1.0 → atomicguard-1.1.0}/src/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.1.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-0.1.0 → atomicguard-1.1.0}/src/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-0.1.0 → atomicguard-1.1.0}/src/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
         )