Class ScriptedCodeSession

packages/java/dev/axllm/ax/Conformance.java:310–387

  }

  static final class ScriptedCodeSession implements AxCodeSession {
    final ScriptedCodeRuntime runtime;
    Map<String, Object> globals;
    Map<String, Object> createOptions;
    boolean closed;

    ScriptedCodeSession(ScriptedCodeRuntime runtime, Map<String, Object> globals, Map<String, Object> options) {
      this.runtime = runtime;
      this.globals = new LinkedHashMap<>(globals == null ? Map.of() : globals);
      this.createOptions = new LinkedHashMap<>(options == null ? Map.of() : options);
    }

    public Object execute(String code, Map<String, Object> options) {
      if (closed) return new LinkedHashMap<>(Map.of("is_error", true, "error_category", "session_closed", "error", "session closed"));
      if (runtime.script.isEmpty()) throw new RuntimeException("scripted runtime exhausted");
      Map<String, Object> step = Core.asMap(runtime.script.remove(0));
      Object expected = step.get("expected_code");
      if (expected != null && !String.valueOf(expected).equals(code)) throw new RuntimeException("expected code " + expected + ", got " + code);
      if (step.containsKey("expected_options_subset")) assertSubset(options == null ? Map.of() : options, step.get("expected_options_subset"), "runtime execute options");
      runtime.executed.add(code);
      runtime.executeOptions.add(new LinkedHashMap<>(options == null ? Map.of() : options));
      globals.putAll(Core.asMap(step.get("bindings_patch")));
      if (Boolean.TRUE.equals(step.get("close_before_result"))) closed = true;
      return step.getOrDefault("result", new LinkedHashMap<>(Map.of("kind", "result", "result", new LinkedHashMap<>(globals))));
    }

    public Object inspectGlobals(Map<String, Object> options) {
      if (Boolean.FALSE.equals(runtime.capabilities.get("inspect"))) {
        return "[runtime state inspection unavailable: runtime session does not implement inspectGlobals()]";
      }
      return new LinkedHashMap<>(globals);
    }

    public Object snapshotGlobals(Map<String, Object> options) {
      if (Boolean.FALSE.equals(runtime.capabilities.get("snapshot"))) {
        throw new RuntimeException("AxCodeSession.snapshotGlobals() is required to export AxAgent state");
      }
      List<Object> entries = new ArrayList<>();
      for (Map.Entry<String, Object> entry : globals.entrySet()) {
        entries.add(new LinkedHashMap<>(Map.of(
          "name", entry.getKey(),
          "type", entry.getValue() == null ? "null" : entry.getValue().getClass().getSimpleName(),
          "preview", String.valueOf(entry.getValue())
        )));
      }
      Map<String, Object> out = new LinkedHashMap<>();
      out.put("version", 1);
      out.put("entries", entries);
      out.put("bindings", new LinkedHashMap<>(globals));
      out.put("globals", new LinkedHashMap<>(globals));
      out.put("closed", closed);
      return out;
    }

    public Object patchGlobals(Object snapshot, Map<String, Object> options) {
      if (Boolean.FALSE.equals(runtime.capabilities.get("patch"))) {
        throw new RuntimeException("AxCodeSession.patchGlobals() is required to restore AxAgent state");
      }