Publish tiny_computer_agent/ from marimo

tiny_computer_agent/__init__.py

@@ -63,20 +63,34 @@ from tiny_computer_agent.models import (
     make_model_brain,
     model_cached,
     model_dependencies,
+    pick_dtype,
     resolve_device,
 )
+from tiny_computer_agent.demo import fake_desktop, write_fake_desktop
 from tiny_computer_agent.observe import (
     Element,
     FakeScreen,
     GrabScreen,
+    ImageScreen,
     Observation,
     ScreenSource,
     ScreenUnavailable,
     normalise_label,
 )
+from tiny_computer_agent.desktop import (
+    BRAIN_NAMES,
+    DEFAULT_KILL_SWITCH,
+    Check,
+    build_agent,
+    display_available,
+    doctor,
+    run_checks,
+)
 from tiny_computer_agent.perception import (
+    AutoDetector,
     Detector,
     DetectorUnavailable,
+    GridDetector,
     IndexedScreen,
     OcrDetector,
     detector_available,
@@ -101,17 +115,23 @@ from tiny_computer_agent.trace import (
 __version__ = "0.3.0"
 
 __all__ = [
-    "ACTION_KINDS", "ACTION_MENU", "Action", "ActionError", "ActionResult",
-    "AgentLoop", "Brain", "BrainError", "DEFAULT_TEXT_MODEL", "DEFAULT_VISION_MODEL",
-    "DecisionMenu", "Detector", "DetectorUnavailable", "DryRunExecutor", "Element",
-    "Episode", "Executor", "FakeScreen", "GoalUnreachable", "GrabScreen",
-    "IndexedScreen", "InputUnavailable", "MODEL_BRAINS", "MenuItem", "MockBrain",
-    "ModelBrain", "Observation", "OcrDetector", "PlannerBrain", "Policy",
-    "PolicyViolation", "PyAutoGuiExecutor", "Scenario", "Scoreboard", "ScreenSource",
-    "ScreenUnavailable", "Step", "TextPlannerBrain", "Trace", "TraceStep", "VLMBrain",
-    "Verdict", "benchmark", "benchmark_traces", "detector_available",
-    "extract_json_object", "extract_json_objects", "load_traces", "make_detector",
+    "ACTION_KINDS", "ACTION_MENU", "Action", "ActionError",
+    "ActionResult", "AgentLoop", "AutoDetector", "BRAIN_NAMES",
+    "Brain", "BrainError", "Check", "DEFAULT_KILL_SWITCH",
+    "DEFAULT_TEXT_MODEL", "DEFAULT_VISION_MODEL", "DecisionMenu", "Detector",
+    "DetectorUnavailable", "DryRunExecutor", "Element", "Episode",
+    "Executor", "FakeScreen", "GoalUnreachable", "GrabScreen",
+    "GridDetector", "ImageScreen", "IndexedScreen", "InputUnavailable",
+    "MODEL_BRAINS", "MenuItem", "MockBrain", "ModelBrain",
+    "Observation", "OcrDetector", "PlannerBrain", "Policy",
+    "PolicyViolation", "PyAutoGuiExecutor", "Scenario", "Scoreboard",
+    "ScreenSource", "ScreenUnavailable", "Step", "TextPlannerBrain",
+    "Trace", "TraceStep", "VLMBrain", "Verdict",
+    "__version__", "benchmark", "benchmark_traces", "build_agent",
+    "detector_available", "display_available", "doctor", "extract_json_object",
+    "extract_json_objects", "fake_desktop", "load_traces", "make_detector",
     "make_model_brain", "model_cached", "model_dependencies", "normalise_label",
-    "parse_action", "parse_decision", "resolve_device", "resolve_point",
-    "run_scenario", "score", "score_replay", "score_trace", "__version__",
+    "parse_action", "parse_decision", "pick_dtype", "resolve_device",
+    "resolve_point", "run_checks", "run_scenario", "score",
+    "score_replay", "score_trace", "write_fake_desktop",
 ]

tiny_computer_agent/brain.py

@@ -428,6 +428,11 @@ class ModelBrain:
         self._last_fingerprint = fingerprint
         return action
 
+    @property
+    def loaded(self) -> bool:
+        """True once the weights are in memory - they load lazily, on the first decide()."""
+        return getattr(self, "_model", None) is not None
+
     def _generate(self, prompt: str, observation: Observation) -> str:
         raise NotImplementedError(
             "subclass ModelBrain and implement _generate(prompt, observation)"