atomicguard 2.30.0__tar.gz → 2.32.0__tar.gz

{atomicguard-2.30.0/src/atomicguard.egg-info → atomicguard-2.32.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomicguard
-Version: 2.30.0
+Version: 2.32.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>

{atomicguard-2.30.0 → atomicguard-2.32.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "atomicguard"
-version = "2.30.0"
+version = "2.32.0"
 description = "A Dual-State Agent Framework for reliable LLM code generation with guard-validated loops"
 readme = "README.md"
 license = { text = "MIT" }

{atomicguard-2.30.0 → atomicguard-2.32.0}/src/atomicguard/__init__.py

@@ -84,7 +84,7 @@ from atomicguard.infrastructure.persistence import (
 # Visualization
 from atomicguard.visualization import export_workflow_config_html, export_workflow_html
 
-__version__ = "2.30.0"
+__version__ = "2.32.0"
 
 __all__ = [
     # Version

{atomicguard-2.30.0 → atomicguard-2.32.0}/src/atomicguard/contrib/dsl/ast.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from dataclasses import dataclass, field
 
-type Atom = str | int | bool | tuple["Atom", ...]
+type Atom = str | int | float | bool | tuple["Atom", ...]
 
 
 @dataclass(frozen=True)
@@ -171,6 +171,8 @@ class WorkflowNode:
     model: str = ""
     provider: str = ""
     rmax: int = 3
+    intent_examples: tuple[str, ...] = ()
+    intent_threshold: float | None = None
     action_pairs: tuple[ActionPairNode, ...] = ()
     guards: tuple[GuardFormNode, ...] = ()
     duplicate_fields: tuple[str, ...] = ()

{atomicguard-2.30.0 → atomicguard-2.32.0}/src/atomicguard/contrib/dsl/codegen.py

@@ -74,6 +74,8 @@ def _translate_workflow(
             model=node.model,
             provider=node.provider,
             rmax=node.rmax,
+            intent_examples=node.intent_examples,
+            intent_threshold=node.intent_threshold,
             action_pairs=tuple(action_pairs),
         ),
         contexts,

{atomicguard-2.30.0 → atomicguard-2.32.0}/src/atomicguard/contrib/dsl/parser.py

@@ -88,6 +88,8 @@ class _ASTTransformer(Transformer[Any, Any]):
             model=cast(str, fields.get(":model", "")),
             provider=cast(str, fields.get(":provider", "")),
             rmax=cast(int, fields.get(":rmax", 3)),
+            intent_examples=cast("tuple[str, ...]", fields.get(":intent-examples", ())),
+            intent_threshold=cast("float | None", fields.get(":intent-threshold")),
             action_pairs=tuple(action_pairs),
             guards=tuple(guards),
             duplicate_fields=tuple(duplicates),
@@ -114,6 +116,15 @@ class _ASTTransformer(Transformer[Any, Any]):
     def workflow_rmax(self, items: list[Any]) -> _Field:
         return _Field(":rmax", items[0])
 
+    def workflow_intent_examples(self, items: list[Any]) -> _Field:
+        return _Field(":intent-examples", items[0])
+
+    def workflow_intent_threshold(self, items: list[Any]) -> _Field:
+        return _Field(":intent-threshold", items[0])
+
+    def string_list(self, items: list[Any]) -> tuple[str, ...]:
+        return tuple(cast(str, item) for item in items)
+
     def spec_source(self, items: list[Any]) -> SpecSource:
         value = items[0]
         if isinstance(value, str):
@@ -462,12 +473,18 @@ class _ASTTransformer(Transformer[Any, Any]):
     def INTEGER(self, token: Token) -> int:  # noqa: N802
         return int(str(token))
 
+    def FLOAT(self, token: Token) -> float:  # noqa: N802
+        return float(str(token))
+
     def BOOLEAN(self, token: Token) -> bool:  # noqa: N802
         return str(token) == "true"
 
     def integer(self, items: list[Any]) -> Atom:
         return cast(Atom, items[0])
 
+    def float(self, items: list[Any]) -> Atom:
+        return cast(Atom, items[0])
+
     def boolean(self, items: list[Any]) -> Atom:
         return cast(Atom, items[0])
 

{atomicguard-2.30.0 → atomicguard-2.32.0}/src/atomicguard/contrib/dsl/semantic.py

@@ -35,6 +35,16 @@ def validate_guard_forms(node: WorkflowNode) -> None:
 
 def validate_definition(definition: WorkflowDefinition) -> None:
     """Validate DS-PDDL semantics that depend on the whole workflow."""
+    if definition.intent_threshold is not None and not (
+        0.0 <= definition.intent_threshold <= 1.0
+    ):
+        raise DSPDDLValidationError(
+            f"Workflow '{definition.slug}' has invalid intent_threshold: "
+            f"{definition.intent_threshold}. Expected 0.0 <= intent_threshold <= 1.0. "
+            "This checks type/range only — it makes no claim about whether the "
+            "value is reachable by any particular classifier."
+        )
+
     ap_ids: set[str] = set()
     for ap_def in definition.action_pairs:
         if ap_def.ap_id in ap_ids:

atomicguard-2.32.0/src/atomicguard/contrib/generators/pydantic_ai_generator.py

@@ -0,0 +1,108 @@
+"""Generator that wraps a PydanticAI Agent for LLM calls.
+
+The generator calls the LLM and returns the raw text response.
+Validation is delegated to the guard (JSONSchemaGuard).
+
+Configurable from DS-PDDL via :config:
+    model — LLM model name (e.g. deepseek/deepseek-v4-flash)
+    provider — API provider (openrouter, openai, ollama, huggingface)
+    retries — PydanticAI Agent retries (default 0 — DSA owns correction)
+"""
+
+from __future__ import annotations
+
+import asyncio
+import os
+import re
+import uuid
+from datetime import datetime
+
+from pydantic_ai import Agent
+
+from atomicguard.domain.interfaces import GeneratorInterface
+from atomicguard.domain.models import (
+    ActionPairId,
+    Artifact,
+    ArtifactId,
+    ArtifactStatus,
+    Context,
+    ContextSnapshot,
+    WorkflowId,
+)
+from atomicguard.domain.prompts import PromptTemplate
+from atomicguard.infrastructure.llm.factory import create_llm_model
+
+
+class PydanticAIGenerator(GeneratorInterface):
+    """LLM generator.
+
+    Config (from DS-PDDL :config):
+        model — LLM model name (required)
+        provider — API provider (default: openrouter)
+        retries — PydanticAI retry count (default: 0)
+        api_key — optional, falls back to LLM_API_KEY env var
+        base_url — optional, falls back to OpenRouter default
+    """
+
+    def __init__(
+        self,
+        *,
+        model: str = "",
+        provider: str = "",
+        retries: int = 0,
+        api_key: str = "",
+        base_url: str = "",
+        timeout: float = 180.0,
+    ) -> None:
+        resolved_provider = provider or "openrouter"
+        resolved_api_key = (
+            api_key
+            or os.environ.get("LLM_API_KEY", "")
+            or os.environ.get("OPENAI_API_KEY", "")
+            or os.environ.get("OPENROUTER_API_KEY", "")
+        )
+        resolved_base_url = base_url or "https://openrouter.ai/api/v1"
+
+        ai_model = create_llm_model(
+            provider=resolved_provider,
+            model=model,
+            base_url=resolved_base_url,
+            api_key=resolved_api_key,
+            timeout=timeout,
+        )
+        self._agent = Agent(ai_model, output_type=str, retries=retries)
+
+    def generate(
+        self,
+        context: Context,
+        template: PromptTemplate,
+        action_pair_id: str = "unknown",
+        workflow_id: str = "unknown",
+        workflow_ref: str | None = None,  # noqa: ARG002
+    ) -> Artifact:
+        prompt = template.render(context)
+        result = asyncio.run(self._agent.run(prompt))
+
+        content = result.output.strip()
+        match = re.search(r"```(?:json)?\n(.*?)\n```", content, re.DOTALL)
+        if match:
+            content = match.group(1).strip()
+
+        return Artifact(
+            artifact_id=ArtifactId(str(uuid.uuid4())),
+            workflow_id=WorkflowId(workflow_id),
+            content=content,
+            previous_attempt_id=None,
+            parent_action_pair_id=None,
+            action_pair_id=ActionPairId(action_pair_id),
+            created_at=datetime.now().isoformat(),
+            attempt_number=1,
+            status=ArtifactStatus.PENDING,
+            guard_result=None,
+            context=ContextSnapshot(
+                workflow_id=workflow_id,
+                specification=context.specification,
+                constraints="",
+                feedback_history=(),
+            ),
+        )

atomicguard-2.32.0/src/atomicguard/contrib/guards/json_schema_guard.py

@@ -0,0 +1,47 @@
+"""Guard that validates artifact JSON against a JSON Schema file."""
+
+from __future__ import annotations
+
+import json
+import re
+from pathlib import Path
+
+from atomicguard.domain.interfaces import GuardInterface
+from atomicguard.domain.models import GuardResult
+
+
+class JSONSchemaGuard(GuardInterface):
+    """Validates artifact JSON content against a JSON Schema file.
+
+    Config (:config in DS-PDDL):
+        schema — path to JSON Schema file (resolved from project root)
+    """
+
+    def __init__(self, schema: str = "") -> None:
+        self._schema_path = schema
+
+    def validate(self, artifact: object, **kwargs: object) -> GuardResult:  # noqa: ARG002
+        import jsonschema  # noqa: PLC0415
+
+        content = getattr(artifact, "content", "")
+        if not isinstance(content, str):
+            return GuardResult(
+                passed=False, feedback="Artifact content is not a string"
+            )
+
+        content = content.strip()
+        match = re.search(r"```(?:json)?\n(.*?)\n```", content, re.DOTALL)
+        if match:
+            content = match.group(1).strip()
+
+        try:
+            schema = json.loads(Path(self._schema_path).read_text())
+            data = json.loads(content)
+            jsonschema.validate(data, schema)
+        except json.JSONDecodeError as exc:
+            return GuardResult(passed=False, feedback=f"Invalid JSON: {exc}")
+        except jsonschema.ValidationError as exc:
+            return GuardResult(passed=False, feedback=f"Schema violation: {exc}")
+        except FileNotFoundError as exc:
+            return GuardResult(passed=False, feedback=f"Schema file not found: {exc}")
+        return GuardResult(passed=True)

{atomicguard-2.30.0 → atomicguard-2.32.0}/src/atomicguard/domain/models.py

@@ -511,6 +511,8 @@ class WorkflowDefinition:
     model: str = ""
     provider: str = ""
     rmax: int = 3
+    intent_examples: tuple[str, ...] = ()
+    intent_threshold: float | None = None
     action_pairs: tuple[ActionPairDefinition, ...] = ()
     source_policy_slug: str = ""  # provenance: policy that generated this workflow
     policy_derived_requires: tuple[

{atomicguard-2.30.0 → atomicguard-2.32.0/src/atomicguard.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atomicguard
-Version: 2.30.0
+Version: 2.32.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>

{atomicguard-2.30.0 → atomicguard-2.32.0}/src/atomicguard.egg-info/SOURCES.txt

@@ -42,11 +42,13 @@ src/atomicguard/contrib/effectors/bash_executor.py
 src/atomicguard/contrib/generators/__init__.py
 src/atomicguard/contrib/generators/command_template.py
 src/atomicguard/contrib/generators/embedding_intent.py
+src/atomicguard/contrib/generators/pydantic_ai_generator.py
 src/atomicguard/contrib/guards/__init__.py
 src/atomicguard/contrib/guards/all_passed_guard.py
 src/atomicguard/contrib/guards/always_pass_guard.py
 src/atomicguard/contrib/guards/exit_code_guard.py
 src/atomicguard/contrib/guards/intent_guard.py
+src/atomicguard/contrib/guards/json_schema_guard.py
 src/atomicguard/contrib/guards/parallel.py
 src/atomicguard/contrib/guards/pattern_guard.py
 src/atomicguard/contrib/guards/rendered_command_guard.py