feat: add openadapt-types dependency and _AgentOutput schema

openadapt_evals/training/trl_rollout.py

@@ -71,7 +71,9 @@
 import logging
 import re
 import time
-from typing import Any, Callable
+from typing import Any, Callable, Optional
+
+from pydantic import BaseModel
 
 from openadapt_evals.adapters.base import BenchmarkAction, BenchmarkObservation
 from openadapt_evals.adapters.rl_env import RLEnvironment, ResetConfig
@@ -99,6 +101,28 @@
 ACTION_REGEX = r"Thought: [^\n]+\nAction: (" + _ACTION_RE + r")"
 
 
+# ---------------------------------------------------------------------------
+# JSON schema for future Outlines JSON-mode constrained decoding
+# ---------------------------------------------------------------------------
+# When the model is SFT'd on JSON format (not DSL), switch constrained
+# decoding to: outlines.json(model, _AgentOutput) instead of regex.
+# This is NOT the default — the default uses DSL regex (ACTION_REGEX).
+class _AgentOutput(BaseModel):
+    """Pydantic schema for Outlines JSON-mode constrained decoding.
+
+    Use with: ``outlines.json(model, _AgentOutput)`` once the model has
+    been SFT'd on JSON action format. Currently unused — default is DSL
+    regex via ACTION_REGEX.
+    """
+
+    reasoning: str
+    type: str  # click, type, key, scroll, wait, done
+    x: Optional[float] = None
+    y: Optional[float] = None
+    text: Optional[str] = None
+    key: Optional[str] = None
+
+
 def _build_outlines_generator(model: Any, processor: Any) -> Any | None:
     """Build an Outlines Generator for constrained generation.
 

pyproject.toml

@@ -42,6 +42,7 @@ dependencies = [
     "pyyaml>=6.0",
     "openadapt-consilium>=0.3.2",
     "openadapt-telemetry>=0.2.0",
+    "openadapt-types>=0.1.0",
 ]
 
 [project.optional-dependencies]
@@ -148,6 +149,7 @@ Documentation = "https://github.com/OpenAdaptAI/openadapt-evals#readme"
 [tool.uv.sources]
 openadapt-consilium = { git = "https://github.com/OpenAdaptAI/openadapt-consilium.git" }
 openadapt-ml = { path = "../openadapt-ml", editable = true }
+openadapt-types = { path = "../openadapt-types", editable = true }
 
 [tool.hatch.build.targets.wheel]
 packages = ["openadapt_evals", "scripts"]