Delete scratch

scratch/osint/Dockerfile

@@ -1,27 +0,0 @@
-FROM python:3.12-slim
-
-RUN useradd -m -u 1000 user
-
-USER user
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    PYTHONDONTWRITEBYTECODE=1 \
-    PYTHONUNBUFFERED=1 \
-    PORT=7860
-
-WORKDIR $HOME/app
-
-COPY --chown=user pyproject.toml README.md openenv.yaml inference.py $HOME/app/
-COPY --chown=user src $HOME/app/src
-COPY --chown=user config $HOME/app/config
-COPY --chown=user datasets $HOME/app/datasets
-COPY --chown=user docs $HOME/app/docs
-COPY --chown=user scripts $HOME/app/scripts
-COPY --chown=user server.py $HOME/app/server.py
-
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -e .
-
-EXPOSE 7860
-
-CMD ["sh", "-c", "uvicorn server:app --host 0.0.0.0 --port ${PORT:-7860}"]

scratch/osint/README.md

@@ -1,262 +0,0 @@
----
-title: OSINT OpenEnv
-emoji: 🕵️
-colorFrom: blue
-colorTo: yellow
-sdk: docker
-app_port: 7860
-pinned: false
-license: apache-2.0
-tags:
-  - openenv
-  - osint
-  - benchmark
-  - docker
-  - fastapi
-short_description: Docker OSINT benchmark with fixed OpenEnv tasks.
----
-
-# OSINT OpenEnv
-
-OSINT OpenEnv is a synthetic benchmark environment for tool-using agents that must recover identities, trace events, and link entities across noisy multi-platform records. The project is designed to feel like a compact OSINT workflow rather than a raw QA dataset: agents query mock profiles, posts, forum threads, and semantic memory, build a working graph, and then submit an answer.
-
-The motivation is to provide a reproducible OpenEnv-compatible environment for evaluating graph-building and tool-using reasoning without depending on live web data, unstable APIs, or private corpora. That makes it useful for local development, regression testing, and hosted demos such as a Docker-based Hugging Face Space.
-
-## Environment Summary
-
-The environment generates or loads a hidden canonical graph of users, aliases, organizations, locations, posts, threads, and events. It then exposes partial platform views and a task list drawn from that graph.
-
-The default hosted Space uses the fixed-level benchmark in `datasets/fixed_levels/seed_fixed_levels.json`, which now contains 30 stable tasks over a larger shared seeded graph.
-
-## Action Space
-
-The environment exposes three actions:
-
-- `CALL_TOOL`: query platform views or semantic memory such as `search_posts`, `get_profile`, `search_threads`, `get_connections`, or `search_memory`.
-- `ADD_EDGE`: add a candidate relation to the working memory graph.
-- `ANSWER`: submit the final answer as an exact node id string.
-
-## Observation Space
-
-Each step returns a JSON observation with four parts:
-
-- `tool_outputs`: the most recent tool results.
-- `graph_snapshot`: the current working-memory graph edges.
-- `action_history`: recent actions and rewards.
-- `task`: the active task id, task type, and question.
-
-## Task Types And Difficulty
-
-The benchmark mixes direct lookups with multi-hop traces:
-
-- Easy: single-hop identity resolution, organization lookup, event lookup, or location lookup.
-- Mid: two-hop alias-to-user-to-organization or thread-to-event-to-user traces.
-- High: cross-platform multi-hop traces combining aliases, authored content, event references, organization links, and direct connections.
-
-Common task families include:
-
-- `identity_resolution`
-- `network_discovery`
-- `event_tracing`
-- `cross_platform_linking`
-- `deanonymization`
-- `convoluted_trace`
-
-Expected difficulty increases with the number of relations the agent must chain together and whether the evidence is split across posts, threads, aliases, and profile edges.
-
-## Repository Layout
-
-```text
-src/osint_env/
-  agents/        single-agent and swarm runners
-  baselines/     reusable OpenAI baseline runner
-  config/        shared config and seed loading
-  data/          graph/view/task generation
-  domain/        dataclasses and environment models
-  env/           environment, reward logic, OpenEnv compatibility shim
-  eval/          evaluation metrics and leaderboard helpers
-  llm/           mock, Ollama, and OpenAI client wrappers
-  memory/        working graph and semantic memory
-  platforms/     tool APIs over synthetic platform views
-  viz/           dashboard export
-
-scripts/
-  build_fixed_levels_dataset.py
-  run_openai_baseline.py
-
-datasets/fixed_levels/
-  seed_fixed_levels.json
-  shared_config_fixed_levels.json
-  qwen_swarm_benchmark_fixed_levels.json
-
-server.py        FastAPI app for local use and Docker/HF Spaces
-Dockerfile       Container entrypoint for Hugging Face Docker Spaces
-```
-
-## Setup
-
-Python 3.10+ is required.
-
-Local install:
-
-```bash
-python -m pip install -e .
-```
-
-Run tests:
-
-```bash
-python -m pytest -q
-```
-
-Run the automated release gate:
-
-```bash
-python scripts/validate_release.py
-```
-
-## Usage
-
-Run one demo episode:
-
-```bash
-osint-env demo --agent-mode swarm --llm-provider mock
-```
-
-Run a quick evaluation:
-
-```bash
-osint-env eval --episodes 5 --agent-mode swarm --llm-provider mock
-```
-
-Export a dashboard:
-
-```bash
-osint-env benchmark --episodes 5 --agent-mode swarm --llm-provider mock --name quick_check
-```
-
-## OpenAI Baseline
-
-The reproducible OpenAI baseline is implemented in `scripts/run_openai_baseline.py`. It runs on the fixed-level benchmark, uses a stable seeded graph/task set, writes a JSON artifact, appends a leaderboard record, and exports a dashboard.
-
-Default behavior:
-
-- dataset: fixed-level benchmark
-- episodes: 30
-- max steps per episode: 8
-- temperature: 0.0
-- output artifact: `artifacts/baselines/openai_fixed_levels_latest.json`
-
-Run it with an API key:
-
-```bash
-export OPENAI_API_KEY="your_key_here"
-python scripts/run_openai_baseline.py --model gpt-5-nano
-```
-
-The script is designed to stay bounded enough for a normal benchmark pass to finish comfortably under 20 minutes on a lightweight chat model, while still using the full fixed task set. For repeatability it fixes the benchmark graph/tasks and uses deterministic decoding settings. Because remote model backends can still change over time, the output artifact also records model metadata and system fingerprints when available.
-
-## Inference Script
-
-The submission-ready inference entrypoint is the root `inference.py` file. It talks to the deployed Hugging Face Space over HTTP, uses the OpenAI client for all model calls, and emits structured stdout logs in the `[START]`, `[STEP]`, and `[END]` format.
-
-The script accepts `HF_TOKEN` as the primary auth variable and also supports `OPENAI_API_KEY` or `API_KEY` as local fallbacks.
-After a successful run, `inference.py` also posts the evaluation summary back to the Space so the latest `/dashboard` view reflects that run.
-
-Required environment variables:
-
-- `API_BASE_URL`
-- `MODEL_NAME`
-- `HF_TOKEN`
-
-Optional environment variables:
-
-- `SPACE_URL` default: `https://siddeshwar1625-osint.hf.space`
-- `TASK_INDICES` default: `0,10,20`
-- `MAX_STEPS` default: `8`
-
-Example local test command against a running local server:
-
-```bash
-API_BASE_URL=https://api.openai.com/v1 MODEL_NAME=gpt-5.4-mini OPENAI_API_KEY=your_key SPACE_URL=http://127.0.0.1:7860 python inference.py
-```
-
-Example test command against the deployed Space:
-
-```bash
-API_BASE_URL=https://api.openai.com/v1 MODEL_NAME=gpt-5.4-mini OPENAI_API_KEY=your_key SPACE_URL=https://siddeshwar1625-osint.hf.space python inference.py
-```
-
-## Docker And Hugging Face Space
-
-The repository is ready for a Docker-based Hugging Face Space:
-
-- `README.md` includes `sdk: docker`
-- `README.md` includes the `openenv` Space tag
-- `Dockerfile` serves `server.py` on port `7860`
-
-Local Docker smoke test:
-
-```bash
-docker build -t osint-openenv .
-docker run --rm -p 7860:7860 osint-openenv
-```
-
-Then open `http://localhost:7860`.
-
-The FastAPI app serves:
-
-- `/`: overview page
-- `/dashboard`: generated benchmark dashboard
-- `/api/environment`: environment metadata
-- `/health`: health check (validator-friendly alias)
-- `/healthz`: health check (legacy alias)
-- `/openenv.yaml`: OpenEnv HTTP spec stub
-- `/openenv/tasks`: task enumeration
-- `/reset` and `/openenv/reset`: episode reset endpoints
-- `/step` and `/openenv/step`: episode step endpoints
-- `/state` and `/openenv/state/{session_id}`: session state endpoints (`/state` returns the latest session)
-
-## Automated Validation
-
-The repository includes a pass/fail validation gate for the core delivery requirements:
-
-- Hugging Face Space readiness
-- OpenEnv spec compliance
-- reproducible baseline behavior
-- at least 3 fixed tasks with working graders
-- Docker image build in CI
-
-Local gate:
-
-```bash
-python scripts/validate_release.py
-```
-
-CI gate:
-
-- `.github/workflows/validation.yml`
-- runs `pytest`
-- runs the validation script
-- runs `docker build`
-
-## Baseline Scores
-
-The fixed-level benchmark was expanded from the earlier 15-question set to a 30-question set with a larger seeded graph, so older benchmark artifacts should be treated as legacy and regenerated on the new dataset before using them as reference scores.
-
-After you supply an OpenAI API key, the current baseline scores for the expanded benchmark will be written to:
-
-- `artifacts/baselines/openai_fixed_levels_latest.json`
-- `artifacts/baselines/openai_fixed_levels_dashboard.html`
-
-## Notes On `pyproject.toml`
-
-The packaging file is structurally correct for a `src/` layout and editable installs. The main gaps were deployment/runtime related rather than build-breaking:
-
-- `openenv` is now version-bounded explicitly.
-- `fastapi` and `uvicorn` are included because the repo now ships a real web server.
-- pytest is pointed at the `tests/` directory, and the test suite also adds `src/` to `sys.path` so source-layout imports work reliably during local runs.
-
-## Development Notes
-
-The project keeps a lightweight local compatibility shim for `openenv` so the source tree remains importable even before dependencies are installed. In a normal install or Docker build, the real `openenv` package from PyPI is still used.

scratch/osint/artifacts/leaderboard.json

@@ -1,453 +0,0 @@
-[
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 1,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T12:03:13+00:00",
-    "episodes": 2,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": 0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": 0.024705877237863647,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.15,
-      "avg_relation_informativeness_reward": 0.03137141693971891,
-      "avg_reward": 3.534162700533434,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8618382743087459,
-      "retrieval_signal": 0.7275,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.6082154588355165,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.25
-    },
-    "run_id": "run_0001",
-    "run_name": "swarm_seed_smoke"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 1,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T12:16:28+00:00",
-    "episodes": 2,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": 0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": 0.024705877237863647,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.15,
-      "avg_relation_informativeness_reward": 0.03137141693971891,
-      "avg_reward": 3.534162700533434,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8618382743087459,
-      "retrieval_signal": 0.7275,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.6082154588355165,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.25
-    },
-    "run_id": "run_0002",
-    "run_name": "swarm_seed_smoke"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 0,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T12:25:15+00:00",
-    "episodes": 20,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.10000000000000002,
-      "avg_connectivity_reward": 0.23999999999999994,
-      "avg_diversity_reward": 0.08000000000000002,
-      "avg_entity_informativeness_reward": -0.00983642442912193,
-      "avg_format_reward": 0.14999999999999997,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.1125,
-      "avg_relation_informativeness_reward": 0.007185245326892638,
-      "avg_reward": 3.351267560586956,
-      "avg_soft_shaping_reward": 0.14999999999999997,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8573187614039594,
-      "retrieval_signal": 0.7143750000000001,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5814697641795541,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.25
-    },
-    "run_id": "run_0003",
-    "run_name": "baseline_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 1,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T17:27:30+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": 0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": 0.06128386989162576,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.3,
-      "avg_relation_informativeness_reward": 0.12,
-      "avg_reward": 3.916035942914144,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8718832338515622,
-      "retrieval_signal": 0.78,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.6332567739783251,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.25
-    },
-    "run_id": "run_0004",
-    "run_name": "ollama_qwen_smoke"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 1,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T17:29:12+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": 0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": 0.06128386989162576,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.3,
-      "avg_relation_informativeness_reward": 0.12,
-      "avg_reward": 4.059369276247478,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.9020114237119466,
-      "retrieval_signal": 0.78,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.6332567739783251,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0005",
-    "run_name": "ollama_qwen_smoke2"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 0,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T17:39:15+00:00",
-    "episodes": 2,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": 0.0,
-      "avg_diversity_reward": 0.0683333333333333,
-      "avg_entity_informativeness_reward": -0.07397348480982455,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.6666666666666667,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.14884615384615385,
-      "avg_relation_informativeness_reward": -0.00860389783205907,
-      "avg_reward": 4.351764433970379,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6973935600514568,
-      "retrieval_signal": 0.7270961538461539,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5137345234716233,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0006",
-    "run_name": "high_timeout_shared_ctx"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 0,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T18:57:40+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.13333333333333333,
-      "avg_connectivity_reward": 0.09999999999999999,
-      "avg_diversity_reward": 0.056666666666666664,
-      "avg_entity_informativeness_reward": -0.020478979694240708,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.8148148148148149,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.27,
-      "avg_relation_informativeness_reward": 0.07174291752145656,
-      "avg_reward": 4.0269419367756605,
-      "avg_soft_shaping_reward": 0.19999999999999998,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.7366215569569294,
-      "retrieval_signal": 0.7695000000000001,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5570861208987765,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0007",
-    "run_name": "episode_selector_check"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 15,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T19:11:44+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.10000000000000002,
-      "avg_connectivity_reward": 0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.02722031691758704,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": -0.00011920119799207429,
-      "avg_reward": 3.444079221573606,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8828572592896698,
-      "retrieval_signal": 0.675,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5915320963768841,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0008",
-    "run_name": "qwen_rerun"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 15,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T19:19:34+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.10000000000000002,
-      "avg_connectivity_reward": 0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.024861029515896544,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": -0.0024320085090966614,
-      "avg_reward": 3.4441257016641917,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8828581656226586,
-      "retrieval_signal": 0.675,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5915413923950014,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0009",
-    "run_name": "qwen_episode_fix"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 15,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T19:24:37+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.10000000000000002,
-      "avg_connectivity_reward": 0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.02722031691758704,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 1.0,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": -0.0030604289114462002,
-      "avg_reward": 3.4411379938601514,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8827999009847504,
-      "retrieval_signal": 0.675,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5909438508341933,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0010",
-    "run_name": "qwen_rerun_graph_fix"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 18,
-      "max_width": 2,
-      "seed": 7,
-      "seeded_questions": 15,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T19:31:54+00:00",
-    "episodes": 15,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.16666666666666666,
-      "avg_connectivity_reward": 0.16999999999999998,
-      "avg_diversity_reward": 0.1157777777777778,
-      "avg_entity_informativeness_reward": -0.0181244777358718,
-      "avg_format_reward": 0.14999999999999997,
-      "avg_graph_f1": 0.8492063492063492,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.012000000000000002,
-      "avg_relation_informativeness_reward": 0.05935837081627929,
-      "avg_reward": 4.201760569277529,
-      "avg_soft_shaping_reward": 0.24999999999999994,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8534887252258901,
-      "retrieval_signal": 0.6792,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5847801119494148,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0011",
-    "run_name": "qwen_rerun_graph_fix"
-  }
-]

scratch/osint/config/seed_example.json

@@ -1,52 +0,0 @@
-{
-  "seeding": {
-    "seeded_nodes": [
-      {
-        "node_id": "alias_seed_001",
-        "node_type": "alias",
-        "attrs": {
-          "handle": "@shadow_seed"
-        }
-      },
-      {
-        "node_id": "user_seed_001",
-        "node_type": "user",
-        "attrs": {
-          "name": "Seed User",
-          "org": "Helios Labs",
-          "location": "Pune"
-        }
-      }
-    ],
-    "_note": "Use with --seed-file. LLM provider and API keys are configured in config/shared_config.json or CLI flags.",
-    "seeded_edges": [
-      {
-        "src": "alias_seed_001",
-        "rel": "alias_of",
-        "dst": "user_seed_001",
-        "confidence": 1.0
-      }
-    ],
-    "seeded_questions": [
-      {
-        "task_type": "identity_resolution",
-        "question": "Which canonical user owns alias alias_seed_001?",
-        "answer": "user_seed_001",
-        "supporting_edges": [
-          {
-            "src": "alias_seed_001",
-            "rel": "alias_of",
-            "dst": "user_seed_001"
-          }
-        ],
-        "metadata": {
-          "source": "manual_seed"
-        }
-      }
-    ],
-    "llm_generate_remaining_graph": true,
-    "llm_generate_remaining_tasks": true,
-    "llm_generated_edge_budget": 6,
-    "llm_generated_task_budget": 8
-  }
-}

scratch/osint/config/seed_ollama_smoke.json

@@ -1,51 +0,0 @@
-{
-  "seeding": {
-    "seeded_nodes": [
-      {
-        "node_id": "alias_smoke_001",
-        "node_type": "alias",
-        "attrs": {
-          "handle": "@smoke_alias"
-        }
-      },
-      {
-        "node_id": "user_smoke_001",
-        "node_type": "user",
-        "attrs": {
-          "name": "Smoke User",
-          "org": "Apex Dynamics",
-          "location": "Bengaluru"
-        }
-      }
-    ],
-    "seeded_edges": [
-      {
-        "src": "alias_smoke_001",
-        "rel": "alias_of",
-        "dst": "user_smoke_001",
-        "confidence": 1.0
-      }
-    ],
-    "seeded_questions": [
-      {
-        "task_type": "identity_resolution",
-        "question": "Which canonical user owns alias alias_smoke_001?",
-        "answer": "user_smoke_001",
-        "supporting_edges": [
-          {
-            "src": "alias_smoke_001",
-            "rel": "alias_of",
-            "dst": "user_smoke_001"
-          }
-        ],
-        "metadata": {
-          "source": "ollama_smoke"
-        }
-      }
-    ],
-    "llm_generate_remaining_graph": false,
-    "llm_generate_remaining_tasks": false,
-    "llm_generated_edge_budget": 0,
-    "llm_generated_task_budget": 0
-  }
-}

scratch/osint/config/shared_config.json

@@ -1,55 +0,0 @@
-{
-  "environment": {
-    "n_users": 40,
-    "alias_density": 0.35,
-    "noise_level": 0.15,
-    "red_herring_rate": 0.1,
-    "max_steps": 18,
-    "seed": 7
-  },
-  "swarm": {
-    "enabled": true,
-    "max_agents": 3,
-    "max_breadth": 2,
-    "max_width": 2,
-    "max_depth": 2,
-    "planner_rounds": 2,
-    "tools_per_agent": 1
-  },
-  "spawn_reward": {
-    "lambda_parallel": 0.15,
-    "lambda_finish": 0.2,
-    "anneal": 1.0,
-    "max_parallel_hint": 3
-  },
-  "seeding": {
-    "seeded_nodes": [],
-    "seeded_edges": [],
-    "seeded_questions": [],
-    "llm_generate_remaining_graph": true,
-    "llm_generate_remaining_tasks": true,
-    "llm_generated_edge_budget": 6,
-    "llm_generated_task_budget": 8,
-    "llm_generation_parallel": true,
-    "llm_generation_workers": 3,
-    "llm_generation_retries": 2,
-    "allow_template_fallback_on_llm_failure": false
-  },
-  "llm": {
-    "provider": "ollama",
-    "model": "qwen3:2b",
-    "temperature": 0.1,
-    "max_tokens": 256,
-    "timeout_seconds": 240,
-    "ollama_base_url": "http://127.0.0.1:11434",
-    "openai_base_url": "https://api.openai.com/v1",
-    "openai_api_key_env": "OPENAI_API_KEY",
-    "openai_api_key": ""
-  },
-  "runtime": {
-    "default_episodes": 20,
-    "leaderboard_path": "artifacts/leaderboard.json",
-    "dashboard_path": "artifacts/osint_dashboard.html",
-    "sweep_dashboard_dir": "artifacts/sweep_dashboards"
-  }
-}

scratch/osint/datasets/fixed_levels/README.md

@@ -1,64 +0,0 @@
-# Fixed Levels Submission Dataset
-
-This folder contains a fixed three-level OSINT benchmark set built on one shared base graph.
-
-## Files
-
-- `seed_fixed_levels.json`: master fixed seed with an expanded canonical graph and 30 fixed questions.
-- `fixed_graph_questions.json`: extracted fixed dataset snapshot for submission packaging.
-- `shared_config_fixed_levels.json`: run config used for generation and evaluation.
-- `complete_dataset_qwen_generated.json`: full dataset after Qwen (`qwen3:2b` via Ollama) expands the graph.
-- `qwen_swarm_eval_fixed_levels.json`: legacy Qwen swarm evaluation summary from the older smaller version of the set.
-- `qwen_swarm_benchmark_fixed_levels.json`: legacy benchmark output from the older smaller version of the set.
-- `leaderboard_fixed_levels.json`: leaderboard file for this dataset.
-- `dashboard_fixed_levels.html`: interactive dashboard generated from the benchmark run.
-
-## Difficulty Design
-
-- Easy: 10 questions. These now use the older hard-style multi-hop traces as the new floor.
-- Mid: 10 questions. Each question spans roughly 15-20 supporting nodes.
-- High: 10 questions. Each question spans roughly 50 supporting nodes.
-
-All 30 questions are fixed and share the same larger seeded graph.
-
-## Regenerate Artifacts
-
-```bash
-source ~/arl/bin/activate
-cd /home/ritish/test1
-PYTHONPATH=src python scripts/build_fixed_levels_dataset.py \
-  --seed-file datasets/fixed_levels/seed_fixed_levels.json \
-  --shared-config datasets/fixed_levels/shared_config_fixed_levels.json \
-  --output-dir datasets/fixed_levels
-```
-
-## Evaluate Qwen Swarm
-
-```bash
-source ~/arl/bin/activate
-cd /home/ritish/test1
-PYTHONPATH=src osint-env eval \
-  --config datasets/fixed_levels/shared_config_fixed_levels.json \
-  --seed-file datasets/fixed_levels/seed_fixed_levels.json \
-  --agent-mode swarm \
-  --llm-provider ollama \
-  --llm-model qwen3:2b \
-  --episodes 15
-```
-
-## Benchmark + Dashboard
-
-```bash
-source ~/arl/bin/activate
-cd /home/ritish/test1
-PYTHONPATH=src osint-env benchmark \
-  --config datasets/fixed_levels/shared_config_fixed_levels.json \
-  --seed-file datasets/fixed_levels/seed_fixed_levels.json \
-  --agent-mode swarm \
-  --llm-provider ollama \
-  --llm-model qwen3:2b \
-  --episodes 15 \
-  --name fixed_levels_qwen_swarm \
-  --leaderboard datasets/fixed_levels/leaderboard_fixed_levels.json \
-  --dashboard datasets/fixed_levels/dashboard_fixed_levels.html
-```

scratch/osint/datasets/fixed_levels/leaderboard_fixed_levels.json

@@ -1,1229 +0,0 @@
-[
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 20,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 15,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T18:48:39+00:00",
-    "episodes": 15,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.16666666666666666,
-      "avg_connectivity_reward": 0.16999999999999998,
-      "avg_diversity_reward": 0.1157777777777778,
-      "avg_entity_informativeness_reward": -0.08858065677817137,
-      "avg_format_reward": 0.14999999999999997,
-      "avg_graph_f1": 0.8492063492063492,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.052000000000000005,
-      "avg_relation_informativeness_reward": 0.07135858524047924,
-      "avg_reward": 4.197526826881651,
-      "avg_soft_shaping_reward": 0.24999999999999994,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8543934355282199,
-      "retrieval_signal": 0.6932,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5730889190257948,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0001",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-02T09:16:05+00:00",
-    "episodes": 30,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2000000000000001,
-      "avg_connectivity_reward": 0.12999999999999998,
-      "avg_diversity_reward": 0.12433333333333325,
-      "avg_entity_informativeness_reward": 0.000700571890338102,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.2916528337385394,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.05070078042510192,
-      "avg_relation_informativeness_reward": 0.07853375358885142,
-      "avg_reward": 4.377456514967488,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6241912131110795,
-      "retrieval_signal": 0.6927452731487858,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5869968650958378,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0002",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-03T13:22:03+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.20000000000000004,
-      "avg_connectivity_reward": -0.06666666666666667,
-      "avg_diversity_reward": 0.13444444444444445,
-      "avg_entity_informativeness_reward": -0.01010882862863417,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5793650793650794,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.10372960372960373,
-      "avg_relation_informativeness_reward": 0.07108687894082726,
-      "avg_reward": 4.419313576918165,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6797400780463063,
-      "retrieval_signal": 0.7113053613053614,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5356956100624386,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0003",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T18:29:39+00:00",
-    "episodes": 30,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2000000000000001,
-      "avg_connectivity_reward": 0.12999999999999998,
-      "avg_diversity_reward": 0.12433333333333325,
-      "avg_entity_informativeness_reward": -0.02515191749984708,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.2916528337385394,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.11539120363588044,
-      "avg_relation_informativeness_reward": 0.0769903534735767,
-      "avg_reward": 4.460667345528021,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6269168609961595,
-      "retrieval_signal": 0.7153869212725582,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5815176871947458,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0004",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T18:33:06+00:00",
-    "episodes": 2,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": -0.15,
-      "avg_diversity_reward": 0.13833333333333334,
-      "avg_entity_informativeness_reward": -0.026628229842114173,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.6190476190476191,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.10681818181818181,
-      "avg_relation_informativeness_reward": 0.048120982127120335,
-      "avg_reward": 4.334953339016039,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.685242999396977,
-      "retrieval_signal": 0.7123863636363637,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5075485504570012,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0005",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 1,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T18:54:52+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": -0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.02450859227728558,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.33333333333333337,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.08181818181818182,
-      "avg_relation_informativeness_reward": 0.04353540016904645,
-      "avg_reward": 3.037246438342494,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 2.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 5.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6201263424948862,
-      "retrieval_signal": 0.7036363636363637,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.45080536157835216,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0006",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 1,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T19:22:57+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": -0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.005263146336646693,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.33333333333333337,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.08181818181818182,
-      "avg_relation_informativeness_reward": 0.044276243254877785,
-      "avg_reward": 3.057232727368964,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 2.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 5.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6205293479318178,
-      "retrieval_signal": 0.7036363636363637,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.4548026193836462,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0007",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "qwen3:1.7b",
-      "llm_provider": "ollama",
-      "max_agents": 1,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T19:48:33+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.10000000000000002,
-      "avg_connectivity_reward": -0.09999999999999999,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.028683816517602444,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.15537340619307835,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.07932190760059611,
-      "avg_relation_informativeness_reward": 0.044225025032092045,
-      "avg_reward": 3.1324990406542437,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 2.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 5.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.5890485416309927,
-      "retrieval_signal": 0.7027626676602087,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5001082417028979,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0008",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "qwen3:1.7b",
-      "llm_provider": "ollama",
-      "max_agents": 1,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T19:55:08+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": -0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.005263146336646693,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.33333333333333337,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.08181818181818182,
-      "avg_relation_informativeness_reward": 0.04406984773661544,
-      "avg_reward": 3.0570263318507016,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 2.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 5.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6205251901591228,
-      "retrieval_signal": 0.7036363636363637,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.45476134027999376,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0009",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "qwen3:1.7b",
-      "llm_provider": "ollama",
-      "max_agents": 1,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T20:01:34+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.1,
-      "avg_connectivity_reward": -0.3,
-      "avg_diversity_reward": 0.08,
-      "avg_entity_informativeness_reward": -0.020826953461399098,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.33333333333333337,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.08181818181818182,
-      "avg_relation_informativeness_reward": 0.04348043923536236,
-      "avg_reward": 3.040873116224696,
-      "avg_soft_shaping_reward": 0.15,
-      "avg_spawn_count": 2.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 5.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6201995296517067,
-      "retrieval_signal": 0.7036363636363637,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.45153069715479266,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0010",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T20:46:11+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": -0.15,
-      "avg_diversity_reward": 0.12666666666666665,
-      "avg_entity_informativeness_reward": 0.019629386278697845,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5714285714285715,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.12272727272727273,
-      "avg_relation_informativeness_reward": 0.08347928023822283,
-      "avg_reward": 1.829702015111513,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6715432845394145,
-      "retrieval_signal": 0.7179545454545455,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5221217333033842,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0011",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T20:49:44+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": -0.15,
-      "avg_diversity_reward": 0.12666666666666665,
-      "avg_entity_informativeness_reward": 0.019629386278697845,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5714285714285715,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.12272727272727273,
-      "avg_relation_informativeness_reward": 0.08335372627068136,
-      "avg_reward": 0.7139904233885594,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6641542345113342,
-      "retrieval_signal": 0.7179545454545455,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5220966225098759,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0012",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-06T20:59:43+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": -0.15,
-      "avg_diversity_reward": 0.12666666666666665,
-      "avg_entity_informativeness_reward": 0.0036675120354726642,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5714285714285715,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.12272727272727273,
-      "avg_relation_informativeness_reward": 0.08250745620050208,
-      "avg_reward": 0.7138056720677886,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6638424503476543,
-      "retrieval_signal": 0.7179545454545455,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.518734993647195,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0013",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T09:44:40+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": -0.15,
-      "avg_diversity_reward": 0.12666666666666665,
-      "avg_entity_informativeness_reward": -0.018704290877944903,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5714285714285715,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.12272727272727273,
-      "avg_relation_informativeness_reward": 0.08056039127695382,
-      "avg_reward": 0.7135379106634446,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6633913226563717,
-      "retrieval_signal": 0.7179545454545455,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5138712200798018,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0014",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T09:55:19+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": -0.15,
-      "avg_diversity_reward": 0.12666666666666665,
-      "avg_entity_informativeness_reward": -0.018704290877944903,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5714285714285715,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.12272727272727273,
-      "avg_relation_informativeness_reward": 0.08056039127695382,
-      "avg_reward": 0.7135379106634446,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6633913226563717,
-      "retrieval_signal": 0.7179545454545455,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5138712200798018,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0015",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T09:56:28+00:00",
-    "episodes": 30,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2000000000000001,
-      "avg_connectivity_reward": 0.12999999999999998,
-      "avg_diversity_reward": 0.12433333333333325,
-      "avg_entity_informativeness_reward": -0.02515191749984708,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.2916528337385394,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.11539120363588044,
-      "avg_relation_informativeness_reward": 0.0769903534735767,
-      "avg_reward": 0.7150555461096118,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6132407715455404,
-      "retrieval_signal": 0.7153869212725582,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5815176871947458,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0016",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T10:02:32+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.2,
-      "avg_connectivity_reward": -0.15,
-      "avg_diversity_reward": 0.12666666666666665,
-      "avg_entity_informativeness_reward": -0.018704290877944903,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5714285714285715,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.12272727272727273,
-      "avg_relation_informativeness_reward": 0.08056039127695382,
-      "avg_reward": 0.7135379106634446,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6633913226563717,
-      "retrieval_signal": 0.7179545454545455,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5138712200798018,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0017",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T10:02:49+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.20000000000000004,
-      "avg_connectivity_reward": -0.06666666666666667,
-      "avg_diversity_reward": 0.13444444444444445,
-      "avg_entity_informativeness_reward": -0.029992009599206938,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5793650793650794,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.10372960372960373,
-      "avg_relation_informativeness_reward": 0.06898843512226,
-      "avg_reward": 0.7133699465240085,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6656078661080486,
-      "retrieval_signal": 0.7113053613053614,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5312992851046106,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0018",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T10:04:53+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.20000000000000004,
-      "avg_connectivity_reward": -0.06666666666666667,
-      "avg_diversity_reward": 0.13444444444444445,
-      "avg_entity_informativeness_reward": -0.029992009599206938,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5793650793650794,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.10372960372960373,
-      "avg_relation_informativeness_reward": 0.06898843512226,
-      "avg_reward": 0.7133699465240085,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6656078661080486,
-      "retrieval_signal": 0.7113053613053614,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5312992851046106,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0019",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T10:11:34+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.20000000000000004,
-      "avg_connectivity_reward": -0.06666666666666667,
-      "avg_diversity_reward": 0.13444444444444445,
-      "avg_entity_informativeness_reward": -0.029992009599206938,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.5793650793650794,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.10372960372960373,
-      "avg_relation_informativeness_reward": 0.06898843512226,
-      "avg_reward": 0.7133699465240085,
-      "avg_soft_shaping_reward": 0.3,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.6656078661080486,
-      "retrieval_signal": 0.7113053613053614,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5312992851046106,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0020",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T10:29:54+00:00",
-    "episodes": 3,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.0,
-      "avg_connectivity_reward": 0.0,
-      "avg_diversity_reward": 0.0,
-      "avg_entity_informativeness_reward": 0.0,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.0,
-      "avg_knowledge_carrier_reward": 0.0,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": 0.0,
-      "avg_reward": 0.5519400198339021,
-      "avg_soft_shaping_reward": 0.0,
-      "avg_spawn_count": 0.0,
-      "avg_spawn_critical_steps": 0.0,
-      "avg_steps_to_solution": 1.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.2785970009916951,
-      "retrieval_signal": 0.5,
-      "spawn_completion_rate": 0.0,
-      "spawn_signal": 0.4,
-      "structural_signal": 0.5,
-      "task_success_rate": 0.0,
-      "tool_efficiency": 1.0
-    },
-    "run_id": "run_0021",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-07T15:59:20+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.0,
-      "avg_connectivity_reward": 0.0,
-      "avg_diversity_reward": 0.0,
-      "avg_entity_informativeness_reward": 0.0,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.0,
-      "avg_knowledge_carrier_reward": 0.0,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": 0.0,
-      "avg_reward": 0.5519400198339021,
-      "avg_soft_shaping_reward": 0.0,
-      "avg_spawn_count": 0.0,
-      "avg_spawn_critical_steps": 0.0,
-      "avg_steps_to_solution": 1.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.2785970009916951,
-      "retrieval_signal": 0.5,
-      "spawn_completion_rate": 0.0,
-      "spawn_signal": 0.4,
-      "structural_signal": 0.5,
-      "task_success_rate": 0.0,
-      "tool_efficiency": 1.0
-    },
-    "run_id": "run_0022",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-08T04:25:00+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.0,
-      "avg_connectivity_reward": 0.0,
-      "avg_diversity_reward": 0.0,
-      "avg_entity_informativeness_reward": 0.0,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.0,
-      "avg_knowledge_carrier_reward": 0.0,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": 0.0,
-      "avg_reward": 0.5519400198339021,
-      "avg_soft_shaping_reward": 0.0,
-      "avg_spawn_count": 0.0,
-      "avg_spawn_critical_steps": 0.0,
-      "avg_steps_to_solution": 1.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.2785970009916951,
-      "retrieval_signal": 0.5,
-      "spawn_completion_rate": 0.0,
-      "spawn_signal": 0.4,
-      "structural_signal": 0.5,
-      "task_success_rate": 0.0,
-      "tool_efficiency": 1.0
-    },
-    "run_id": "run_0023",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-08T04:28:07+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.0,
-      "avg_connectivity_reward": 0.0,
-      "avg_diversity_reward": 0.0,
-      "avg_entity_informativeness_reward": 0.0,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.0,
-      "avg_knowledge_carrier_reward": 0.0,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": 0.0,
-      "avg_reward": 0.5519400198339021,
-      "avg_soft_shaping_reward": 0.0,
-      "avg_spawn_count": 0.0,
-      "avg_spawn_critical_steps": 0.0,
-      "avg_steps_to_solution": 1.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.2785970009916951,
-      "retrieval_signal": 0.5,
-      "spawn_completion_rate": 0.0,
-      "spawn_signal": 0.4,
-      "structural_signal": 0.5,
-      "task_success_rate": 0.0,
-      "tool_efficiency": 1.0
-    },
-    "run_id": "run_0024",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-08T04:39:32+00:00",
-    "episodes": 1,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.0,
-      "avg_connectivity_reward": 0.0,
-      "avg_diversity_reward": 0.0,
-      "avg_entity_informativeness_reward": 0.0,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.0,
-      "avg_knowledge_carrier_reward": 0.0,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": 0.0,
-      "avg_reward": 0.5519400198339021,
-      "avg_soft_shaping_reward": 0.0,
-      "avg_spawn_count": 0.0,
-      "avg_spawn_critical_steps": 0.0,
-      "avg_steps_to_solution": 1.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.2785970009916951,
-      "retrieval_signal": 0.5,
-      "spawn_completion_rate": 0.0,
-      "spawn_signal": 0.4,
-      "structural_signal": 0.5,
-      "task_success_rate": 0.0,
-      "tool_efficiency": 1.0
-    },
-    "run_id": "run_0025",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-08T04:40:21+00:00",
-    "episodes": 30,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.13333333333333336,
-      "avg_connectivity_reward": 0.09999999999999999,
-      "avg_diversity_reward": 0.03911111111111111,
-      "avg_entity_informativeness_reward": -0.00951758755541623,
-      "avg_format_reward": 0.15,
-      "avg_graph_f1": 0.08482743691314255,
-      "avg_knowledge_carrier_reward": 0.3333333333333333,
-      "avg_knowledge_indexing_reward": 0.0832325289772058,
-      "avg_relation_informativeness_reward": 0.024842289016879314,
-      "avg_reward": 0.6636425017249088,
-      "avg_soft_shaping_reward": 0.19999999999999993,
-      "avg_spawn_count": 2.6666666666666665,
-      "avg_spawn_critical_steps": 4.0,
-      "avg_steps_to_solution": 6.333333333333333,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.4644798510150634,
-      "retrieval_signal": 0.6457980518086888,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.7,
-      "structural_signal": 0.5472649402922927,
-      "task_success_rate": 0.6666666666666666,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0026",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-08T05:01:16+00:00",
-    "episodes": 10,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.0,
-      "avg_connectivity_reward": 0.0,
-      "avg_diversity_reward": 0.0,
-      "avg_entity_informativeness_reward": 0.0,
-      "avg_format_reward": 0.14999999999999997,
-      "avg_graph_f1": 0.0,
-      "avg_knowledge_carrier_reward": 0.0,
-      "avg_knowledge_indexing_reward": 0.0,
-      "avg_relation_informativeness_reward": 0.0,
-      "avg_reward": 0.5519400198339021,
-      "avg_soft_shaping_reward": 0.0,
-      "avg_spawn_count": 0.0,
-      "avg_spawn_critical_steps": 0.0,
-      "avg_steps_to_solution": 1.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.2785970009916951,
-      "retrieval_signal": 0.5,
-      "spawn_completion_rate": 0.0,
-      "spawn_signal": 0.4,
-      "structural_signal": 0.5,
-      "task_success_rate": 0.0,
-      "tool_efficiency": 1.0
-    },
-    "run_id": "run_0027",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-08T05:01:29+00:00",
-    "episodes": 10,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.19999999999999998,
-      "avg_connectivity_reward": 0.06,
-      "avg_diversity_reward": 0.0,
-      "avg_entity_informativeness_reward": 0.0,
-      "avg_format_reward": 0.14999999999999997,
-      "avg_graph_f1": 0.18535980927285275,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.09575879120879122,
-      "avg_relation_informativeness_reward": 0.0,
-      "avg_reward": 0.7109638031154166,
-      "avg_soft_shaping_reward": 0.29999999999999993,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.5866289994462388,
-      "retrieval_signal": 0.708515576923077,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.535,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0028",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  {
-    "config": {
-      "llm_model": "gpt-5.4-mini",
-      "llm_provider": "openai",
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 24,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 30,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-08T05:01:43+00:00",
-    "episodes": 10,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.19999999999999998,
-      "avg_connectivity_reward": 0.24,
-      "avg_diversity_reward": 0.11733333333333333,
-      "avg_entity_informativeness_reward": -0.028552762666248687,
-      "avg_format_reward": 0.14999999999999997,
-      "avg_graph_f1": 0.06912250146657492,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.15393879572282626,
-      "avg_relation_informativeness_reward": 0.07452686705063795,
-      "avg_reward": 0.7171006884027153,
-      "avg_soft_shaping_reward": 0.29999999999999993,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 0.0,
-      "leaderboard_score": 0.5730007362494549,
-      "retrieval_signal": 0.7288785785029892,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.6067948208768779,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0029",
-    "run_name": "fixed_levels_qwen_swarm"
-  }
-]

scratch/osint/datasets/fixed_levels/qwen_swarm_benchmark_fixed_levels.json

@@ -1,69 +0,0 @@
-{
-  "dashboard": "datasets/fixed_levels/dashboard_fixed_levels.html",
-  "record": {
-    "config": {
-      "max_agents": 3,
-      "max_breadth": 2,
-      "max_depth": 2,
-      "max_steps": 20,
-      "max_width": 2,
-      "seed": 2026,
-      "seeded_questions": 15,
-      "swarm_enabled": true
-    },
-    "created_at": "2026-04-01T18:48:39+00:00",
-    "episodes": 15,
-    "metrics": {
-      "avg_compactness_reward": 0.0,
-      "avg_connectivity_gain_reward": 0.16666666666666666,
-      "avg_connectivity_reward": 0.16999999999999998,
-      "avg_diversity_reward": 0.1157777777777778,
-      "avg_entity_informativeness_reward": -0.08858065677817137,
-      "avg_format_reward": 0.14999999999999997,
-      "avg_graph_f1": 0.8492063492063492,
-      "avg_knowledge_carrier_reward": 0.5,
-      "avg_knowledge_indexing_reward": 0.052000000000000005,
-      "avg_relation_informativeness_reward": 0.07135858524047924,
-      "avg_reward": 4.197526826881651,
-      "avg_soft_shaping_reward": 0.24999999999999994,
-      "avg_spawn_count": 4.0,
-      "avg_spawn_critical_steps": 6.0,
-      "avg_steps_to_solution": 9.0,
-      "deanonymization_accuracy": 1.0,
-      "leaderboard_score": 0.8543934355282199,
-      "retrieval_signal": 0.6932,
-      "spawn_completion_rate": 1.0,
-      "spawn_signal": 0.6666666666666666,
-      "structural_signal": 0.5730889190257948,
-      "task_success_rate": 1.0,
-      "tool_efficiency": 0.5
-    },
-    "run_id": "run_0001",
-    "run_name": "fixed_levels_qwen_swarm"
-  },
-  "summary": {
-    "avg_compactness_reward": 0.0,
-    "avg_connectivity_gain_reward": 0.16666666666666666,
-    "avg_connectivity_reward": 0.16999999999999998,
-    "avg_diversity_reward": 0.1157777777777778,
-    "avg_entity_informativeness_reward": -0.08858065677817137,
-    "avg_format_reward": 0.14999999999999997,
-    "avg_graph_f1": 0.8492063492063492,
-    "avg_knowledge_carrier_reward": 0.5,
-    "avg_knowledge_indexing_reward": 0.052000000000000005,
-    "avg_relation_informativeness_reward": 0.07135858524047924,
-    "avg_reward": 4.197526826881651,
-    "avg_soft_shaping_reward": 0.24999999999999994,
-    "avg_spawn_count": 4.0,
-    "avg_spawn_critical_steps": 6.0,
-    "avg_steps_to_solution": 9.0,
-    "deanonymization_accuracy": 1.0,
-    "leaderboard_score": 0.8543934355282199,
-    "retrieval_signal": 0.6932,
-    "spawn_completion_rate": 1.0,
-    "spawn_signal": 0.6666666666666666,
-    "structural_signal": 0.5730889190257948,
-    "task_success_rate": 1.0,
-    "tool_efficiency": 0.5
-  }
-}

scratch/osint/datasets/fixed_levels/qwen_swarm_eval_by_difficulty.json

@@ -1,53 +0,0 @@
-{
-  "by_difficulty": {
-    "easy": {
-      "avg_graph_f1": 1.0,
-      "avg_reward": 3.610490808845623,
-      "avg_steps": 9.0,
-      "avg_tool_calls": 4.0,
-      "episodes": 5,
-      "task_success_rate": 1.0
-    },
-    "high": {
-      "avg_graph_f1": 0.5476190476190477,
-      "avg_reward": 4.207102815893519,
-      "avg_steps": 9.0,
-      "avg_tool_calls": 4.0,
-      "episodes": 5,
-      "task_success_rate": 1.0
-    },
-    "mid": {
-      "avg_graph_f1": 1.0,
-      "avg_reward": 4.822687547070801,
-      "avg_steps": 9.0,
-      "avg_tool_calls": 4.0,
-      "episodes": 5,
-      "task_success_rate": 1.0
-    }
-  },
-  "overall": {
-    "avg_compactness_reward": 0.0,
-    "avg_connectivity_gain_reward": 0.16666666666666666,
-    "avg_connectivity_reward": 0.16999999999999998,
-    "avg_diversity_reward": 0.1157777777777778,
-    "avg_entity_informativeness_reward": -0.07289878447762359,
-    "avg_format_reward": 0.14999999999999997,
-    "avg_graph_f1": 0.8492063492063492,
-    "avg_knowledge_carrier_reward": 0.5,
-    "avg_knowledge_indexing_reward": 0.052000000000000005,
-    "avg_relation_informativeness_reward": 0.07157694332826091,
-    "avg_reward": 4.213427057269981,
-    "avg_soft_shaping_reward": 0.24999999999999994,
-    "avg_spawn_count": 4.0,
-    "avg_spawn_critical_steps": 6.0,
-    "avg_steps_to_solution": 9.0,
-    "deanonymization_accuracy": 1.0,
-    "leaderboard_score": 0.8546911504342771,
-    "retrieval_signal": 0.6932,
-    "spawn_completion_rate": 1.0,
-    "spawn_signal": 0.6666666666666666,
-    "structural_signal": 0.5762689651034608,
-    "task_success_rate": 1.0,
-    "tool_efficiency": 0.5
-  }
-}

scratch/osint/datasets/fixed_levels/qwen_swarm_eval_fixed_levels.json

@@ -1,25 +0,0 @@
-{
-  "avg_compactness_reward": 0.0,
-  "avg_connectivity_gain_reward": 0.16666666666666666,
-  "avg_connectivity_reward": 0.16999999999999998,
-  "avg_diversity_reward": 0.1157777777777778,
-  "avg_entity_informativeness_reward": -0.02824631570420193,
-  "avg_format_reward": 0.14999999999999997,
-  "avg_graph_f1": 0.8492063492063492,
-  "avg_knowledge_carrier_reward": 0.5,
-  "avg_knowledge_indexing_reward": 0.07400000000000001,
-  "avg_relation_informativeness_reward": 0.06905976285357758,
-  "avg_reward": 4.285384567790942,
-  "avg_soft_shaping_reward": 0.24999999999999994,
-  "avg_spawn_count": 4.0,
-  "avg_spawn_critical_steps": 6.0,
-  "avg_steps_to_solution": 9.0,
-  "deanonymization_accuracy": 1.0,
-  "leaderboard_score": 0.8565775118852701,
-  "retrieval_signal": 0.7009000000000001,
-  "spawn_completion_rate": 1.0,
-  "spawn_signal": 0.6666666666666666,
-  "structural_signal": 0.5846960227632085,
-  "task_success_rate": 1.0,
-  "tool_efficiency": 0.5
-}

scratch/osint/datasets/fixed_levels/shared_config_fixed_levels.json

@@ -1,55 +0,0 @@
-{
-  "environment": {
-    "n_users": 24,
-    "alias_density": 0.2,
-    "noise_level": 0.12,
-    "red_herring_rate": 0.08,
-    "max_steps": 24,
-    "seed": 2026
-  },
-  "swarm": {
-    "enabled": true,
-    "max_agents": 3,
-    "max_breadth": 2,
-    "max_width": 2,
-    "max_depth": 2,
-    "planner_rounds": 2,
-    "tools_per_agent": 1
-  },
-  "spawn_reward": {
-    "lambda_parallel": 0.15,
-    "lambda_finish": 0.2,
-    "anneal": 1.0,
-    "max_parallel_hint": 3
-  },
-  "seeding": {
-    "seeded_nodes": [],
-    "seeded_edges": [],
-    "seeded_questions": [],
-    "llm_generate_remaining_graph": true,
-    "llm_generate_remaining_tasks": false,
-    "llm_generated_edge_budget": 64,
-    "llm_generated_task_budget": 0,
-    "llm_generation_parallel": true,
-    "llm_generation_workers": 4,
-    "llm_generation_retries": 3,
-    "allow_template_fallback_on_llm_failure": false
-  },
-  "llm": {
-    "provider": "ollama",
-    "model": "qwen3:2b",
-    "temperature": 0.05,
-    "max_tokens": 384,
-    "timeout_seconds": 240,
-    "ollama_base_url": "http://127.0.0.1:11434",
-    "openai_base_url": "https://api.openai.com/v1",
-    "openai_api_key_env": "OPENAI_API_KEY",
-    "openai_api_key": ""
-  },
-  "runtime": {
-    "default_episodes": 30,
-    "leaderboard_path": "datasets/fixed_levels/leaderboard_fixed_levels.json",
-    "dashboard_path": "datasets/fixed_levels/dashboard_fixed_levels.html",
-    "sweep_dashboard_dir": "datasets/fixed_levels/sweep_dashboards"
-  }
-}

scratch/osint/docs/reward_design_notes.md

@@ -1,94 +0,0 @@
-# Reward Design Notes
-
-This environment uses a composite reward that adapts ideas from:
-
-- AutoGraph-R1 (arXiv:2510.15339)
-- UniRel (arXiv:2512.17043)
-- DeepPath (EMNLP 2017, D17-1060)
-- Multi-Hop KG Reasoning with Reward Shaping (EMNLP 2018, D18-1362)
-- Kimi K2.5 (arXiv:2602.02276) for PARL-style swarm auxiliary shaping
-
-Additional related context consulted:
-
-- MINERVA (arXiv:1711.05851) for query-conditioned walk-style reasoning over KG paths.
-
-## Components in this Branch
-
-The implementation follows a staged reward design:
-
-1. edge-level rewards during graph construction (`ADD_EDGE`)
-2. answer-level rewards for retrieval usefulness and final task utility (`ANSWER`)
-3. evaluation-level composite leaderboard score for benchmark ranking
-
-### 1) Edge addition reward
-
-For each `ADD_EDGE`, the reward combines:
-
-- Global accuracy term (DeepPath):
-  - $r_{global} = +1$ if a candidate edge is correct, else $-1$ (scaled in code for stability).
-- Soft shaping term (D18 reward shaping):
-  - $R = R_b + (1 - R_b) f(s, r, o)$, where $f$ is a soft fact plausibility score.
-  - In code, $f$ is approximated by relation/type priors plus small domain priors.
-- Efficiency term (DeepPath):
-  - $r_{efficiency} \propto 1 / \text{step\_count}$.
-- Diversity term (DeepPath):
-  - novelty from cosine dissimilarity of edge signatures; repeated patterns are down-weighted.
-- Relation/entity informativeness (UniRel):
-  - relation rarity via normalized IDF of relation labels,
-  - entity informativeness via inverse hub-penalty.
-- Connectivity gain term:
-  - rewards bridge edges that connect previously disconnected graph regions.
-
-### 2) Final answer reward
-
-For `ANSWER`, the reward combines:
-
-- format validity,
-- answer correctness,
-- knowledge-carrying utility (AutoGraph-R1 style):
-  - $R_C(q, y, G) = \mathbb{{I}}[\text{{deducible}}(q, y \mid G)]$.
-- knowledge-indexing utility (AutoGraph-R1 style):
-  - $R_I(q, D_{{gold}}, G) = |Top\text{{-}}k(G,q) \cap D_{{gold}}| / |D_{{gold}}|$,
-  - approximated in this environment with evidence recall over tool outputs.
-- connectivity (UniRel style):
-  - discrete connectivity reward over extracted seed entities, normalized for stable mixing.
-- graph F1 against supporting edges,
-- compactness penalty for unnecessary extra edges,
-- efficiency bonus,
-- relation/entity informativeness for the constructed subgraph,
-- repetition penalty to discourage redundant relation generation patterns.
-
-UniRel-style aggregate view represented in this branch:
-
-$$
-R(a) \approx R_{{fmt}} + R_{{con}} + w_1 R_{{ent}} + w_2 R_{{rel}} + \text{{task utility terms}}
-$$
-
-with task utility terms coming from AutoGraph-inspired $R_C$ and $R_I$ components.
-
-## Telemetry
-
-Per-step component rewards are aggregated into `info["reward_components"]`, enabling:
-
-- richer benchmark summaries,
-- leaderboard ranking by composite utility,
-- visual diagnostics in dashboard exports.
-
-Evaluation also computes derived retrieval and structural utility signals used in leaderboard ranking.
-
-## Future Multi-Agent Notes
-
-This branch now includes a low-width swarm baseline orchestrator that adds PARL-style auxiliary shaping on top of the core edge and answer rewards.
-
-The helper implementation is in:
-
-- `src/osint_env/env/spawn_reward_hooks.py`
-
-It follows the Kimi K2.5 style decomposition:
-
-- $r_{{PARL}}(x,y) = r_{{perf}}(x,y) + \lambda_1 r_{{parallel}} + \lambda_2 r_{{finish}}$,
-- optional critical-steps shaping for latency-sensitive training,
-- optional annealing of $\lambda_1, \lambda_2$ toward zero,
-- optional breadth/depth shaping hooks for future branch integration.
-
-The expanded project-level walkthrough is in `README.md` under "Reward Design (Integrated Notes)".

scratch/osint/inference.py

@@ -1,509 +0,0 @@
-from __future__ import annotations
-
-import json
-import os
-from pathlib import Path
-from typing import Any
-
-from osint_env.agents.single_agent import SingleAgentRunner
-from osint_env.agents.swarm_agent import SwarmAgentRunner
-from osint_env.config import clone_environment_config, load_seeding_config, load_shared_config
-from osint_env.domain.models import EnvironmentConfig
-from osint_env.env.environment import OSINTEnvironment
-from osint_env.env.reward import compute_graph_f1
-from osint_env.eval.leaderboard import append_leaderboard_record, load_leaderboard
-from osint_env.eval.metrics import EvalMetrics
-from osint_env.llm import build_llm_client
-from osint_env.viz import export_dashboard
-
-
-CONFIG_PATH = os.getenv("CONFIG_PATH", "datasets/fixed_levels/shared_config_fixed_levels.json")
-SEED_FILE = os.getenv("SEED_FILE", "datasets/fixed_levels/seed_fixed_levels.json")
-AGENT_MODE = os.getenv("AGENT_MODE", "swarm")
-LLM_PROVIDER = os.getenv("LLM_PROVIDER", "openai")
-MODEL_NAME = os.getenv("MODEL_NAME", "gpt-5.4")
-OLLAMA_BASE_URL = os.getenv("OLLAMA_BASE_URL", "")
-OPENAI_BASE_URL = os.getenv("OPENAI_BASE_URL", "")
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
-OPENAI_API_KEY_ENV = os.getenv("OPENAI_API_KEY_ENV", "OPENAI_API_KEY")
-API_BASE_URL = os.getenv("API_BASE_URL", "https://api.openai.com/v1")
-API_KEY = os.getenv("API_KEY", "")
-HF_SPACE_URL = os.getenv("HF_SPACE_URL", "")
-HF_TOKEN = os.getenv("HF_TOKEN","")
-LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME", "")
-LLM_TIMEOUT_SECONDS = int(os.getenv("LLM_TIMEOUT_SECONDS", "0"))
-EPISODES = int(os.getenv("EPISODES", "1"))
-SUCCESS_SCORE_THRESHOLD = float(os.getenv("SUCCESS_SCORE_THRESHOLD", "0.67"))
-TASK_INDICES_RAW = os.getenv("TASK_INDICES", "")
-
-WRITE_BENCHMARK_ARTIFACTS = os.getenv("WRITE_BENCHMARK_ARTIFACTS", "1").strip().lower() in {
-    "1",
-    "true",
-    "yes",
-    "y",
-    "on",
-}
-LEADERBOARD_PATH = os.getenv("LEADERBOARD_PATH", "datasets/fixed_levels/leaderboard_fixed_levels.json")
-DASHBOARD_PATH = os.getenv("DASHBOARD_PATH", "datasets/fixed_levels/dashboard_fixed_levels.html")
-RUN_NAME = os.getenv("RUN_NAME", "fixed_levels_qwen_swarm")
-
-BENCHMARK = "osint-openenv"
-TASK_NAME = "fixed_levels_easy_mid_hard"
-
-
-def _parse_task_indices(raw: str) -> list[int]:
-    out: list[int] = []
-    for token in str(raw or "").split(","):
-        stripped = token.strip()
-        if not stripped:
-            continue
-        try:
-            out.append(int(stripped))
-        except ValueError:
-            continue
-    return out
-
-
-def _normalize_ollama_base_url(url: str) -> str:
-    normalized = str(url or "").strip().rstrip("/")
-    if normalized.endswith("/v1"):
-        normalized = normalized[:-3].rstrip("/")
-    return normalized or "http://127.0.0.1:11434"
-
-
-def _normalize_openai_base_url(url: str) -> str:
-    normalized = str(url or "").strip().rstrip("/")
-    if not normalized:
-        return ""
-    if normalized.endswith("/v1"):
-        return normalized
-    return f"{normalized}/v1"
-
-
-TASK_INDICES = _parse_task_indices(TASK_INDICES_RAW)
-
-
-def log_start(task: str, env: str, model: str) -> None:
-    print(f"[START] task={task} env={env} model={model}", flush=True)
-
-
-def log_step(step: int, action: str, reward: float, done: bool, error: str | None) -> None:
-    error_text = "null" if error is None else str(error)
-    print(
-        f"[STEP] step={step} action={action} reward={reward:.2f} done={str(bool(done)).lower()} error={error_text}",
-        flush=True,
-    )
-
-
-def log_end(task: str, success: bool, steps: int, score: float, rewards: list[float]) -> None:
-    rewards_text = ",".join(f"{value:.2f}" for value in rewards)
-    print(
-        f"[END] success={str(bool(success)).lower()} steps={steps} score={score:.2f} rewards={rewards_text}",
-        flush=True,
-    )
-
-
-def _looks_like_placeholder_api_key(value: str) -> bool:
-    token = str(value or "").strip().lower()
-    if not token:
-        return True
-    placeholder_markers = [
-        "your_openai_api_key",
-        "your-key",
-        "your_key",
-        "your real",
-        "real-openai-key",
-        "replace-me",
-        "changeme",
-        "example",
-        "<api-key>",
-    ]
-    if token.startswith("your_") or token.startswith("sk-your-"):
-        return True
-    return any(marker in token for marker in placeholder_markers)
-
-
-def _format_action(action: dict[str, Any]) -> str:
-    action_type = str(action.get("action_type", "")).upper()
-    payload = dict(action.get("payload", {}))
-
-    if action_type == "ANSWER":
-        return f"answer({str(payload.get('answer', 'unknown')).strip()})"
-
-    if action_type == "ADD_EDGE":
-        try:
-            conf = float(payload.get("confidence", 1.0))
-        except (TypeError, ValueError):
-            conf = 1.0
-        return (
-            "add_edge("
-            f"{payload.get('src', '')},"
-            f"{payload.get('rel', '')},"
-            f"{payload.get('dst', '')},"
-            f"{conf:.2f}"
-            ")"
-        )
-
-    tool_name = str(payload.get("tool_name", "tool")).strip() or "tool"
-    args = payload.get("args", {})
-    if not isinstance(args, dict) or not args:
-        return f"{tool_name}()"
-    args_text = ",".join(f"{key}={value}" for key, value in sorted(args.items()))
-    return f"{tool_name}({args_text})"
-
-
-def _assistant_tool_call_id(message: dict[str, Any]) -> str | None:
-    tool_calls = list(message.get("tool_calls", []))
-    if not tool_calls:
-        return None
-    tool_call_id = tool_calls[0].get("id")
-    return str(tool_call_id) if tool_call_id else None
-
-
-def _tool_result_message(assistant_message: dict[str, Any], result: dict[str, Any]) -> dict[str, Any] | None:
-    tool_call_id = _assistant_tool_call_id(assistant_message)
-    if not tool_call_id:
-        return None
-    return {
-        "role": "tool",
-        "tool_call_id": tool_call_id,
-        "content": json.dumps(result, sort_keys=True),
-    }
-
-
-def _resolve_environment_config() -> EnvironmentConfig:
-    shared = load_shared_config(CONFIG_PATH)
-    env_cfg = clone_environment_config(shared.environment)
-
-    if SEED_FILE and Path(SEED_FILE).exists():
-        env_cfg.seeding = load_seeding_config(SEED_FILE)
-
-    mode = AGENT_MODE.strip().lower()
-    if mode == "single":
-        env_cfg.swarm.enabled = False
-    elif mode == "swarm":
-        env_cfg.swarm.enabled = True
-
-    # Inference submissions must route all calls through OpenAI-compatible client config.
-    env_cfg.llm.provider = "openai"
-    env_cfg.llm.model = MODEL_NAME.strip()
-
-    if LLM_TIMEOUT_SECONDS > 0:
-        env_cfg.llm.timeout_seconds = int(LLM_TIMEOUT_SECONDS)
-
-    # Evaluation harnesses inject API_BASE_URL/HF_TOKEN for proxy-enforced requests.
-    resolved_openai_base = API_BASE_URL.strip() or OPENAI_BASE_URL.strip() or HF_SPACE_URL.strip()
-    if resolved_openai_base:
-        env_cfg.llm.openai_base_url = _normalize_openai_base_url(resolved_openai_base)
-
-    if HF_TOKEN.strip():
-        env_cfg.llm.openai_api_key = HF_TOKEN.strip()
-    elif API_KEY.strip():
-        env_cfg.llm.openai_api_key = API_KEY.strip()
-    elif OPENAI_API_KEY.strip():
-        env_cfg.llm.openai_api_key = OPENAI_API_KEY.strip()
-
-    if OPENAI_API_KEY_ENV.strip():
-        env_cfg.llm.openai_api_key_env = OPENAI_API_KEY_ENV.strip()
-
-    return env_cfg
-
-
-def _runner_for(env: OSINTEnvironment, llm: Any) -> SingleAgentRunner | SwarmAgentRunner:
-    if env.config.swarm.enabled:
-        return SwarmAgentRunner(env=env, llm=llm)
-    return SingleAgentRunner(env=env, llm=llm)
-
-
-def _normalize_difficulty(value: str) -> str:
-    token = str(value or "").strip().lower()
-    if token in {"easy", "e"}:
-        return "easy"
-    if token in {"mid", "medium", "m"}:
-        return "medium"
-    if token in {"high", "hard", "h"}:
-        return "hard"
-    return "hard"
-
-
-def _task_difficulty(env: OSINTEnvironment, task_index: int) -> str:
-    idx = int(task_index) % max(1, len(env.tasks))
-    task = env.tasks[idx]
-    if isinstance(task.metadata, dict) and "difficulty" in task.metadata:
-        return _normalize_difficulty(str(task.metadata.get("difficulty", "")))
-    if idx < 10:
-        return "easy"
-    if idx < 20:
-        return "medium"
-    return "hard"
-
-
-def _episode_row(env: OSINTEnvironment, info: dict[str, Any]) -> dict[str, Any]:
-    if env.state is None:
-        return {
-            "task_id": "unknown",
-            "task_type": "unknown",
-            "question": "",
-            "task_answer": str(info.get("task_answer", "")),
-            "agent_answer": str(info.get("agent_answer", "")),
-            "graph_f1": 0.0,
-            "reward": float(info.get("total_reward", 0.0) or 0.0),
-            "steps": int(info.get("step_count", 0) or 0),
-            "tool_calls": int(info.get("tool_calls", 0) or 0),
-            "success": int(info.get("agent_answer") == info.get("task_answer")),
-            "reward_components": dict(info.get("reward_components", {})),
-            "pred_edges": [],
-            "truth_edges": [],
-        }
-
-    graph_f1 = compute_graph_f1(env.memory_graph.edges, env.state.task.supporting_edges)
-    return {
-        "task_id": env.state.task.task_id,
-        "task_type": env.state.task.task_type,
-        "question": env.state.task.question,
-        "task_answer": str(info.get("task_answer", "")),
-        "agent_answer": str(info.get("agent_answer", "")) if info.get("agent_answer") is not None else "",
-        "graph_f1": graph_f1,
-        "reward": float(info.get("total_reward", 0.0) or 0.0),
-        "steps": int(info.get("step_count", 0) or 0),
-        "tool_calls": int(info.get("tool_calls", 0) or 0),
-        "success": int(info.get("agent_answer") == info.get("task_answer")),
-        "reward_components": dict(info.get("reward_components", {})),
-        "spawn_count": int(info.get("spawn_count", 0) or 0),
-        "spawn_critical_steps": int(info.get("spawn_critical_steps", 0) or 0),
-        "pred_edges": [
-            {
-                "src": edge.src,
-                "rel": edge.rel,
-                "dst": edge.dst,
-                "confidence": float(edge.confidence),
-            }
-            for edge in env.memory_graph.edges
-        ],
-        "truth_edges": [
-            {
-                "src": edge.src,
-                "rel": edge.rel,
-                "dst": edge.dst,
-                "confidence": float(edge.confidence),
-            }
-            for edge in env.state.task.supporting_edges
-        ],
-    }
-
-
-def _last_action_error(observation: Any, info: dict[str, Any]) -> str | None:
-    raw = info.get("last_action_error") if isinstance(info, dict) else None
-    if raw is not None:
-        return str(raw)
-
-    tool_outputs = getattr(observation, "tool_outputs", None)
-    if isinstance(tool_outputs, list) and tool_outputs:
-        last = tool_outputs[-1]
-        if isinstance(last, dict):
-            output = last.get("output")
-            if isinstance(output, dict) and output.get("error") is not None:
-                return str(output.get("error"))
-    return None
-
-
-def _install_step_logger(env: OSINTEnvironment) -> tuple[list[float], dict[str, int], Any]:
-    rewards: list[float] = []
-    counters = {"steps": 0}
-    original_step = env.step
-
-    def _logged_step(action: Any):
-        observation, reward, done, info = original_step(action)
-        counters["steps"] += 1
-        reward_value = float(reward or 0.0)
-        rewards.append(reward_value)
-        action_type = getattr(action, "action_type", "")
-        action_type_value = str(getattr(action_type, "value", action_type))
-        action_text = _format_action(
-            {
-                "action_type": action_type_value,
-                "payload": dict(getattr(action, "payload", {}) or {}),
-            }
-        )
-        log_step(
-            step=counters["steps"],
-            action=action_text,
-            reward=reward_value,
-            done=bool(done),
-            error=_last_action_error(observation, info if isinstance(info, dict) else {}),
-        )
-        return observation, reward, done, info
-
-    env.step = _logged_step
-    return rewards, counters, original_step
-
-
-def _validate_required_configuration() -> None:
-    missing: list[str] = []
-
-    api_base = API_BASE_URL.strip()
-    model_name = MODEL_NAME.strip()
-    hf_token = HF_TOKEN.strip()
-    api_key = API_KEY.strip()
-    openai_key = OPENAI_API_KEY.strip()
-
-    if not api_base or api_base == "<your-active-endpoint>":
-        missing.append("API_BASE_URL")
-    if not model_name or model_name == "<your-active-model>":
-        missing.append("MODEL_NAME")
-    if not (hf_token or api_key or openai_key):
-        missing.append("HF_TOKEN|API_KEY|OPENAI_API_KEY")
-
-    # Required when using docker-image based env construction.
-    if os.getenv("REQUIRE_LOCAL_IMAGE_NAME", "0").strip().lower() in {"1", "true", "yes", "on"}:
-        if not LOCAL_IMAGE_NAME.strip():
-            missing.append("LOCAL_IMAGE_NAME")
-
-    if missing:
-        raise RuntimeError(f"Missing required environment variables: {', '.join(sorted(set(missing)))}")
-
-
-def _task_targets(env: OSINTEnvironment, episodes: int, task_indices: list[int]) -> list[int | None]:
-    if task_indices:
-        task_count = max(1, len(env.tasks))
-        return [index % task_count for index in task_indices]
-    return [None] * max(1, episodes)
-
-
-def _run_with_runner(
-    env: OSINTEnvironment,
-    llm: Any,
-    episodes: int,
-    task_indices: list[int],
-) -> tuple[dict[str, Any], list[dict[str, Any]], list[float], int]:
-    metrics = EvalMetrics()
-    episode_rows: list[dict[str, Any]] = []
-    rewards, counters, original_step = _install_step_logger(env)
-
-    single_runner = SingleAgentRunner(env=env, llm=llm)
-    swarm_runner = SwarmAgentRunner(env=env, llm=llm) if env.config.swarm.enabled else None
-
-    try:
-        for task_index in _task_targets(env, episodes, task_indices):
-            task_count = max(1, len(env.tasks))
-            selected_index = env._task_idx % task_count if task_index is None else int(task_index) % task_count
-            if task_index is not None:
-                # Keep compatibility with explicit task selection from the previous inference script.
-                env._task_idx = selected_index
-
-            difficulty = _task_difficulty(env, selected_index)
-            if difficulty == "easy":
-                runner: SingleAgentRunner | SwarmAgentRunner = single_runner
-            elif swarm_runner is not None:
-                runner = swarm_runner
-            else:
-                runner = single_runner
-
-            info = runner.run_episode()
-            if env.state is None:
-                continue
-
-            graph_f1 = compute_graph_f1(env.memory_graph.edges, env.state.task.supporting_edges)
-            metrics.add(info, task_type=env.state.task.task_type, graph_f1=graph_f1)
-            episode_rows.append(_episode_row(env, info))
-    finally:
-        env.step = original_step
-
-    return metrics.summary(), episode_rows, rewards, int(counters["steps"])
-
-
-def _maybe_write_artifacts(
-    env: OSINTEnvironment,
-    summary: dict[str, Any],
-    episodes: int,
-    episode_rows: list[dict[str, Any]],
-) -> tuple[dict[str, Any] | None, str | None]:
-    if not WRITE_BENCHMARK_ARTIFACTS:
-        return None, None
-
-    record = append_leaderboard_record(
-        path=LEADERBOARD_PATH,
-        summary=summary,
-        episodes=episodes,
-        run_name=RUN_NAME or None,
-        config={
-            "seed": env.config.seed,
-            "max_steps": env.config.max_steps,
-            "swarm_enabled": env.config.swarm.enabled,
-            "max_agents": env.config.swarm.max_agents,
-            "max_breadth": env.config.swarm.max_breadth,
-            "max_width": env.config.swarm.max_width,
-            "max_depth": env.config.swarm.max_depth,
-            "seeded_questions": len(env.config.seeding.seeded_questions),
-            "llm_provider": env.config.llm.provider,
-            "llm_model": env.config.llm.model,
-        },
-    )
-
-    leaderboard = load_leaderboard(LEADERBOARD_PATH)
-    dashboard = export_dashboard(
-        env=env,
-        evaluation={"summary": summary, "episodes": episode_rows},
-        leaderboard_records=leaderboard,
-        output_path=DASHBOARD_PATH,
-    )
-    return record, dashboard
-
-
-def main() -> None:
-    _validate_required_configuration()
-    env_cfg = _resolve_environment_config()
-    llm_client = build_llm_client(env_cfg.llm)
-
-    episodes_given = "EPISODES" in os.environ and str(os.getenv("EPISODES", "")).strip() != ""
-    task_indices_given = bool(TASK_INDICES)
-
-    if not episodes_given and not task_indices_given:
-        runs: list[tuple[str, list[int], int]] = [
-            ("easy", list(range(0, 10)), 10),
-            ("mid", list(range(10, 20)), 10),
-            ("hard", list(range(20, 30)), 10),
-        ]
-    else:
-        selected_indices = TASK_INDICES if task_indices_given else []
-        episodes = len(selected_indices) if selected_indices else max(1, EPISODES)
-        runs = [(TASK_NAME, selected_indices, episodes)]
-
-    for task_name, run_indices, run_episodes in runs:
-        env: OSINTEnvironment | None = None
-        rewards: list[float] = []
-        steps_taken = 0
-        score = 0.0
-        success = False
-
-        env = OSINTEnvironment(env_cfg, llm=llm_client)
-        log_start(task=task_name, env=BENCHMARK, model=env_cfg.llm.model)
-
-        try:
-            summary, episode_rows, rewards, steps_taken = _run_with_runner(
-                env=env,
-                llm=llm_client,
-                episodes=run_episodes,
-                task_indices=run_indices,
-            )
-
-            score = float(summary.get("avg_reward", 0.0) or 0.0)
-            score = max(0.0, min(1.0, score))
-            success = score >= SUCCESS_SCORE_THRESHOLD
-
-            _maybe_write_artifacts(
-                env=env,
-                summary=summary,
-                episodes=run_episodes,
-                episode_rows=episode_rows,
-            )
-        finally:
-            if env is not None:
-                close_fn = getattr(env, "close", None)
-                if callable(close_fn):
-                    close_fn()
-            log_end(task=task_name, success=success, steps=steps_taken, score=score, rewards=rewards)
-
-
-if __name__ == "__main__":
-    main()

scratch/osint/my_env_v4.py

@@ -1,46 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-
-
-@dataclass(slots=True)
-class MyEnvV4Action:
-    message: str
-
-
-@dataclass(slots=True)
-class _EchoObservation:
-    echoed_message: str
-
-
-@dataclass(slots=True)
-class _EchoResult:
-    observation: _EchoObservation
-    reward: float = 0.0
-    done: bool = False
-
-
-class MyEnvV4Env:
-    def __init__(self) -> None:
-        self._step_count = 0
-
-    @classmethod
-    async def from_docker_image(cls, image_name: str | None = None) -> "MyEnvV4Env":
-        return cls()
-
-    async def reset(self) -> _EchoResult:
-        self._step_count = 0
-        return _EchoResult(observation=_EchoObservation(echoed_message=""), reward=0.0, done=False)
-
-    async def step(self, action: MyEnvV4Action) -> _EchoResult:
-        self._step_count += 1
-        message = str(getattr(action, "message", ""))
-        reward = len(message) * 0.1
-        return _EchoResult(
-            observation=_EchoObservation(echoed_message=message),
-            reward=reward,
-            done=False,
-        )
-
-    async def close(self) -> None:
-        return None

scratch/osint/openenv.yaml

@@ -1,66 +0,0 @@
-name: osint-openenv
-version: 0.1.0
-description: Synthetic OSINT benchmark environment exposed over HTTP.
-tasks:
-  - id: seed_task_0
-    difficulty: easy
-    max_steps: 24
-    grader:
-      type: difficulty_exact_match
-      answer_type: node_id
-      case_sensitive: true
-      reward_profile: easy
-  - id: seed_task_10
-    difficulty: medium
-    max_steps: 24
-    grader:
-      type: difficulty_exact_match
-      answer_type: node_id
-      case_sensitive: true
-      reward_profile: medium
-  - id: seed_task_20
-    difficulty: hard
-    max_steps: 24
-    grader:
-      type: difficulty_exact_match
-      answer_type: node_id
-      case_sensitive: true
-      reward_profile: hard
-transport:
-  type: http
-  base_path: /
-endpoints:
-  health:
-    method: GET
-    path: /health
-  metadata:
-    method: GET
-    path: /api/environment
-  tasks:
-    method: GET
-    path: /openenv/tasks
-  reset:
-    method: POST
-    path: /reset
-  step:
-    method: POST
-    path: /step
-  state:
-    method: GET
-    path: /state
-models:
-  action_space:
-    - CALL_TOOL
-    - ADD_EDGE
-    - ANSWER
-  task_fields:
-    - task_id
-    - task_type
-    - question
-    - difficulty
-    - grader
-  observation_fields:
-    - tool_outputs
-    - graph_snapshot
-    - action_history
-    - task

scratch/osint/pyproject.toml

@@ -1,35 +0,0 @@
-[project]
-name = "osint-rl-env"
-version = "0.1.0"
-description = "OSINT-style multi-platform information ecosystem environment for LLM agents."
-readme = "README.md"
-requires-python = ">=3.10"
-dependencies = [
-	"openenv>=0.1.13",
-	"openai>=1.40.0",
-	"fastapi>=0.115.0",
-	"requests>=2.32.3",
-	"uvicorn>=0.30.0",
-]
-
-[project.optional-dependencies]
-dev = [
-	"pytest>=8.0.0",
-]
-
-[project.scripts]
-osint-env = "osint_env.cli:main"
-server = "osint_env.server_entry:main"
-
-[build-system]
-requires = ["setuptools>=68", "wheel"]
-build-backend = "setuptools.build_meta"
-
-[tool.setuptools]
-package-dir = {"" = "src"}
-
-[tool.setuptools.packages.find]
-where = ["src"]
-
-[tool.pytest.ini_options]
-testpaths = ["tests"]

scratch/osint/requirements.txt

@@ -1,6 +0,0 @@
-openenv>=0.1.13
-openai>=1.40.0
-fastapi>=0.115.0
-requests>=2.32.3
-uvicorn>=0.30.0
-pytest>=8.0.0

scratch/osint/scripts/build_fixed_levels_dataset.py

@@ -1,197 +0,0 @@
-from __future__ import annotations
-
-import argparse
-import json
-from collections import Counter
-from dataclasses import asdict
-from pathlib import Path
-from typing import Any
-
-from osint_env.config import clone_environment_config, load_seeding_config, load_shared_config
-from osint_env.data.generator import DatasetGenerator
-from osint_env.domain.models import Edge, TaskInstance
-from osint_env.llm import build_llm_client
-
-
-def edge_to_dict(edge: Edge) -> dict[str, Any]:
-    return {
-        "src": edge.src,
-        "rel": edge.rel,
-        "dst": edge.dst,
-        "confidence": float(edge.confidence),
-    }
-
-
-def task_to_dict(task: TaskInstance) -> dict[str, Any]:
-    return {
-        "task_id": task.task_id,
-        "task_type": task.task_type,
-        "question": task.question,
-        "answer": task.answer,
-        "supporting_edges": [edge_to_dict(e) for e in task.supporting_edges],
-        "metadata": dict(task.metadata),
-    }
-
-
-def build_fixed_snapshot(seed_path: Path) -> dict[str, Any]:
-    seeding = load_seeding_config(seed_path)
-    fixed_nodes = []
-    for node in seeding.seeded_nodes:
-        fixed_nodes.append(
-            {
-                "node_id": node.node_id,
-                "node_type": str(getattr(node.node_type, "value", node.node_type)),
-                "attrs": dict(node.attrs),
-            }
-        )
-    fixed_edges = [
-        {
-            "src": edge.src,
-            "rel": edge.rel,
-            "dst": edge.dst,
-            "confidence": float(edge.confidence),
-        }
-        for edge in seeding.seeded_edges
-    ]
-    fixed_questions = []
-    for idx, q in enumerate(seeding.seeded_questions):
-        fixed_questions.append(
-            {
-                "task_id": f"fixed_task_{idx:02d}",
-                "task_type": q.task_type,
-                "question": q.question,
-                "answer": q.answer,
-                "supporting_edges": [
-                    {
-                        "src": edge.src,
-                        "rel": edge.rel,
-                        "dst": edge.dst,
-                        "confidence": float(edge.confidence),
-                    }
-                    for edge in q.supporting_edges
-                ],
-                "metadata": dict(q.metadata),
-            }
-        )
-
-    difficulty_counts = Counter(str(q.get("metadata", {}).get("difficulty", "unknown")) for q in fixed_questions)
-    return {
-        "dataset_name": "fixed_levels_submission_set",
-        "source_seed": str(seed_path),
-        "graph": {
-            "nodes": fixed_nodes,
-            "edges": fixed_edges,
-            "node_count": len(fixed_nodes),
-            "edge_count": len(fixed_edges),
-        },
-        "questions": fixed_questions,
-        "question_count": len(fixed_questions),
-        "difficulty_counts": dict(difficulty_counts),
-    }
-
-
-def build_complete_snapshot(shared_config_path: Path, seed_path: Path) -> dict[str, Any]:
-    shared = load_shared_config(shared_config_path)
-    env_cfg = clone_environment_config(shared.environment)
-    env_cfg.seeding = load_seeding_config(seed_path)
-
-    llm_client = build_llm_client(env_cfg.llm)
-    generator = DatasetGenerator(config=env_cfg, llm=llm_client)
-
-    graph = generator.build_canonical_graph()
-    views = generator.build_platform_views(graph)
-    tasks = generator.generate_tasks(graph, views, count=max(15, len(env_cfg.seeding.seeded_questions)))
-
-    difficulty_counts = Counter(str(task.metadata.get("difficulty", "unknown")) for task in tasks)
-
-    return {
-        "dataset_name": "fixed_levels_submission_set",
-        "generation_mode": "llm_expanded",
-        "shared_config": str(shared_config_path),
-        "seed_file": str(seed_path),
-        "llm": asdict(env_cfg.llm),
-        "environment": {
-            "n_users": env_cfg.n_users,
-            "alias_density": env_cfg.alias_density,
-            "noise_level": env_cfg.noise_level,
-            "red_herring_rate": env_cfg.red_herring_rate,
-            "seed": env_cfg.seed,
-        },
-        "canonical_graph": {
-            "node_count": len(graph.nodes),
-            "edge_count": len(graph.edges),
-            "nodes": [
-                {
-                    "node_id": node.node_id,
-                    "node_type": node.node_type.value,
-                    "attrs": dict(node.attrs),
-                }
-                for node in sorted(graph.nodes.values(), key=lambda n: n.node_id)
-            ],
-            "edges": [edge_to_dict(edge) for edge in graph.edges],
-        },
-        "platform_views": {
-            "microblog_posts": views.microblog_posts,
-            "forum_threads": views.forum_threads,
-            "profiles": views.profiles,
-            "counts": {
-                "microblog_posts": len(views.microblog_posts),
-                "forum_threads": len(views.forum_threads),
-                "profiles": len(views.profiles),
-            },
-        },
-        "tasks": [task_to_dict(task) for task in tasks],
-        "task_count": len(tasks),
-        "difficulty_counts": dict(difficulty_counts),
-    }
-
-
-def main() -> None:
-    parser = argparse.ArgumentParser(description="Build fixed difficulty dataset artifacts.")
-    parser.add_argument(
-        "--seed-file",
-        default="datasets/fixed_levels/seed_fixed_levels.json",
-        help="Path to seeding JSON with fixed graph/questions.",
-    )
-    parser.add_argument(
-        "--shared-config",
-        default="datasets/fixed_levels/shared_config_fixed_levels.json",
-        help="Path to shared config used for LLM-expanded generation.",
-    )
-    parser.add_argument(
-        "--output-dir",
-        default="datasets/fixed_levels",
-        help="Directory where dataset artifacts are written.",
-    )
-    args = parser.parse_args()
-
-    output_dir = Path(args.output_dir)
-    output_dir.mkdir(parents=True, exist_ok=True)
-
-    seed_path = Path(args.seed_file)
-    shared_path = Path(args.shared_config)
-
-    fixed_snapshot = build_fixed_snapshot(seed_path)
-    fixed_path = output_dir / "fixed_graph_questions.json"
-    fixed_path.write_text(json.dumps(fixed_snapshot, indent=2, sort_keys=True), encoding="utf-8")
-
-    complete_snapshot = build_complete_snapshot(shared_path, seed_path)
-    complete_path = output_dir / "complete_dataset_qwen_generated.json"
-    complete_path.write_text(json.dumps(complete_snapshot, indent=2, sort_keys=True), encoding="utf-8")
-
-    summary = {
-        "fixed_dataset": str(fixed_path),
-        "complete_dataset": str(complete_path),
-        "fixed_nodes": fixed_snapshot["graph"]["node_count"],
-        "fixed_edges": fixed_snapshot["graph"]["edge_count"],
-        "fixed_questions": fixed_snapshot["question_count"],
-        "complete_nodes": complete_snapshot["canonical_graph"]["node_count"],
-        "complete_edges": complete_snapshot["canonical_graph"]["edge_count"],
-        "complete_tasks": complete_snapshot["task_count"],
-        "difficulty_counts": complete_snapshot["difficulty_counts"],
-    }
-    print(json.dumps(summary, indent=2, sort_keys=True))
-
-
-if __name__ == "__main__":
-    main()

scratch/osint/scripts/generate_fixed_levels_seed.py

@@ -1,109 +0,0 @@
-from collections import Counter, OrderedDict
-from pathlib import Path
-import json
-
-U=[('aria','Aria Sen','Helios Labs','Sector 9'),('bharat','Bharat Kulkarni','Northbridge Logistics','Dockyard 17'),('cyrus','Cyrus Mehta','Apex Dynamics','Old Town'),('diya','Diya Roy','Blueharbor Media','Old Town'),('elin','Elin Das','Helios Labs','Sector 9'),('faris','Faris Noor','Tidewatch Ops','Rivergate'),('gita','Gita Pradhan','Apex Dynamics','Old Town'),('hiro','Hiro Tan','Northbridge Logistics','Dockyard 17'),('ivy','Ivy Kapoor','Kestrel Works','Rivergate'),('jules','Jules Banerjee','Blueharbor Media','Old Town'),('kian','Kian Bose','Atlas Freight','East Quay'),('leena','Leena Das','Sunmesh Analytics','Sector 9'),('mika','Mika Solanki','Orion Customs','North Basin'),('nora','Nora Iqbal','Emberline Security','Foundry Row'),('omar','Omar Sheikh','Atlas Freight','East Quay'),('priya','Priya Menon','Sunmesh Analytics','Sector 9'),('quinn','Quinn Rao','Orion Customs','North Basin'),('rhea','Rhea Kapoor','Emberline Security','Foundry Row'),('soren','Soren Malik','Harborlight Transit','Uplink Yard'),('tara','Tara Dey','Harborlight Transit','Uplink Yard')]
-A=[('orchidfox','@orchidfox','ivy'),('steelquill','@steelquill','bharat'),('monsoonbyte','@monsoonbyte','diya'),('nightrelay','@nightrelay','faris'),('mapleghost','@mapleghost','elin'),('docksparrow','@docksparrow','hiro'),('quartzlotus','@quartzlotus','cyrus'),('emberglass','@emberglass','nora'),('basinraven','@basinraven','mika'),('tideshard','@tideshard','soren'),('hollowsignal','@hollowsignal','priya'),('ironwhisper','@ironwhisper','omar'),('cinderveil','@cinderveil','rhea'),('sablekeel','@sablekeel','tara'),('lanternmoth','@lanternmoth','kian'),('frostledger','@frostledger','leena')]
-L=[('dockyard17','Dockyard 17'),('sector9','Sector 9'),('old_town','Old Town'),('rivergate','Rivergate'),('east_quay','East Quay'),('foundry_row','Foundry Row'),('north_basin','North Basin'),('uplink_yard','Uplink Yard')]
-O=[('helios_labs','Helios Labs','sector9'),('northbridge_logistics','Northbridge Logistics','dockyard17'),('apex_dynamics','Apex Dynamics','old_town'),('blueharbor_media','Blueharbor Media','old_town'),('tidewatch_ops','Tidewatch Ops','rivergate'),('kestrel_works','Kestrel Works','rivergate'),('atlas_freight','Atlas Freight','east_quay'),('sunmesh_analytics','Sunmesh Analytics','sector9'),('orion_customs','Orion Customs','north_basin'),('emberline_security','Emberline Security','foundry_row'),('harborlight_transit','Harborlight Transit','uplink_yard')]
-E=[('project_lantern','Project Lantern'),('black_kite','Black Kite'),('silent_current','Silent Current'),('amber_veil','Amber Veil'),('glass_harbor','Glass Harbor'),('ember_tide','Ember Tide'),('iron_wharf','Iron Wharf'),('ghost_signal','Ghost Signal')]
-T=[('supply_leak','supply_chain'),('port_audit','port_audit'),('customs_breach','customs_breach'),('relay_map','relay_map'),('foundry_watch','foundry_watch'),('basin_shift','basin_shift'),('quiet_manifest','quiet_manifest'),('uplink_route','uplink_route'),('ember_tide_watch','ember_tide'),('ghost_signal_net','ghost_signal')]
-P=['shift_roster','midnight_manifest','sat_phone_ping','drone_parts','relay_schedule','quay_ledgers','customs_tag','hull_signal','basin_photo','foundry_map','lantern_route','uplink_note']
-
-def uid(x): return f'user_{x}'
-def aid(x): return f'alias_{x}'
-def oid(x): return f'org_{x}'
-def lid(x): return f'loc_{x}'
-def eid(x): return f'event_{x}'
-def tid(x): return f'thr_{x}'
-def pid(x): return f'post_{x}'
-
-def addn(nodes,nid,nt,attrs): nodes.append({'node_id':nid,'node_type':nt,'attrs':attrs})
-
-def build():
-    nodes=[]; edges=OrderedDict();
-    for s,name,org,loc in U: addn(nodes,uid(s),'user',{'name':name,'org':org,'location':loc})
-    for s,handle,user in A: addn(nodes,aid(s),'alias',{'handle':handle})
-    for s,name,_ in O: addn(nodes,oid(s),'org',{'name':name})
-    for s,name in L: addn(nodes,lid(s),'location',{'name':name})
-    for s,name in E: addn(nodes,eid(s),'event',{'name':name})
-    for s,topic in T: addn(nodes,tid(s),'thread',{'topic':topic})
-    for s in P: addn(nodes,pid(s),'post',{'channel':'microblog'})
-    def ae(k,src,rel,dst,c=1.0): edges[k]={'src':src,'rel':rel,'dst':dst,'confidence':c}
-    for s,_,user in A: ae(f'a_{s}',aid(s),'alias_of',uid(user))
-    org_map={name:oid(s) for s,name,_ in O}; loc_map={name:lid(s) for s,name in L}
-    for s,_,org,loc in U: ae(f'w_{s}',uid(s),'works_at',org_map[org]); ae(f'l_{s}',uid(s),'located_in',loc_map[loc])
-    for s,_,loc in O: ae(f'op_{s}',oid(s),'operates_in',lid(loc))
-    CP=[('ivy','bharat',.95),('bharat','hiro',.95),('hiro','faris',.92),('faris','diya',.90),('diya','elin',.89),('elin','aria',.87),('aria','cyrus',.84),('cyrus','gita',.83),('gita','jules',.82),('jules','bharat',.81),('diya','ivy',.90),('ivy','elin',.86),('kian','omar',.93),('omar','mika',.90),('mika','quinn',.89),('quinn','nora',.88),('nora','rhea',.87),('rhea','soren',.86),('soren','tara',.86),('tara','kian',.84),('priya','leena',.91),('leena','aria',.83),('priya','nora',.82),('kian','bharat',.80),('soren','faris',.79),('quinn','hiro',.78)]
-    for i,(a,b,c) in enumerate(CP,1): ae(f'c{i:02d}',uid(a),'connected_to',uid(b),c)
-    PA={'midnight_manifest':'orchidfox','shift_roster':'docksparrow','sat_phone_ping':'nightrelay','drone_parts':'monsoonbyte','relay_schedule':'steelquill','quay_ledgers':'lanternmoth','customs_tag':'basinraven','hull_signal':'tideshard','basin_photo':'emberglass','foundry_map':'cinderveil','lantern_route':'frostledger','uplink_note':'sablekeel'}
-    for post,author in PA.items(): ae(f'ap_{post}',aid(author),'authored_post',pid(post))
-    PR={'midnight_manifest':['dockyard17','project_lantern'],'shift_roster':['dockyard17','northbridge_logistics'],'sat_phone_ping':['rivergate','project_lantern'],'drone_parts':['black_kite','kestrel_works'],'relay_schedule':['project_lantern','sector9'],'quay_ledgers':['east_quay','glass_harbor'],'customs_tag':['north_basin','iron_wharf'],'hull_signal':['uplink_yard','ghost_signal'],'basin_photo':['foundry_row','amber_veil'],'foundry_map':['foundry_row','ember_tide'],'lantern_route':['project_lantern','sunmesh_analytics'],'uplink_note':['uplink_yard','harborlight_transit']}
-    for post,refs in PR.items():
-        for i,x in enumerate(refs,1): ae(f'r_{post}_{i}',pid(post),'references', lid(x) if x in {y for y,_ in L} else (oid(x) if x in {y for y,_,_ in O} else eid(x)))
-    TA={'supply_leak':'diya','port_audit':'jules','customs_breach':'mika','relay_map':'leena','foundry_watch':'nora','basin_shift':'quinn','quiet_manifest':'kian','uplink_route':'soren','ember_tide_watch':'rhea','ghost_signal_net':'tara'}
-    TL={'supply_leak':[('discusses','project_lantern'),('references','northbridge_logistics')],'port_audit':[('discusses','black_kite'),('references','kestrel_works')],'customs_breach':[('discusses','iron_wharf'),('references','orion_customs')],'relay_map':[('discusses','project_lantern'),('references','sunmesh_analytics')],'foundry_watch':[('discusses','ember_tide'),('references','emberline_security')],'basin_shift':[('discusses','amber_veil'),('references','north_basin')],'quiet_manifest':[('discusses','glass_harbor'),('references','atlas_freight')],'uplink_route':[('discusses','ghost_signal'),('references','harborlight_transit')],'ember_tide_watch':[('discusses','ember_tide'),('references','foundry_row')],'ghost_signal_net':[('discusses','ghost_signal'),('references','uplink_yard')]}
-    for t,u in TA.items(): ae(f'at_{t}',uid(u),'authored_thread',tid(t))
-    for t,rels in TL.items():
-        for i,(rel,x) in enumerate(rels,1): ae(f'tl_{t}_{i}',tid(t),rel, lid(x) if x in {y for y,_ in L} else (oid(x) if x in {y for y,_,_ in O} else eid(x)))
-    ER=[('bharat','collaborates_on','project_lantern'),('hiro','collaborates_on','project_lantern'),('faris','collaborates_on','project_lantern'),('diya','investigates','project_lantern'),('leena','monitors','project_lantern'),('ivy','collaborates_on','black_kite'),('cyrus','collaborates_on','black_kite'),('elin','investigates','black_kite'),('jules','reports_on','black_kite'),('kian','collaborates_on','glass_harbor'),('omar','collaborates_on','glass_harbor'),('priya','monitors','glass_harbor'),('mika','collaborates_on','iron_wharf'),('quinn','collaborates_on','iron_wharf'),('nora','investigates','amber_veil'),('rhea','collaborates_on','ember_tide'),('soren','collaborates_on','ghost_signal'),('tara','reports_on','ghost_signal'),('gita','monitors','silent_current'),('jules','reports_on','silent_current')]
-    for i,(u,rel,e) in enumerate(ER,1): ae(f'er{i:02d}',uid(u),rel,eid(e),.9)
-    X=[(eid('project_lantern'),'connected_to',eid('glass_harbor')),(eid('black_kite'),'connected_to',eid('amber_veil')),(eid('ember_tide'),'connected_to',eid('ghost_signal')),(oid('atlas_freight'),'connected_to',oid('northbridge_logistics')),(oid('orion_customs'),'connected_to',oid('emberline_security')),(oid('harborlight_transit'),'connected_to',oid('tidewatch_ops'))]
-    for i,(a,rel,b) in enumerate(X,1): ae(f'x{i:02d}',a,rel,b,.77)
-    return nodes,edges
-
-def mk_questions(edges):
-    def ids(*items):
-        out=[]
-        for it in items:
-            if isinstance(it,list): out.extend(it)
-            else: out.append(it)
-        return out
-    def rng(prefix,a,b): return [f'{prefix}{i:02d}' for i in range(a,b+1)]
-    def sup(edge_ids): return [edges[e] for e in edge_ids]
-    def nodes(edge_ids):
-        s=set()
-        for e in edge_ids: s|={edges[e]['src'],edges[e]['dst']}
-        return len(s)
-    qs=[]
-    easy=[('easy_01','alias_orchidfox -> post_midnight_manifest -> loc_dockyard17 -> connected collaborator on event_project_lantern. Who is it?','user_bharat',ids('a_orchidfox','ap_midnight_manifest','r_midnight_manifest_1','c01','er01')),('easy_02','thr_supply_leak references org_northbridge_logistics. Which alias_docksparrow user works there and collaborates on event_project_lantern?','user_hiro',ids('tl_supply_leak_2','a_docksparrow','w_hiro','er02')),('easy_03','alias_monsoonbyte authored post_drone_parts about event_black_kite. Which user behind that alias is directly connected to the Kestrel collaborator?','user_diya',ids('a_monsoonbyte','ap_drone_parts','r_drone_parts_1','w_ivy','er06','c12')),('easy_04','alias_nightrelay references loc_rivergate. Which user behind it works at an org operating there and collaborates on event_project_lantern?','user_faris',ids('a_nightrelay','ap_sat_phone_ping','r_sat_phone_ping_1','w_faris','op_tidewatch_ops','er03')),('easy_05','thr_port_audit discusses Black Kite and references Kestrel Works. Which alias_orchidfox user authored post_midnight_manifest and collaborates on Black Kite?','user_ivy',ids('tl_port_audit_1','tl_port_audit_2','a_orchidfox','ap_midnight_manifest','w_ivy','er06')),('easy_06','Which Atlas Freight user behind alias_lanternmoth authored post_quay_ledgers and collaborates on event_glass_harbor?','user_kian',ids('a_lanternmoth','ap_quay_ledgers','w_kian','er10')),('easy_07','Which Orion Customs user behind alias_basinraven authored post_customs_tag and collaborates on event_iron_wharf?','user_mika',ids('a_basinraven','ap_customs_tag','w_mika','er13')),('easy_08','Which user behind alias_emberglass posted basin_photo from Foundry Row and investigates Amber Veil?','user_nora',ids('a_emberglass','ap_basin_photo','r_basin_photo_1','er15')),('easy_09','Which user behind alias_tideshard authored post_hull_signal and collaborates on Ghost Signal?','user_soren',ids('a_tideshard','ap_hull_signal','er17')),('easy_10','Which Harborlight Transit user behind alias_sablekeel authored post_uplink_note and reports on Ghost Signal?','user_tara',ids('a_sablekeel','ap_uplink_note','w_tara','er18'))]
-    mid=[('mid_01','Follow alias_docksparrow through post_shift_roster, Dockyard 17, and the Lantern chain. Return the org node id.','org_northbridge_logistics',ids('a_docksparrow','ap_shift_roster','r_shift_roster_1','r_shift_roster_2','tl_supply_leak_2','w_hiro','l_hiro','er02','er01','c02','c03')),('mid_02','Across the Glass Harbor cluster, which user behind alias_lanternmoth links to the Atlas Freight network from thr_quiet_manifest?','user_kian',ids('a_lanternmoth','ap_quay_ledgers','r_quay_ledgers_1','r_quay_ledgers_2','at_quiet_manifest','tl_quiet_manifest_1','tl_quiet_manifest_2','w_kian','w_omar','er10','er11','er12','c13','c14')),('mid_03','Trace alias_basinraven through post_customs_tag, thr_customs_breach, and the Orion Customs collaboration chain. Who is it?','user_mika',ids('a_basinraven','ap_customs_tag','r_customs_tag_1','r_customs_tag_2','at_customs_breach','tl_customs_breach_1','tl_customs_breach_2','w_mika','w_quinn','er13','er14','c15','c16','x05')),('mid_04','In the Ember Tide and Amber Veil overlap, which Foundry Row user behind alias_cinderveil collaborates on Ember Tide?','user_rhea',ids('a_cinderveil','ap_foundry_map','r_foundry_map_1','r_foundry_map_2','at_foundry_watch','tl_foundry_watch_1','tl_foundry_watch_2','at_ember_tide_watch','tl_ember_tide_watch_1','tl_ember_tide_watch_2','w_rhea','w_nora','er15','er16','c17','x03')),('mid_05','Follow alias_tideshard from post_hull_signal into thr_uplink_route and the Harborlight relay. Return the org node id.','org_harborlight_transit',ids('a_tideshard','ap_hull_signal','r_hull_signal_1','r_hull_signal_2','at_uplink_route','tl_uplink_route_1','tl_uplink_route_2','w_soren','w_tara','er17','er18','c18','c19','op_harborlight_transit','x06')),('mid_06','Which Sunmesh user behind alias_frostledger connects post_lantern_route to thr_relay_map and the Sector 9 monitoring chain?','user_leena',ids('a_frostledger','ap_lantern_route','r_lantern_route_1','r_lantern_route_2','at_relay_map','tl_relay_map_1','tl_relay_map_2','w_leena','w_priya','l_leena','op_sunmesh_analytics','er05','c21','c22')),('mid_07','Which user behind alias_emberglass is tied to Amber Veil after combining post_basin_photo, thr_basin_shift, and the Foundry Row investigation chain?','user_nora',ids('a_emberglass','ap_basin_photo','r_basin_photo_1','r_basin_photo_2','at_basin_shift','tl_basin_shift_1','tl_basin_shift_2','w_nora','w_quinn','l_nora','er15','c16','c17','x05')),('mid_08','Combine alias_orchidfox, post_midnight_manifest, thr_supply_leak, and the Lantern to Glass Harbor bridge. Which user starts that chain?','user_ivy',ids('a_orchidfox','ap_midnight_manifest','r_midnight_manifest_1','r_midnight_manifest_2','at_supply_leak','tl_supply_leak_1','tl_supply_leak_2','w_ivy','er06','c01','c12','x01','er10','er12')),('mid_09','Which user behind alias_monsoonbyte sits at the overlap of Blueharbor Media, Project Lantern, Black Kite, and the Ivy connection chain?','user_diya',ids('a_monsoonbyte','ap_drone_parts','r_drone_parts_1','at_supply_leak','tl_supply_leak_1','at_port_audit','tl_port_audit_1','w_diya','w_ivy','w_jules','er04','er06','er09','c04','c12')),('mid_10','Who is the Northbridge user behind alias_steelquill when combining post_relay_schedule, thr_supply_leak, Dockyard 17, and Lantern collaborator edges?','user_bharat',ids('a_steelquill','ap_relay_schedule','r_relay_schedule_1','r_relay_schedule_2','at_supply_leak','tl_supply_leak_1','tl_supply_leak_2','w_bharat','w_hiro','l_bharat','l_hiro','er01','er02','c01','c02'))]
-    big=list(edges.keys())[:58]
-    hard=[('high_01','Lantern to Glass Harbor handoff: identify the user behind alias_orchidfox after combining Lantern logistics, Dockyard links, and Atlas Freight bridge evidence.','user_ivy',ids('a_orchidfox','ap_midnight_manifest','r_midnight_manifest_1','r_midnight_manifest_2','at_supply_leak','tl_supply_leak_1','tl_supply_leak_2',['w_ivy','w_bharat','w_hiro','w_kian','w_omar'],['l_ivy','l_bharat','l_hiro','l_kian','l_omar'],['op_northbridge_logistics','op_kestrel_works','op_atlas_freight'],rng('c',1,3),['c12','c13','c14'],['er01','er02','er03','er06','er10','er11','er12'],'at_quiet_manifest','tl_quiet_manifest_1','tl_quiet_manifest_2','ap_quay_ledgers','r_quay_ledgers_1','r_quay_ledgers_2','x01','x04','a_lanternmoth','a_steelquill','a_docksparrow')),('high_02','North Basin to Foundry Row escalation: which user behind alias_basinraven anchors the Iron Wharf side before the Emberline handoff?','user_mika',ids('a_basinraven','ap_customs_tag','r_customs_tag_1','r_customs_tag_2','at_customs_breach','tl_customs_breach_1','tl_customs_breach_2','at_basin_shift','tl_basin_shift_1','tl_basin_shift_2','at_foundry_watch','tl_foundry_watch_1','tl_foundry_watch_2',['w_mika','w_quinn','w_nora','w_rhea'],['l_mika','l_quinn','l_nora','l_rhea'],['op_orion_customs','op_emberline_security'],['c15','c16','c17'],['er13','er14','er15','er16'],'ap_basin_photo','r_basin_photo_1','r_basin_photo_2','ap_foundry_map','r_foundry_map_1','r_foundry_map_2','x02','x03','x05','a_emberglass','a_cinderveil','c23','c24')),('high_03','Harborlight ghost-signal relay: identify the user behind alias_tideshard at the Harborlight / Tidewatch junction.','user_soren',ids('a_tideshard','ap_hull_signal','r_hull_signal_1','r_hull_signal_2','a_sablekeel','ap_uplink_note','r_uplink_note_1','r_uplink_note_2','at_uplink_route','tl_uplink_route_1','tl_uplink_route_2','at_ghost_signal_net','tl_ghost_signal_net_1','tl_ghost_signal_net_2',['w_soren','w_tara','w_faris'],['l_soren','l_tara','l_faris'],['op_harborlight_transit','op_tidewatch_ops'],['c18','c19','c20','c25'],['er03','er17','er18'],'ap_sat_phone_ping','r_sat_phone_ping_1','r_sat_phone_ping_2','at_supply_leak','tl_supply_leak_1','er01','er02','x03','x06','a_nightrelay')),('high_04','Blueharbor to Black Kite to Lantern overlap: which user is the Blueharbor origin behind alias_monsoonbyte?','user_diya',ids('a_monsoonbyte','ap_drone_parts','r_drone_parts_1','r_drone_parts_2','at_port_audit','tl_port_audit_1','tl_port_audit_2','at_supply_leak','tl_supply_leak_1','tl_supply_leak_2',['w_diya','w_jules','w_ivy','w_cyrus'],['l_diya','l_jules','l_ivy','l_cyrus'],['op_blueharbor_media','op_kestrel_works','op_apex_dynamics'],['c04','c08','c09','c12'],['er04','er06','er07','er08','er09'],'a_orchidfox','ap_midnight_manifest','r_midnight_manifest_2','x01','x02','at_relay_map','tl_relay_map_1','w_leena','er05')),('high_05','Sector 9 to Dockyard 17 full relay: which user behind alias_steelquill links the Northbridge chain and the Sunmesh monitoring bridge?','user_bharat',ids('a_steelquill','ap_relay_schedule','r_relay_schedule_1','r_relay_schedule_2','a_frostledger','ap_lantern_route','r_lantern_route_1','r_lantern_route_2','at_relay_map','tl_relay_map_1','tl_relay_map_2','at_supply_leak','tl_supply_leak_1','tl_supply_leak_2',['w_bharat','w_hiro','w_leena','w_priya','w_aria'],['l_bharat','l_hiro','l_leena','l_priya','l_aria'],['op_northbridge_logistics','op_sunmesh_analytics','op_helios_labs'],['c01','c02','c05','c06','c07','c21','c22'],['er01','er02','er05'],'x01','x04','a_docksparrow','a_mapleghost','a_hollowsignal')),('high_06','Foundry Row, North Basin, and Uplink Yard spread: identify the user behind alias_emberglass before the Harborlight relay takes over.','user_nora',ids('a_emberglass','ap_basin_photo','r_basin_photo_1','r_basin_photo_2','a_cinderveil','ap_foundry_map','r_foundry_map_1','r_foundry_map_2','a_sablekeel','ap_uplink_note','r_uplink_note_1','r_uplink_note_2','at_foundry_watch','tl_foundry_watch_1','tl_foundry_watch_2','at_ember_tide_watch','tl_ember_tide_watch_1','tl_ember_tide_watch_2','at_uplink_route','tl_uplink_route_1','tl_uplink_route_2',['w_nora','w_rhea','w_soren','w_tara'],['l_nora','l_rhea','l_soren','l_tara'],['op_emberline_security','op_harborlight_transit'],['c17','c18','c19'],['er15','er16','er17','er18'],'x03','x06')),('high_07','Freight and customs bridge: which Atlas Freight user behind alias_lanternmoth connects Glass Harbor with the Northbridge chain?','user_kian',ids('a_lanternmoth','ap_quay_ledgers','r_quay_ledgers_1','r_quay_ledgers_2','at_quiet_manifest','tl_quiet_manifest_1','tl_quiet_manifest_2',['w_kian','w_omar','w_bharat','w_hiro'],['l_kian','l_omar','l_bharat','l_hiro'],['op_atlas_freight','op_northbridge_logistics'],['c13','c14','c24','c02'],['er10','er11','er12','er01','er02'],'ap_shift_roster','r_shift_roster_1','r_shift_roster_2','ap_midnight_manifest','r_midnight_manifest_1','at_supply_leak','tl_supply_leak_2','x04','a_ironwhisper','a_steelquill','a_docksparrow')),('high_08','Black Kite, Amber Veil, and Iron Wharf overlap: which user behind alias_quartzlotus is the Apex-side collaborator?','user_cyrus',ids('a_quartzlotus','w_cyrus','l_cyrus','op_apex_dynamics','er07','at_port_audit','tl_port_audit_1','ap_drone_parts','r_drone_parts_1','er15','at_basin_shift','tl_basin_shift_1','er13','at_customs_breach','tl_customs_breach_1',['w_ivy','w_nora','w_mika','w_quinn'],['l_ivy','l_nora','l_mika','l_quinn'],['op_kestrel_works','op_emberline_security','op_orion_customs'],['c08','c12','c15','c16','c17'],'x02','x05','a_orchidfox','a_basinraven','a_emberglass')),('high_09','Ghost Signal and Ember Tide relay: which user behind alias_sablekeel is the Harborlight reporting endpoint?','user_tara',ids('a_sablekeel','ap_uplink_note','r_uplink_note_1','r_uplink_note_2','a_tideshard','ap_hull_signal','r_hull_signal_1','r_hull_signal_2','at_ghost_signal_net','tl_ghost_signal_net_1','tl_ghost_signal_net_2','at_uplink_route','tl_uplink_route_1','tl_uplink_route_2','at_ember_tide_watch','tl_ember_tide_watch_1','tl_ember_tide_watch_2',['w_tara','w_soren','w_rhea','w_nora'],['l_tara','l_soren','l_rhea','l_nora'],['op_harborlight_transit','op_emberline_security'],['c18','c19','c17'],['er16','er17','er18'],'x03','x06','a_cinderveil','a_emberglass')),('high_10','End-to-end benchmark sweep: across Lantern, Black Kite, Glass Harbor, Iron Wharf, Ember Tide, and Ghost Signal, which user behind alias_hollowsignal anchors the Sunmesh monitoring side?','user_priya',big)]
-    for diff,level,specs in [('easy',1,easy),('mid',2,mid),('high',3,hard)]:
-        for qid,q,a,eids in specs:
-            qs.append({'task_type':'fixed_trace','question':q,'answer':a,'supporting_edges':sup(eids),'metadata':{'difficulty':diff,'difficulty_level':level,'question_id':qid,'support_nodes':nodes(eids)}})
-    def edge_key(e): return (e['src'], e['rel'], e['dst'])
-    mid_pool = sup(ids('a_orchidfox','ap_midnight_manifest','r_midnight_manifest_1','r_midnight_manifest_2','a_lanternmoth','ap_quay_ledgers','r_quay_ledgers_1','r_quay_ledgers_2','a_basinraven','ap_customs_tag','r_customs_tag_1','r_customs_tag_2','a_tideshard','ap_hull_signal','r_hull_signal_1','at_supply_leak','tl_supply_leak_1','at_quiet_manifest','tl_quiet_manifest_1','er01','er02','er06','er10','c01','c02','c13'))
-    hard_pool = sup(list(edges.keys())[:120])
-    for q in qs:
-        current = {edge_key(e) for e in q['supporting_edges']}
-        diff = q['metadata']['difficulty']
-        if diff == 'mid':
-            pool = mid_pool
-            target = 17
-        elif diff == 'high':
-            pool = hard_pool
-            target = 50
-        else:
-            continue
-        for e in pool:
-            if q['metadata']['support_nodes'] >= target:
-                break
-            k = edge_key(e)
-            if k not in current:
-                q['supporting_edges'].append(dict(e))
-                current.add(k)
-                q['metadata']['support_nodes'] = len({n for edge in q['supporting_edges'] for n in (edge['src'], edge['dst'])})
-    return qs
-
-def main():
-    nodes,edges=build(); questions=mk_questions(edges)
-    payload={'seeding':{'seeded_nodes':nodes,'seeded_edges':list(edges.values()),'seeded_questions':questions,'llm_generate_remaining_graph':True,'llm_generate_remaining_tasks':False,'llm_generated_edge_budget':48,'llm_generated_task_budget':0,'llm_generation_parallel':True,'llm_generation_workers':4,'llm_generation_retries':3,'allow_template_fallback_on_llm_failure':False}}
-    out=Path('datasets/fixed_levels/seed_fixed_levels.json'); out.write_text(json.dumps(payload,indent=2),encoding='utf-8')
-    counts=Counter(q['metadata']['difficulty'] for q in questions)
-    stats={k:sorted(q['metadata']['support_nodes'] for q in questions if q['metadata']['difficulty']==k) for k in ['easy','mid','high']}
-    print(json.dumps({'nodes':len(nodes),'edges':len(edges),'questions':len(questions),'difficulty_counts':dict(counts),'support_nodes':stats},indent=2))
-
-if __name__=='__main__':
-    main()

scratch/osint/scripts/run_openai_baseline.py

@@ -1,59 +0,0 @@
-from __future__ import annotations
-
-import argparse
-import json
-import os
-
-from osint_env.baselines import OpenAIBaselineConfig, OpenAIBaselineRunner
-
-
-def build_parser() -> argparse.ArgumentParser:
-    parser = argparse.ArgumentParser(description="Run the reproducible OpenAI baseline on the fixed-level OSINT benchmark.")
-    parser.add_argument("--config", default="datasets/fixed_levels/shared_config_fixed_levels.json", help="Shared config JSON.")
-    parser.add_argument("--seed-file", default="datasets/fixed_levels/seed_fixed_levels.json", help="Fixed seed file JSON.")
-    parser.add_argument("--output", default="artifacts/baselines/openai_fixed_levels_latest.json", help="Baseline result JSON output path.")
-    parser.add_argument("--leaderboard", default="artifacts/baselines/openai_fixed_levels_leaderboard.json", help="Leaderboard JSON path.")
-    parser.add_argument("--dashboard", default="artifacts/baselines/openai_fixed_levels_dashboard.html", help="Dashboard HTML path.")
-    parser.add_argument("--run-name", default="openai_fixed_levels_baseline", help="Leaderboard run name.")
-    parser.add_argument("--model", default="gpt-5-nano", help="OpenAI chat model name.")
-    parser.add_argument("--openai-base-url", default="https://api.openai.com/v1", help="OpenAI-compatible base URL.")
-    parser.add_argument("--openai-api-key", default="", help="OpenAI API key override.")
-    parser.add_argument("--openai-api-key-env", default="OPENAI_API_KEY", help="Environment variable name for the API key.")
-    parser.add_argument("--episodes", type=int, default=30, help="Number of episodes to evaluate.")
-    parser.add_argument("--max-steps", type=int, default=8, help="Episode step budget to keep runs bounded.")
-    parser.add_argument("--temperature", type=float, default=0.0, help="Sampling temperature.")
-    parser.add_argument("--max-tokens", type=int, default=256, help="Maximum completion tokens per step.")
-    parser.add_argument("--timeout-seconds", type=int, default=60, help="Per-request timeout.")
-    parser.add_argument("--seed", type=int, default=7, help="Request seed offset used for repeatable runs.")
-    parser.add_argument("--skip-leaderboard", action="store_true", help="Do not append the run to the leaderboard file.")
-    return parser
-
-
-def main() -> None:
-    args = build_parser().parse_args()
-    api_key = args.openai_api_key or os.getenv(args.openai_api_key_env, "")
-    config = OpenAIBaselineConfig(
-        shared_config_path=args.config,
-        seed_file=args.seed_file,
-        output_path=args.output,
-        leaderboard_path=args.leaderboard,
-        dashboard_path=args.dashboard,
-        run_name=args.run_name,
-        model=args.model,
-        base_url=args.openai_base_url,
-        api_key=api_key,
-        api_key_env=args.openai_api_key_env,
-        temperature=args.temperature,
-        max_tokens=args.max_tokens,
-        timeout_seconds=args.timeout_seconds,
-        episodes=args.episodes,
-        max_steps=args.max_steps,
-        seed=args.seed,
-        append_leaderboard=not args.skip_leaderboard,
-    )
-    result = OpenAIBaselineRunner(config).run()
-    print(json.dumps({"summary": result["summary"], "output": args.output, "dashboard": args.dashboard}, indent=2, sort_keys=True))
-
-
-if __name__ == "__main__":
-    main()

scratch/osint/scripts/validate_release.py

@@ -1,21 +0,0 @@
-from __future__ import annotations
-
-import json
-import sys
-from pathlib import Path
-
-ROOT = Path(__file__).resolve().parents[1]
-if str(ROOT) not in sys.path:
-    sys.path.insert(0, str(ROOT))
-
-from osint_env.validation import run_validation_suite
-
-
-def main() -> int:
-    result = run_validation_suite()
-    print(json.dumps(result, indent=2, sort_keys=True))
-    return 0 if result["passed"] else 1
-
-
-if __name__ == "__main__":
-    raise SystemExit(main())

scratch/osint/server.py

@@ -1,564 +0,0 @@
-from __future__ import annotations
-
-import json
-import os
-from collections import Counter
-from functools import lru_cache
-from pathlib import Path
-from threading import Lock
-from typing import Any
-from uuid import uuid4
-
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.responses import FileResponse, HTMLResponse, JSONResponse
-
-from osint_env.api import (
-    OpenEnvActionRequest,
-    OpenEnvInferenceReportRequest,
-    OpenEnvInferenceReportResponse,
-    OpenEnvObservationModel,
-    OpenEnvResetRequest,
-    OpenEnvResponseEnvelope,
-    OpenEnvTaskSummary,
-)
-from osint_env.config import clone_environment_config, load_seeding_config, load_shared_config
-from osint_env.domain.models import Action, ActionType
-from osint_env.env.environment import OSINTEnvironment
-from osint_env.eval.leaderboard import load_leaderboard
-from osint_env.eval.runner import run_evaluation
-from osint_env.llm import build_llm_client
-from osint_env.viz import export_dashboard
-
-
-SPACE_CONFIG_PATH = Path(os.getenv("OSINT_ENV_CONFIG", "datasets/fixed_levels/shared_config_fixed_levels.json"))
-SPACE_SEED_PATH = Path(os.getenv("OSINT_ENV_SEED_FILE", "datasets/fixed_levels/seed_fixed_levels.json"))
-SPACE_PROVIDER = os.getenv("OSINT_SPACE_LLM_PROVIDER", "mock")
-SPACE_MODEL = os.getenv("OSINT_SPACE_LLM_MODEL", "gpt-4o-mini")
-SPACE_PORT = int(os.getenv("PORT", "7860"))
-SPACE_DASHBOARD = Path("artifacts/space_dashboard.html")
-LATEST_BASELINE_OUTPUT = Path("artifacts/baselines/openai_fixed_levels_latest.json")
-LATEST_EVALUATION_OUTPUT = Path("artifacts/latest_evaluation.json")
-OPENENV_SPEC_PATH = Path("openenv.yaml")
-
-_SESSION_LOCK = Lock()
-_SESSIONS: dict[str, OSINTEnvironment] = {}
-_RESET_COUNTER = 0
-_LATEST_SESSION_ID: str | None = None
-
-
-def _load_json(path: Path) -> dict[str, Any] | None:
-    if not path.exists():
-        return None
-    try:
-        payload = json.loads(path.read_text(encoding="utf-8"))
-    except (OSError, json.JSONDecodeError):
-        return None
-    return payload if isinstance(payload, dict) else None
-
-
-def _path_mtime(path: Path) -> float:
-    try:
-        return path.stat().st_mtime
-    except OSError:
-        return 0.0
-
-
-def _build_environment() -> OSINTEnvironment:
-    shared = load_shared_config(SPACE_CONFIG_PATH)
-    env_cfg = clone_environment_config(shared.environment)
-    if SPACE_SEED_PATH.exists():
-        env_cfg.seeding = load_seeding_config(SPACE_SEED_PATH)
-    env_cfg.llm.provider = SPACE_PROVIDER
-    env_cfg.llm.model = SPACE_MODEL
-    try:
-        llm = build_llm_client(env_cfg.llm)
-    except Exception:
-        env_cfg.llm.provider = "mock"
-        llm = build_llm_client(env_cfg.llm)
-    return OSINTEnvironment(env_cfg, llm=llm)
-
-
-def _serialize_observation(observation: Any) -> OpenEnvObservationModel:
-    return OpenEnvObservationModel(
-        tool_outputs=list(observation.tool_outputs),
-        graph_snapshot=dict(observation.graph_snapshot),
-        action_history=list(observation.action_history),
-        task=dict(observation.task),
-    )
-
-
-def _safe_session_info(info: dict[str, Any]) -> dict[str, Any]:
-    return {
-        "step_count": int(info.get("step_count", 0)),
-        "total_reward": float(info.get("total_reward", 0.0)),
-        "tool_calls": int(info.get("tool_calls", 0)),
-        "redundant_tool_calls": int(info.get("redundant_tool_calls", 0)),
-        "task_answer": str(info.get("task_answer", "")),
-        "agent_answer": "" if info.get("agent_answer") is None else str(info.get("agent_answer", "")),
-        "graph_f1": float(info.get("graph_f1", 0.0)),
-        "reward_components": dict(info.get("reward_components", {})),
-    }
-
-
-def _task_summaries(env: OSINTEnvironment) -> list[OpenEnvTaskSummary]:
-    return [
-        OpenEnvTaskSummary(
-            task_id=task.task_id,
-            task_type=task.task_type,
-            question=task.question,
-            difficulty=str(task.metadata.get("difficulty", "unknown")),
-            grader=(
-                dict(task.metadata.get("grader", {}))
-                if isinstance(task.metadata.get("grader"), dict)
-                else {
-                    "type": "exact_match",
-                    "answer_type": "node_id",
-                    "case_sensitive": True,
-                }
-            ),
-        )
-        for task in env.tasks
-    ]
-
-
-def _resolve_task_index(env: OSINTEnvironment, request: OpenEnvResetRequest) -> int:
-    global _RESET_COUNTER
-    if request.task_index is not None:
-        task_index = int(request.task_index)
-        if task_index < 0 or task_index >= len(env.tasks):
-            raise HTTPException(status_code=400, detail=f"Invalid task_index {task_index}")
-        return task_index
-    if request.task_id:
-        for idx, task in enumerate(env.tasks):
-            if task.task_id == request.task_id:
-                return idx
-        raise HTTPException(status_code=400, detail=f"Unknown task_id {request.task_id}")
-    with _SESSION_LOCK:
-        task_index = _RESET_COUNTER % max(1, len(env.tasks))
-        _RESET_COUNTER += 1
-    return task_index
-
-
-def _get_session_env(session_id: str) -> OSINTEnvironment:
-    with _SESSION_LOCK:
-        env = _SESSIONS.get(session_id)
-    if env is None:
-        raise HTTPException(status_code=404, detail=f"Unknown session_id {session_id}")
-    return env
-
-
-def _store_session(session_id: str, env: OSINTEnvironment) -> None:
-    global _LATEST_SESSION_ID
-    with _SESSION_LOCK:
-        _SESSIONS[session_id] = env
-        _LATEST_SESSION_ID = session_id
-
-
-def _latest_session_id() -> str:
-    with _SESSION_LOCK:
-        if _LATEST_SESSION_ID and _LATEST_SESSION_ID in _SESSIONS:
-            return _LATEST_SESSION_ID
-        if _SESSIONS:
-            return next(reversed(_SESSIONS))
-    raise HTTPException(status_code=404, detail="No active session. Call /reset first.")
-
-
-def _resolve_session_id(session_id: str | None) -> str:
-    token = str(session_id or "").strip()
-    if token:
-        return token
-    return _latest_session_id()
-
-
-def _task_lookup(env: OSINTEnvironment) -> dict[str, Any]:
-    return {task.task_id: task for task in env.tasks}
-
-
-def _normalize_episode_rows(env: OSINTEnvironment, episodes: list[dict[str, Any]]) -> list[dict[str, Any]]:
-    tasks_by_id = _task_lookup(env)
-    normalized: list[dict[str, Any]] = []
-    for episode in episodes:
-        row = dict(episode)
-        task = tasks_by_id.get(str(row.get("task_id", "")))
-        if task is not None:
-            row.setdefault("task_type", task.task_type)
-            row.setdefault("question", task.question)
-            row.setdefault("task_answer", task.answer)
-            row.setdefault(
-                "truth_edges",
-                [
-                    {
-                        "src": edge.src,
-                        "rel": edge.rel,
-                        "dst": edge.dst,
-                        "confidence": float(edge.confidence),
-                    }
-                    for edge in task.supporting_edges
-                ],
-            )
-        row.setdefault("pred_edges", [])
-        row.setdefault("reward_components", {})
-        row.setdefault("graph_f1", 0.0)
-        row.setdefault("reward", 0.0)
-        row.setdefault("steps", 0)
-        row.setdefault("tool_calls", 0)
-        row.setdefault("success", 0)
-        normalized.append(row)
-    return normalized
-
-
-@lru_cache(maxsize=1)
-def _base_environment_snapshot() -> dict[str, Any]:
-    env = _build_environment()
-    difficulty_counts = Counter(str(task.metadata.get("difficulty", "unknown")) for task in env.tasks)
-    return {
-        "task_count": len(env.tasks),
-        "difficulty_counts": dict(difficulty_counts),
-        "action_space": ["CALL_TOOL", "ADD_EDGE", "ANSWER"],
-        "observation_space": {
-            "tool_outputs": "Last tool results and memory hits.",
-            "graph_snapshot": "Current working graph edge snapshot.",
-            "action_history": "Recent action/reward trace.",
-            "task": "Task id, task type, and question.",
-        },
-        "task_types": sorted({task.task_type for task in env.tasks}),
-        "config": {
-            "seed": env.config.seed,
-            "max_steps": env.config.max_steps,
-            "swarm_enabled": env.config.swarm.enabled,
-            "llm_provider": env.config.llm.provider,
-            "llm_model": env.config.llm.model,
-        },
-    }
-
-
-@lru_cache(maxsize=1)
-def _preview_snapshot() -> dict[str, Any]:
-    env = _build_environment()
-    evaluation = run_evaluation(env, episodes=3, return_details=True, llm=build_llm_client(env.config.llm))
-    dashboard_path = export_dashboard(
-        env=env,
-        evaluation=evaluation,
-        leaderboard_records=[],
-        output_path=str(SPACE_DASHBOARD),
-    )
-    snapshot = dict(_base_environment_snapshot())
-    snapshot["summary"] = evaluation["summary"]
-    snapshot["dashboard_path"] = dashboard_path
-    return snapshot
-
-
-def _space_snapshot() -> dict[str, Any]:
-    snapshot = dict(_base_environment_snapshot())
-
-    baseline_payload = _load_json(LATEST_BASELINE_OUTPUT)
-    evaluation_payload = _load_json(LATEST_EVALUATION_OUTPUT)
-
-    candidates: list[tuple[float, str, dict[str, Any]]] = []
-    if baseline_payload is not None and isinstance(baseline_payload.get("summary"), dict):
-        candidates.append((_path_mtime(LATEST_BASELINE_OUTPUT), "baseline_output", baseline_payload))
-    if evaluation_payload is not None and isinstance(evaluation_payload.get("summary"), dict):
-        candidates.append((_path_mtime(LATEST_EVALUATION_OUTPUT), "latest_evaluation", evaluation_payload))
-
-    if candidates:
-        _, source, payload = max(candidates, key=lambda item: item[0])
-        snapshot["summary"] = dict(payload["summary"])
-        snapshot["source"] = source
-        if source == "baseline_output":
-            dashboard_path = Path(
-                str(
-                    ((payload.get("run") or {}).get("dashboard_path"))
-                    or "artifacts/baselines/openai_fixed_levels_dashboard.html"
-                )
-            )
-            if dashboard_path.exists():
-                snapshot["dashboard_path"] = str(dashboard_path)
-            return snapshot
-
-        env = _build_environment()
-        dashboard_path = export_dashboard(
-            env=env,
-            evaluation=payload,
-            leaderboard_records=[],
-            output_path=str(SPACE_DASHBOARD),
-        )
-        snapshot["dashboard_path"] = dashboard_path
-        return snapshot
-
-    preview = _preview_snapshot()
-    preview["source"] = "preview"
-    return preview
-
-
-app = FastAPI(title="OSINT OpenEnv Space", version="0.1.0")
-
-
-@app.get("/", response_class=HTMLResponse)
-def home() -> str:
-    snapshot = _space_snapshot()
-    summary = snapshot["summary"]
-    difficulty_html = "".join(
-        f"<li><strong>{level}</strong>: {count}</li>"
-        for level, count in sorted(snapshot["difficulty_counts"].items())
-    )
-    task_type_html = "".join(f"<li>{task_type}</li>" for task_type in snapshot["task_types"])
-    return f"""<!doctype html>
-<html lang="en">
-<head>
-  <meta charset="utf-8" />
-  <meta name="viewport" content="width=device-width, initial-scale=1" />
-  <title>OSINT OpenEnv Space</title>
-  <style>
-    :root {{
-      --ink: #13212d;
-      --muted: #4d5b69;
-      --line: #d8e2eb;
-      --card: #ffffff;
-      --bg: #f6fafc;
-      --brand: #0f766e;
-      --accent: #b45309;
-    }}
-    * {{ box-sizing: border-box; }}
-    body {{
-      margin: 0;
-      font-family: "Segoe UI", sans-serif;
-      color: var(--ink);
-      background:
-        radial-gradient(circle at top left, rgba(15,118,110,0.12), transparent 30%),
-        radial-gradient(circle at top right, rgba(180,83,9,0.10), transparent 28%),
-        var(--bg);
-    }}
-    .wrap {{ max-width: 1120px; margin: 0 auto; padding: 24px; }}
-    .hero, .grid {{ display: grid; gap: 16px; }}
-    .hero {{ grid-template-columns: 1.5fr 1fr; }}
-    .grid {{ grid-template-columns: repeat(3, minmax(0, 1fr)); margin-top: 16px; }}
-    .card {{
-      background: var(--card);
-      border: 1px solid var(--line);
-      border-radius: 18px;
-      padding: 18px;
-      box-shadow: 0 12px 24px rgba(19, 33, 45, 0.06);
-    }}
-    h1, h2 {{ margin-top: 0; }}
-    .muted {{ color: var(--muted); }}
-    .stats {{ display: grid; grid-template-columns: repeat(2, minmax(0, 1fr)); gap: 10px; }}
-    .stat {{ border: 1px dashed var(--line); border-radius: 12px; padding: 10px; }}
-    .stat .k {{ font-size: 12px; color: var(--muted); text-transform: uppercase; }}
-    .stat .v {{ font-size: 22px; font-weight: 700; }}
-    a.button {{
-      display: inline-block;
-      padding: 10px 14px;
-      border-radius: 12px;
-      text-decoration: none;
-      color: white;
-      background: var(--brand);
-      margin-right: 10px;
-    }}
-    a.link {{
-      color: var(--accent);
-      text-decoration: none;
-      font-weight: 600;
-    }}
-    ul {{ padding-left: 18px; }}
-    code {{
-      background: #f1f5f9;
-      border-radius: 6px;
-      padding: 2px 6px;
-    }}
-    @media (max-width: 900px) {{
-      .hero, .grid {{ grid-template-columns: 1fr; }}
-    }}
-  </style>
-</head>
-<body>
-  <div class="wrap">
-    <div class="hero">
-      <section class="card">
-        <h1>OSINT OpenEnv Space</h1>
-        <p class="muted">A containerized OpenEnv-compatible benchmark for synthetic OSINT reasoning over profiles, forum threads, posts, aliases, organizations, locations, and event links.</p>
-        <p>The Space boots with the fixed-level benchmark so visitors get a stable environment snapshot instead of a different graph every restart.</p>
-        <a class="button" href="/dashboard">Open Dashboard</a>
-        <a class="link" href="/api/environment">Environment JSON</a>
-      </section>
-      <section class="card">
-        <h2>Included Snapshot</h2>
-        <div class="stats">
-          <div class="stat"><div class="k">Tasks</div><div class="v">{snapshot["task_count"]}</div></div>
-          <div class="stat"><div class="k">Provider</div><div class="v">{snapshot["config"]["llm_provider"]}</div></div>
-          <div class="stat"><div class="k">Score</div><div class="v">{summary["leaderboard_score"]:.3f}</div></div>
-          <div class="stat"><div class="k">Success</div><div class="v">{summary["task_success_rate"]:.3f}</div></div>
-        </div>
-      </section>
-    </div>
-
-    <div class="grid">
-      <section class="card">
-        <h2>Action Space</h2>
-        <ul>
-          <li><code>CALL_TOOL</code>: query platform views or semantic memory.</li>
-          <li><code>ADD_EDGE</code>: add a hypothesized relation to the working graph.</li>
-          <li><code>ANSWER</code>: submit the final node id answer.</li>
-        </ul>
-      </section>
-      <section class="card">
-        <h2>Difficulty Mix</h2>
-        <ul>{difficulty_html}</ul>
-      </section>
-      <section class="card">
-        <h2>Task Families</h2>
-        <ul>{task_type_html}</ul>
-      </section>
-    </div>
-  </div>
-</body>
-</html>"""
-
-
-@app.get("/healthz")
-def healthz() -> JSONResponse:
-    return JSONResponse({"status": "ok"})
-
-
-@app.get("/health")
-def health() -> JSONResponse:
-    return healthz()
-
-
-@app.get("/openenv.yaml")
-def openenv_spec() -> FileResponse:
-    return FileResponse(OPENENV_SPEC_PATH, media_type="text/yaml")
-
-
-@app.get("/api/environment")
-def environment_metadata() -> JSONResponse:
-    return JSONResponse(_space_snapshot())
-
-
-@app.get("/openenv/tasks", response_model=list[OpenEnvTaskSummary])
-def openenv_tasks() -> list[OpenEnvTaskSummary]:
-    env = _build_environment()
-    return _task_summaries(env)
-
-
-@app.post("/openenv/reset", response_model=OpenEnvResponseEnvelope)
-@app.post("/openenv/reset/", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-@app.post("/reset", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-@app.post("/reset/", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-async def openenv_reset(request: Request) -> OpenEnvResponseEnvelope:
-    env = _build_environment()
-    raw_body = await request.body()
-    if not raw_body.strip():
-        payload: dict[str, Any] = {}
-    else:
-        try:
-            parsed_payload = json.loads(raw_body)
-        except json.JSONDecodeError as exc:
-            raise HTTPException(status_code=400, detail="Reset body must be valid JSON") from exc
-        if parsed_payload is None:
-            payload = {}
-        elif isinstance(parsed_payload, dict):
-            payload = parsed_payload
-        else:
-            raise HTTPException(status_code=400, detail="Reset body must be a JSON object")
-
-    try:
-        reset_request = OpenEnvResetRequest.model_validate(payload)
-    except Exception as exc:
-        raise HTTPException(status_code=422, detail="Invalid reset request payload") from exc
-
-    env._task_idx = _resolve_task_index(env, reset_request)
-    observation = env.reset()
-    session_id = str(uuid4())
-    _store_session(session_id, env)
-    return OpenEnvResponseEnvelope(
-        session_id=session_id,
-        observation=_serialize_observation(observation),
-        reward=0.0,
-        done=False,
-        info=_safe_session_info(env._info()),
-    )
-
-
-@app.post("/openenv/step", response_model=OpenEnvResponseEnvelope)
-@app.post("/openenv/step/", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-@app.post("/step", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-@app.post("/step/", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-def openenv_step(request: OpenEnvActionRequest) -> OpenEnvResponseEnvelope:
-    session_id = _resolve_session_id(request.session_id)
-    env = _get_session_env(session_id)
-    action_type_raw = request.resolved_action_type().strip()
-    if not action_type_raw:
-        raise HTTPException(status_code=400, detail="Missing action_type")
-    try:
-        action_type = ActionType(action_type_raw)
-    except ValueError as exc:
-        raise HTTPException(status_code=400, detail=f"Unsupported action_type {action_type_raw}") from exc
-    observation, reward, done, info = env.step(Action(action_type=action_type, payload=request.resolved_payload()))
-    return OpenEnvResponseEnvelope(
-        session_id=session_id,
-        observation=_serialize_observation(observation),
-        reward=float(reward),
-        done=bool(done),
-        info=_safe_session_info(info),
-    )
-
-
-def _state_response(session_id: str) -> OpenEnvResponseEnvelope:
-    env = _get_session_env(session_id)
-    if env.state is None:
-        raise HTTPException(status_code=400, detail="Session has not been reset yet")
-    return OpenEnvResponseEnvelope(
-        session_id=session_id,
-        observation=_serialize_observation(env._observation()),
-        reward=0.0,
-        done=bool(env.state.done),
-        info=_safe_session_info(env._info()),
-    )
-
-
-@app.get("/openenv/state/{session_id}", response_model=OpenEnvResponseEnvelope)
-def openenv_state(session_id: str) -> OpenEnvResponseEnvelope:
-    return _state_response(session_id)
-
-
-@app.get("/openenv/state", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-@app.get("/state", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-@app.get("/state/", response_model=OpenEnvResponseEnvelope, include_in_schema=False)
-def openenv_state_latest() -> OpenEnvResponseEnvelope:
-    return _state_response(_latest_session_id())
-
-
-@app.post("/openenv/report_inference", response_model=OpenEnvInferenceReportResponse)
-def openenv_report_inference(request: OpenEnvInferenceReportRequest) -> OpenEnvInferenceReportResponse:
-    env = _build_environment()
-    normalized_episodes = _normalize_episode_rows(env, list(request.episodes))
-    payload = {
-        "run": dict(request.run),
-        "summary": dict(request.summary),
-        "episodes": normalized_episodes,
-    }
-    LATEST_EVALUATION_OUTPUT.parent.mkdir(parents=True, exist_ok=True)
-    LATEST_EVALUATION_OUTPUT.write_text(json.dumps(payload, indent=2, sort_keys=True), encoding="utf-8")
-    dashboard_path = export_dashboard(
-        env=env,
-        evaluation=payload,
-        leaderboard_records=load_leaderboard("artifacts/baselines/openai_fixed_levels_leaderboard.json"),
-        output_path=str(SPACE_DASHBOARD),
-    )
-    return OpenEnvInferenceReportResponse(
-        status="ok",
-        output_path=str(LATEST_EVALUATION_OUTPUT),
-        dashboard_path=str(dashboard_path),
-    )
-
-
-@app.get("/dashboard")
-def dashboard() -> FileResponse:
-    snapshot = _space_snapshot()
-    return FileResponse(snapshot["dashboard_path"], media_type="text/html")
-
-
-if __name__ == "__main__":
-    import uvicorn
-
-    uvicorn.run("server:app", host="0.0.0.0", port=SPACE_PORT)

scratch/osint/server/app.py

@@ -1,26 +0,0 @@
-from __future__ import annotations
-
-import importlib.util
-import os
-from pathlib import Path
-
-import uvicorn
-
-
-_ROOT_SERVER_PATH = Path(__file__).resolve().parents[1] / "server.py"
-_SPEC = importlib.util.spec_from_file_location("osint_root_server", _ROOT_SERVER_PATH)
-if _SPEC is None or _SPEC.loader is None:
-    raise RuntimeError(f"Unable to load server module from {_ROOT_SERVER_PATH}")
-
-_MODULE = importlib.util.module_from_spec(_SPEC)
-_SPEC.loader.exec_module(_MODULE)
-app = _MODULE.app
-
-
-def main() -> None:
-    port = int(os.getenv("PORT", "7860"))
-    uvicorn.run("server.app:app", host="0.0.0.0", port=port)
-
-
-if __name__ == "__main__":
-    main()

scratch/osint/src/osint_env/__init__.py

@@ -1,5 +0,0 @@
-"""OSINT RL environment package."""
-
-from .env.environment import OSINTEnvironment
-
-__all__ = ["OSINTEnvironment"]

scratch/osint/src/osint_env/agents/__init__.py

@@ -1,7 +0,0 @@
-"""Agent implementations."""
-
-from osint_env.agents.single_agent import SingleAgentRunner
-from osint_env.agents.swarm_agent import SwarmAgentRunner
-
-__all__ = ["SingleAgentRunner", "SwarmAgentRunner"]
-

scratch/osint/src/osint_env/agents/single_agent.py

@@ -1,41 +0,0 @@
-from __future__ import annotations
-
-from osint_env.domain.models import Action, ActionType
-from osint_env.env.environment import OSINTEnvironment
-from osint_env.llm.interface import LLMClient, RuleBasedMockLLM
-
-
-class SingleAgentRunner:
-    def __init__(self, env: OSINTEnvironment, llm: LLMClient | None = None):
-        self.env = env
-        self.llm = llm or RuleBasedMockLLM()
-
-    def run_episode(self) -> dict:
-        obs = self.env.reset()
-        done = False
-        info = {}
-        while not done:
-            messages = [{"role": "system", "content": f"question: {obs.task['question']}"}]
-            tools = []
-            try:
-                llm_resp = self.llm.generate(messages, tools)
-                planned_calls = llm_resp.tool_calls[:2]
-            except Exception:
-                planned_calls = []
-
-            for call in planned_calls:
-                obs, _, done, info = self.env.step(Action(ActionType.CALL_TOOL, call))
-                if done:
-                    break
-            if done:
-                break
-            answer_guess = self._heuristic_answer(obs.task["question"])
-            obs, _, done, info = self.env.step(Action(ActionType.ANSWER, {"answer": answer_guess}))
-        return info
-
-    @staticmethod
-    def _heuristic_answer(question: str) -> str:
-        for token in question.replace("?", "").split():
-            if token.startswith("alias_") or token.startswith("user_"):
-                return token
-        return "unknown"

scratch/osint/src/osint_env/agents/swarm_agent.py

@@ -1,209 +0,0 @@
-from __future__ import annotations
-
-import re
-from typing import Any
-
-from osint_env.domain.models import Action, ActionType
-from osint_env.env.environment import OSINTEnvironment
-from osint_env.env.spawn_reward_hooks import critical_steps, parl_style_spawn_reward
-from osint_env.llm.interface import LLMClient, RuleBasedMockLLM
-
-
-class SwarmAgentRunner:
-    """Low-width multi-agent orchestrator over a single environment episode."""
-
-    def __init__(self, env: OSINTEnvironment, llm: LLMClient | None = None):
-        self.env = env
-        self.llm = llm or RuleBasedMockLLM()
-
-    def run_episode(self) -> dict[str, Any]:
-        obs = self.env.reset()
-        done = False
-        info: dict[str, Any] = {}
-
-        swarm_cfg = self.env.config.swarm
-        spawn_cfg = self.env.config.spawn_reward
-
-        spawn_count = 0
-        finished_subtasks = 0
-        depth_used = 0
-        max_breadth_used = 0
-
-        stage_main_steps: list[int] = []
-        stage_sub_steps: list[list[int]] = []
-
-        for _ in range(max(1, swarm_cfg.planner_rounds)):
-            if done:
-                break
-
-            active_agents = max(1, min(swarm_cfg.max_agents, swarm_cfg.max_breadth, swarm_cfg.max_width))
-            max_breadth_used = max(max_breadth_used, active_agents)
-            depth_used += 1
-            spawn_count += active_agents
-            stage_main_steps.append(1)
-
-            stage_steps: list[int] = []
-            for agent_idx in range(active_agents):
-                if done:
-                    break
-
-                steps_for_agent = 0
-                role = self._agent_role(agent_idx)
-                planned_calls = self._tool_plan(
-                    obs=obs,
-                    agent_idx=agent_idx,
-                    role=role,
-                    limit=swarm_cfg.tools_per_agent,
-                )
-                for call in planned_calls:
-                    obs, _, done, info = self.env.step(Action(ActionType.CALL_TOOL, call))
-                    steps_for_agent += 1
-                    if done:
-                        break
-
-                if not done:
-                    edge_payload = self._edge_plan(agent_idx=agent_idx)
-                    if edge_payload is not None:
-                        obs, _, done, info = self.env.step(Action(ActionType.ADD_EDGE, edge_payload))
-                        steps_for_agent += 1
-
-                if steps_for_agent > 0:
-                    finished_subtasks += 1
-                stage_steps.append(steps_for_agent)
-
-            stage_sub_steps.append(stage_steps)
-
-            if depth_used >= swarm_cfg.max_depth:
-                break
-
-        if not done:
-            answer_guess = self._vote_answer()
-            obs, _, done, info = self.env.step(Action(ActionType.ANSWER, {"answer": answer_guess}))
-
-        crit_steps = critical_steps(
-            main_steps=stage_main_steps or [1],
-            parallel_subagent_steps=stage_sub_steps or [[]],
-        )
-
-        base_total = float(info.get("total_reward", 0.0))
-        shaped_total = parl_style_spawn_reward(
-            task_outcome_reward=base_total,
-            spawn_count=spawn_count,
-            finished_subtasks=finished_subtasks,
-            critical_steps=max(1, crit_steps),
-            lambda_parallel=spawn_cfg.lambda_parallel,
-            lambda_finish=spawn_cfg.lambda_finish,
-            anneal=spawn_cfg.anneal,
-            breadth=max_breadth_used,
-            depth=depth_used,
-            max_parallel_hint=spawn_cfg.max_parallel_hint,
-        )
-        spawn_aux = shaped_total - base_total
-
-        components = dict(info.get("reward_components", {}))
-        components["spawn_auxiliary"] = components.get("spawn_auxiliary", 0.0) + float(spawn_aux)
-        components["spawn_count"] = float(spawn_count)
-        components["spawn_finished_subtasks"] = float(finished_subtasks)
-        components["spawn_critical_steps"] = float(crit_steps)
-        components["spawn_depth"] = float(depth_used)
-        components["spawn_breadth"] = float(max_breadth_used)
-
-        info["total_reward"] = shaped_total
-        info["reward_components"] = components
-        info["spawn_count"] = spawn_count
-        info["spawn_finished_subtasks"] = finished_subtasks
-        info["spawn_critical_steps"] = crit_steps
-        info["spawn_depth"] = depth_used
-        info["spawn_breadth"] = max_breadth_used
-        info["swarm_roles"] = [self._agent_role(i) for i in range(max_breadth_used)]
-
-        if self.env.state is not None:
-            self.env.state.total_reward = shaped_total
-            self.env.state.reward_components.update(components)
-
-        return info
-
-    @staticmethod
-    def _agent_role(agent_idx: int) -> str:
-        roles = ["explorer", "linker", "reasoner"]
-        return roles[agent_idx % len(roles)]
-
-    def _tool_plan(self, obs: Any, agent_idx: int, role: str, limit: int) -> list[dict[str, Any]]:
-        messages = [
-            {
-                "role": "system",
-                "content": (
-                    f"question: {obs.task['question']}\n"
-                    f"agent_role: {role}_{agent_idx}\n"
-                    "Return concise tool plan."
-                ),
-            }
-        ]
-        try:
-            response = self.llm.generate(messages, tools=[])
-        except Exception:
-            response = None
-
-        calls: list[dict[str, Any]] = []
-        for call in (response.tool_calls if response is not None else []):
-            if not isinstance(call, dict):
-                continue
-            tool_name = str(call.get("tool_name", "")).strip()
-            args = call.get("args", {})
-            if not tool_name or not isinstance(args, dict):
-                continue
-            calls.append({"tool_name": tool_name, "args": args})
-            if len(calls) >= max(1, limit):
-                break
-
-        if calls:
-            return calls
-
-        question = str(obs.task.get("question", "")).lower()
-        if role == "explorer":
-            if "event" in question:
-                return [{"tool_name": "search_threads", "args": {"topic": "security"}}]
-            return [{"tool_name": "search_posts", "args": {"query": "Update"}}]
-
-        if role == "linker":
-            if "alias" in question:
-                return [{"tool_name": "search_posts", "args": {"query": "alias"}}]
-            return [{"tool_name": "search_people", "args": {"org": "Apex"}}]
-
-        if role == "reasoner":
-            return [{"tool_name": "search_memory", "args": {"query": obs.task.get("question", ""), "k": 5}}]
-
-        if "alias" in question:
-            return [{"tool_name": "search_posts", "args": {"query": "Update"}}]
-
-        user_tokens = re.findall(r"\buser_[a-zA-Z0-9_]+\b", question)
-        if user_tokens:
-            return [{"tool_name": "get_profile", "args": {"user_id": user_tokens[0]}}]
-
-        return [{"tool_name": "search_people", "args": {"org": "Apex"}}]
-
-    def _edge_plan(self, agent_idx: int) -> dict[str, Any] | None:
-        if self.env.state is None or not self.env.state.task.supporting_edges:
-            return None
-        edge = self.env.state.task.supporting_edges[agent_idx % len(self.env.state.task.supporting_edges)]
-        return {
-            "src": edge.src,
-            "rel": edge.rel,
-            "dst": edge.dst,
-            "confidence": float(edge.confidence),
-        }
-
-    def _vote_answer(self) -> str:
-        if self.env.state is None:
-            return "unknown"
-
-        truth = {(e.src, e.rel, e.dst) for e in self.env.state.task.supporting_edges}
-        pred = {(e.src, e.rel, e.dst) for e in self.env.memory_graph.edges}
-        if truth & pred:
-            return self.env.state.task.answer
-
-        question = self.env.state.task.question
-        for token in question.replace("?", "").split():
-            if token.startswith("alias_") or token.startswith("user_"):
-                return token
-        return "unknown"

scratch/osint/src/osint_env/api/__init__.py

@@ -1,19 +0,0 @@
-from osint_env.api.models import (
-    OpenEnvActionRequest,
-    OpenEnvInferenceReportRequest,
-    OpenEnvInferenceReportResponse,
-    OpenEnvObservationModel,
-    OpenEnvResetRequest,
-    OpenEnvResponseEnvelope,
-    OpenEnvTaskSummary,
-)
-
-__all__ = [
-    "OpenEnvActionRequest",
-    "OpenEnvInferenceReportRequest",
-    "OpenEnvInferenceReportResponse",
-    "OpenEnvObservationModel",
-    "OpenEnvResetRequest",
-    "OpenEnvResponseEnvelope",
-    "OpenEnvTaskSummary",
-]

scratch/osint/src/osint_env/api/models.py

@@ -1,73 +0,0 @@
-from __future__ import annotations
-
-from typing import Any
-
-from pydantic import BaseModel, Field
-
-
-class OpenEnvTaskSummary(BaseModel):
-    task_id: str
-    task_type: str
-    question: str
-    difficulty: str = "unknown"
-    grader: dict[str, Any] = Field(default_factory=dict)
-
-
-class OpenEnvObservationModel(BaseModel):
-    tool_outputs: list[dict[str, Any]]
-    graph_snapshot: dict[str, Any]
-    action_history: list[dict[str, Any]]
-    task: dict[str, Any]
-
-
-class OpenEnvResetRequest(BaseModel):
-    task_id: str | None = None
-    task_index: int | None = None
-
-
-class OpenEnvActionRequest(BaseModel):
-    session_id: str | None = Field(
-        default=None,
-        description="Session identifier. Optional for /step compatibility alias, which uses the latest session.",
-    )
-    action_type: str | None = Field(default=None, description="One of CALL_TOOL, ADD_EDGE, ANSWER.")
-    payload: dict[str, Any] = Field(default_factory=dict)
-    action: dict[str, Any] | None = None
-
-    def resolved_action_type(self) -> str:
-        if self.action_type:
-            return str(self.action_type)
-        if isinstance(self.action, dict):
-            nested = self.action.get("action_type")
-            if nested:
-                return str(nested)
-        return ""
-
-    def resolved_payload(self) -> dict[str, Any]:
-        if self.payload:
-            return dict(self.payload)
-        if isinstance(self.action, dict):
-            nested = self.action.get("payload")
-            if isinstance(nested, dict):
-                return dict(nested)
-        return {}
-
-
-class OpenEnvResponseEnvelope(BaseModel):
-    session_id: str
-    observation: OpenEnvObservationModel
-    reward: float
-    done: bool
-    info: dict[str, Any]
-
-
-class OpenEnvInferenceReportRequest(BaseModel):
-    run: dict[str, Any] = Field(default_factory=dict)
-    summary: dict[str, Any]
-    episodes: list[dict[str, Any]] = Field(default_factory=list)
-
-
-class OpenEnvInferenceReportResponse(BaseModel):
-    status: str
-    output_path: str
-    dashboard_path: str

scratch/osint/src/osint_env/baselines/__init__.py

@@ -1,4 +0,0 @@
-from osint_env.baselines.openai_runner import OpenAIBaselineConfig, OpenAIBaselineRunner
-
-__all__ = ["OpenAIBaselineConfig", "OpenAIBaselineRunner"]
-

scratch/osint/src/osint_env/baselines/openai_runner.py

@@ -1,533 +0,0 @@
-from __future__ import annotations
-
-import json
-from dataclasses import asdict, dataclass
-from pathlib import Path
-from time import perf_counter
-from typing import Any
-
-from osint_env.config import clone_environment_config, load_seeding_config, load_shared_config
-from osint_env.domain.models import Action, ActionType, Edge
-from osint_env.env.environment import OSINTEnvironment
-from osint_env.env.reward import compute_graph_f1
-from osint_env.eval.leaderboard import append_leaderboard_record, load_leaderboard
-from osint_env.eval.metrics import EvalMetrics
-from osint_env.viz import export_dashboard
-
-
-SYSTEM_PROMPT = """You are an OSINT benchmark agent operating in a synthetic OpenEnv task.
-
-Available actions are provided as function tools. On every turn, call exactly one tool.
-
-Rules:
-- Solve the question using only tool outputs and the current graph snapshot.
-- When you have enough evidence, call submit_answer with the exact node id string.
-- Questions may contain exact node ids such as alias_*, user_*, post_*, thr_*, org_*, loc_*, and event_*.
-- Prefer direct id lookups when an exact id is present in the question.
-- get_post and get_thread retrieve exact seeded records by id.
-- Use add_edge only for relationships strongly supported by the evidence you have already collected.
-- Prefer concise, high-signal tool queries.
-- Never guess free-form prose when a node id answer is required.
-"""
-
-
-@dataclass(slots=True)
-class OpenAIBaselineConfig:
-    shared_config_path: str = "datasets/fixed_levels/shared_config_fixed_levels.json"
-    seed_file: str = "datasets/fixed_levels/seed_fixed_levels.json"
-    output_path: str = "artifacts/baselines/openai_fixed_levels_latest.json"
-    leaderboard_path: str = "artifacts/baselines/openai_fixed_levels_leaderboard.json"
-    dashboard_path: str = "artifacts/baselines/openai_fixed_levels_dashboard.html"
-    run_name: str = "openai_fixed_levels_baseline"
-    model: str = "gpt-5-nano"
-    base_url: str = "https://api.openai.com/v1"
-    api_key: str = ""
-    api_key_env: str = "OPENAI_API_KEY"
-    temperature: float = 0.0
-    max_tokens: int = 256
-    timeout_seconds: int = 60
-    episodes: int = 30
-    max_steps: int = 8
-    seed: int | None = 7
-    append_leaderboard: bool = True
-
-
-def _tool_schema(
-    name: str,
-    description: str,
-    properties: dict[str, Any],
-    required: list[str],
-) -> dict[str, Any]:
-    return {
-        "type": "function",
-        "function": {
-            "name": name,
-            "description": description,
-            "parameters": {
-                "type": "object",
-                "properties": properties,
-                "required": required,
-                "additionalProperties": False,
-            },
-        },
-    }
-
-
-def build_action_tools() -> list[dict[str, Any]]:
-    return [
-        _tool_schema(
-            "search_posts",
-            "Search microblog posts by substring over post text, post id, author id, canonical user id, or referenced entity ids/names.",
-            {"query": {"type": "string", "description": "Substring to search for in post text."}},
-            ["query"],
-        ),
-        _tool_schema(
-            "get_post",
-            "Fetch a specific microblog post by exact post id.",
-            {"post_id": {"type": "string", "description": "Post node id such as post_midnight_manifest."}},
-            ["post_id"],
-        ),
-        _tool_schema(
-            "get_user_posts",
-            "Fetch posts authored by a user or alias id. Alias ids are resolved to the canonical user and vice versa.",
-            {"user_id": {"type": "string", "description": "User or alias node id."}},
-            ["user_id"],
-        ),
-        _tool_schema(
-            "get_mentions",
-            "Fetch posts that mention a given canonical user id.",
-            {"user_id": {"type": "string", "description": "Canonical user node id."}},
-            ["user_id"],
-        ),
-        _tool_schema(
-            "search_threads",
-            "Search forum threads by exact topic name.",
-            {"topic": {"type": "string", "description": "Thread topic such as security or ai."}},
-            ["topic"],
-        ),
-        _tool_schema(
-            "get_thread",
-            "Fetch a specific forum thread by id.",
-            {"thread_id": {"type": "string", "description": "Thread node id."}},
-            ["thread_id"],
-        ),
-        _tool_schema(
-            "get_user_activity",
-            "Fetch a user's known forum activity.",
-            {"user_id": {"type": "string", "description": "Canonical user node id."}},
-            ["user_id"],
-        ),
-        _tool_schema(
-            "get_profile",
-            "Fetch a profile record by canonical user id or alias id.",
-            {"user_id": {"type": "string", "description": "Canonical user node id or alias id."}},
-            ["user_id"],
-        ),
-        _tool_schema(
-            "search_people",
-            "Search profiles by name, alias id, organization name, or organization id.",
-            {
-                "name": {"type": "string", "description": "Optional name substring.", "default": ""},
-                "org": {"type": "string", "description": "Optional organization substring.", "default": ""},
-            },
-            [],
-        ),
-        _tool_schema(
-            "get_connections",
-            "Fetch explicit profile connections for a user or alias id.",
-            {"user_id": {"type": "string", "description": "Canonical user node id or alias id."}},
-            ["user_id"],
-        ),
-        _tool_schema(
-            "search_memory",
-            "Search semantic memory over prior observations and tool outputs.",
-            {
-                "query": {"type": "string", "description": "Memory retrieval query."},
-                "k": {"type": "integer", "description": "Top-k matches.", "default": 5},
-            },
-            ["query"],
-        ),
-        _tool_schema(
-            "add_edge",
-            "Add a supported graph edge to the working memory graph.",
-            {
-                "src": {"type": "string"},
-                "rel": {"type": "string"},
-                "dst": {"type": "string"},
-                "confidence": {"type": "number", "default": 1.0},
-            },
-            ["src", "rel", "dst"],
-        ),
-        _tool_schema(
-            "submit_answer",
-            "Finish the episode by submitting the exact node id answer.",
-            {"answer": {"type": "string", "description": "Exact node id answer for the task."}},
-            ["answer"],
-        ),
-    ]
-
-
-def _message_text(message: Any) -> str:
-    content = getattr(message, "content", "")
-    if isinstance(content, str):
-        return content
-    if isinstance(content, list):
-        parts: list[str] = []
-        for item in content:
-            if isinstance(item, dict) and item.get("type") == "text":
-                parts.append(str(item.get("text", "")))
-            else:
-                text = getattr(item, "text", None)
-                if text:
-                    parts.append(str(text))
-        return "\n".join(part for part in parts if part)
-    return str(content or "")
-
-
-def _safe_info(info: dict[str, Any]) -> dict[str, Any]:
-    return {
-        "step_count": int(info.get("step_count", 0)),
-        "total_reward": float(info.get("total_reward", 0.0)),
-        "tool_calls": int(info.get("tool_calls", 0)),
-        "redundant_tool_calls": int(info.get("redundant_tool_calls", 0)),
-        "reward_components": dict(info.get("reward_components", {})),
-    }
-
-
-def _observation_payload(env: OSINTEnvironment, observation: Any, step_limit: int) -> dict[str, Any]:
-    task = dict(observation.task)
-    return {
-        "task": {
-            "task_id": task.get("task_id", ""),
-            "task_type": task.get("task_type", ""),
-            "question": task.get("question", ""),
-        },
-        "remaining_steps": max(0, step_limit - int(env.state.step_count if env.state else 0)),
-        "recent_tool_outputs": list(observation.tool_outputs),
-        "graph_snapshot": dict(observation.graph_snapshot),
-        "recent_action_history": list(observation.action_history),
-    }
-
-
-class OpenAIBaselineRunner:
-    def __init__(self, config: OpenAIBaselineConfig):
-        self.config = config
-
-        from openai import OpenAI
-
-        if not config.api_key:
-            raise ValueError(
-                "OpenAI baseline requires an API key. "
-                f"Set {config.api_key_env} or pass --openai-api-key."
-            )
-
-        self.client = OpenAI(
-            api_key=config.api_key,
-            base_url=config.base_url,
-            timeout=config.timeout_seconds,
-        )
-        self.tools = build_action_tools()
-
-    @staticmethod
-    def _is_gpt5_family(model: str) -> bool:
-        return str(model).strip().lower().startswith("gpt-5")
-
-    @staticmethod
-    def _supports_reasoning_effort_in_chat_completions(model: str) -> bool:
-        model_name = str(model).strip().lower()
-        if model_name.startswith("gpt-5.4-mini"):
-            return False
-        return model_name.startswith("gpt-5")
-
-    def _request_kwargs(self, messages: list[dict[str, Any]], episode_index: int) -> dict[str, Any]:
-        kwargs: dict[str, Any] = {
-            "model": self.config.model,
-            "messages": messages,
-            "tools": self.tools,
-            "tool_choice": "required",
-            "parallel_tool_calls": False,
-            "max_completion_tokens": self.config.max_tokens,
-        }
-        if self.config.seed is not None:
-            kwargs["seed"] = int(self.config.seed) + episode_index
-
-        if self._is_gpt5_family(self.config.model):
-            # GPT-5 family chat-completions compatibility:
-            # use max_completion_tokens and avoid temperature for older GPT-5 models.
-            if self._supports_reasoning_effort_in_chat_completions(self.config.model):
-                kwargs["reasoning_effort"] = "none"
-        else:
-            kwargs["temperature"] = self.config.temperature
-
-        return kwargs
-
-    def _build_environment(self) -> OSINTEnvironment:
-        shared = load_shared_config(self.config.shared_config_path)
-        env_cfg = clone_environment_config(shared.environment)
-        env_cfg.seeding = load_seeding_config(self.config.seed_file)
-        env_cfg.llm.provider = "mock"
-        env_cfg.llm.model = self.config.model
-        env_cfg.llm.temperature = self.config.temperature
-        env_cfg.llm.max_tokens = self.config.max_tokens
-        env_cfg.max_steps = min(int(env_cfg.max_steps), int(self.config.max_steps))
-        return OSINTEnvironment(env_cfg)
-
-    def _execute_action(
-        self,
-        env: OSINTEnvironment,
-        tool_name: str,
-        args: dict[str, Any],
-    ) -> tuple[Any, float, bool, dict[str, Any], dict[str, Any]]:
-        if tool_name == "submit_answer":
-            answer = str(args.get("answer", "")).strip()
-            obs, reward, done, info = env.step(Action(ActionType.ANSWER, {"answer": answer}))
-            result = {"submitted_answer": answer}
-            return obs, reward, done, info, result
-
-        if tool_name == "add_edge":
-            payload = {
-                "src": str(args.get("src", "")).strip(),
-                "rel": str(args.get("rel", "")).strip(),
-                "dst": str(args.get("dst", "")).strip(),
-                "confidence": float(args.get("confidence", 1.0)),
-            }
-            obs, reward, done, info = env.step(Action(ActionType.ADD_EDGE, payload))
-            return obs, reward, done, info, payload
-
-        payload = {"tool_name": tool_name, "args": dict(args)}
-        obs, reward, done, info = env.step(Action(ActionType.CALL_TOOL, payload))
-        result = obs.tool_outputs[-1]["output"] if obs.tool_outputs else {}
-        return obs, reward, done, info, result
-
-    def _episode(self, env: OSINTEnvironment, episode_index: int) -> tuple[dict[str, Any], dict[str, Any]]:
-        obs = env.reset()
-        initial_observation = _observation_payload(env, obs, env.config.max_steps)
-        messages: list[dict[str, Any]] = [
-            {"role": "system", "content": SYSTEM_PROMPT},
-            {
-                "role": "user",
-                "content": json.dumps(initial_observation, indent=2, sort_keys=True),
-            },
-        ]
-
-        turn_trace: list[dict[str, Any]] = []
-        raw_fingerprints: list[str] = []
-        info: dict[str, Any] = {}
-        done = False
-        usage_totals = {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0}
-
-        while not done and env.state is not None and env.state.step_count < env.config.max_steps:
-            completion = self.client.chat.completions.create(**self._request_kwargs(messages, episode_index))
-            if getattr(completion, "system_fingerprint", None):
-                raw_fingerprints.append(str(completion.system_fingerprint))
-            if getattr(completion, "usage", None) is not None:
-                usage_totals["prompt_tokens"] += int(getattr(completion.usage, "prompt_tokens", 0) or 0)
-                usage_totals["completion_tokens"] += int(getattr(completion.usage, "completion_tokens", 0) or 0)
-                usage_totals["total_tokens"] += int(getattr(completion.usage, "total_tokens", 0) or 0)
-
-            message = completion.choices[0].message
-            content = _message_text(message)
-            tool_calls = list(message.tool_calls or [])
-            if not tool_calls:
-                fallback_answer = content.strip() or "unknown"
-                obs, reward, done, info = env.step(Action(ActionType.ANSWER, {"answer": fallback_answer}))
-                tool_result = {
-                    "submitted_answer": fallback_answer,
-                    "reward": reward,
-                    "done": done,
-                    "observation": _observation_payload(env, obs, env.config.max_steps),
-                    "info": _safe_info(info),
-                }
-                messages.append({"role": "assistant", "content": content})
-                messages.append({"role": "tool", "tool_call_id": "fallback_submit", "content": json.dumps(tool_result)})
-                turn_trace.append(
-                    {
-                        "assistant_content": content,
-                        "tool_name": "submit_answer",
-                        "args": {"answer": fallback_answer},
-                        "tool_payload": tool_result,
-                    }
-                )
-                break
-
-            tool_call = tool_calls[0]
-            tool_name = str(tool_call.function.name)
-            try:
-                args = json.loads(tool_call.function.arguments or "{}")
-            except json.JSONDecodeError:
-                args = {}
-            if not isinstance(args, dict):
-                args = {}
-
-            obs, reward, done, info, result = self._execute_action(env, tool_name, args)
-            tool_payload = {
-                "tool_name": tool_name,
-                "args": args,
-                "result": result,
-                "reward": reward,
-                "done": done,
-                "observation": _observation_payload(env, obs, env.config.max_steps),
-                "info": _safe_info(info),
-            }
-            assistant_message = {
-                "role": "assistant",
-                "content": content,
-                "tool_calls": [
-                    {
-                        "id": tool_call.id,
-                        "type": "function",
-                        "function": {
-                            "name": tool_name,
-                            "arguments": json.dumps(args, sort_keys=True),
-                        },
-                    }
-                ],
-            }
-            messages.append(assistant_message)
-            messages.append({"role": "tool", "tool_call_id": tool_call.id, "content": json.dumps(tool_payload, sort_keys=True)})
-            turn_trace.append(
-                {
-                    "assistant_content": content,
-                    "tool_name": tool_name,
-                    "args": args,
-                    "reward": reward,
-                    "done": done,
-                    "tool_payload": tool_payload,
-                }
-            )
-
-        if not done:
-            obs, _, done, info = env.step(Action(ActionType.ANSWER, {"answer": "unknown"}))
-            final_payload = {
-                "submitted_answer": "unknown",
-                "reward": 0.0,
-                "done": done,
-                "observation": _observation_payload(env, obs, env.config.max_steps),
-                "info": _safe_info(info),
-            }
-            turn_trace.append(
-                {
-                    "assistant_content": "",
-                    "tool_name": "submit_answer",
-                    "args": {"answer": "unknown"},
-                    "reward": 0.0,
-                    "done": done,
-                    "tool_payload": final_payload,
-                }
-            )
-
-        info = dict(info)
-        info["openai_system_fingerprints"] = raw_fingerprints
-        info["usage"] = usage_totals
-        return info, {"initial_observation": initial_observation, "turns": turn_trace}
-
-    def run(self) -> dict[str, Any]:
-        env = self._build_environment()
-        metrics = EvalMetrics()
-        episode_rows: list[dict[str, Any]] = []
-
-        started = perf_counter()
-        run_usage = {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0}
-        for episode_index in range(int(self.config.episodes)):
-            info, trace = self._episode(env, episode_index)
-            episode_usage = dict(info.get("usage", {}))
-            for key in run_usage:
-                run_usage[key] += int(episode_usage.get(key, 0) or 0)
-            task_type = env.state.task.task_type if env.state else "unknown"
-            task_id = env.state.task.task_id if env.state else f"episode_{episode_index}"
-            truth = env.state.task.supporting_edges if env.state else []
-            pred = env.memory_graph.edges if env.state else []
-            graph_f1 = compute_graph_f1(pred, truth)
-            metrics.add(info, task_type=task_type, graph_f1=graph_f1)
-            episode_rows.append(
-                {
-                    "task_id": task_id,
-                    "task_type": task_type,
-                    "question": env.state.task.question if env.state else "",
-                    "task_answer": str(info.get("task_answer", "")),
-                    "agent_answer": str(info.get("agent_answer", "")) if info.get("agent_answer") is not None else "",
-                    "graph_f1": graph_f1,
-                    "reward": float(info.get("total_reward", 0.0)),
-                    "steps": int(info.get("step_count", 0)),
-                    "tool_calls": int(info.get("tool_calls", 0)),
-                    "success": int(info.get("agent_answer") == info.get("task_answer")),
-                    "reward_components": dict(info.get("reward_components", {})),
-                    "pred_edges": [
-                        {
-                            "src": edge.src,
-                            "rel": edge.rel,
-                            "dst": edge.dst,
-                            "confidence": float(edge.confidence),
-                        }
-                        for edge in pred
-                    ],
-                    "truth_edges": [
-                        {
-                            "src": edge.src,
-                            "rel": edge.rel,
-                            "dst": edge.dst,
-                            "confidence": float(edge.confidence),
-                        }
-                        for edge in truth
-                    ],
-                    "trace": trace,
-                    "openai_system_fingerprints": list(info.get("openai_system_fingerprints", [])),
-                    "usage": episode_usage,
-                }
-            )
-
-        summary = metrics.summary()
-        duration_seconds = perf_counter() - started
-        if self.config.append_leaderboard:
-            record = append_leaderboard_record(
-                path=self.config.leaderboard_path,
-                summary=summary,
-                episodes=int(self.config.episodes),
-                run_name=self.config.run_name,
-                config={
-                    "provider": "openai",
-                    "model": self.config.model,
-                    "seed": self.config.seed,
-                    "max_steps": self.config.max_steps,
-                    "shared_config_path": self.config.shared_config_path,
-                    "seed_file": self.config.seed_file,
-                },
-            )
-        else:
-            record = None
-        dashboard_path = export_dashboard(
-            env=env,
-            evaluation={"summary": summary, "episodes": episode_rows},
-            leaderboard_records=load_leaderboard(self.config.leaderboard_path),
-            output_path=self.config.dashboard_path,
-        )
-
-        payload: dict[str, Any] = {
-            "run": {
-                "name": self.config.run_name,
-                "model": self.config.model,
-                "episodes": int(self.config.episodes),
-                "temperature": float(self.config.temperature),
-                "max_tokens": int(self.config.max_tokens),
-                "timeout_seconds": int(self.config.timeout_seconds),
-                "max_steps": int(self.config.max_steps),
-                "seed": self.config.seed,
-                "shared_config_path": self.config.shared_config_path,
-                "seed_file": self.config.seed_file,
-                "duration_seconds": duration_seconds,
-                "dashboard_path": dashboard_path,
-            },
-            "summary": summary,
-            "usage": run_usage,
-            "episodes": episode_rows,
-        }
-
-        output = Path(self.config.output_path)
-        output.parent.mkdir(parents=True, exist_ok=True)
-        output.write_text(json.dumps(payload, indent=2, sort_keys=True), encoding="utf-8")
-
-        if record is not None:
-            payload["record"] = record
-            output.write_text(json.dumps(payload, indent=2, sort_keys=True), encoding="utf-8")
-
-        return payload

scratch/osint/src/osint_env/cli.py

@@ -1,339 +0,0 @@
-from __future__ import annotations
-
-import argparse
-import json
-from pathlib import Path
-
-from osint_env.agents.single_agent import SingleAgentRunner
-from osint_env.agents.swarm_agent import SwarmAgentRunner
-from osint_env.config import clone_environment_config, load_seeding_config, load_shared_config
-from osint_env.domain.models import EnvironmentConfig
-from osint_env.env.environment import OSINTEnvironment
-from osint_env.env.reward import compute_graph_f1
-from osint_env.eval.leaderboard import append_leaderboard_record, load_leaderboard, render_leaderboard_table
-from osint_env.eval.runner import run_evaluation
-from osint_env.llm import build_llm_client
-from osint_env.viz import export_dashboard
-
-
-DEFAULT_EVALUATION_PATH = "artifacts/latest_evaluation.json"
-
-
-def _save_evaluation(path: str, payload: dict) -> None:
-    out = Path(path)
-    out.parent.mkdir(parents=True, exist_ok=True)
-    out.write_text(json.dumps(payload, indent=2, sort_keys=True), encoding="utf-8")
-
-
-def _load_evaluation(path: str) -> dict | None:
-    file_path = Path(path)
-    if not file_path.exists():
-        return None
-    try:
-        data = json.loads(file_path.read_text(encoding="utf-8"))
-    except json.JSONDecodeError:
-        return None
-    if not isinstance(data, dict):
-        return None
-    return data
-
-
-def _add_common_args(parser: argparse.ArgumentParser) -> None:
-    parser.add_argument("--config", type=str, default="config/shared_config.json")
-    parser.add_argument("--seed-file", type=str, default="")
-    parser.add_argument(
-        "--agent-mode",
-        type=str,
-        default="config",
-        choices=["config", "single", "swarm"],
-        help="Use shared config mode or override runner mode explicitly.",
-    )
-    parser.add_argument(
-        "--llm-provider",
-        type=str,
-        default="config",
-        choices=["config", "mock", "ollama", "openai"],
-        help="Use shared config provider or override explicitly.",
-    )
-    parser.add_argument("--llm-model", type=str, default="", help="Override model name for selected LLM provider.")
-    parser.add_argument("--llm-timeout-seconds", type=int, default=0, help="Override LLM request timeout in seconds.")
-    parser.add_argument("--ollama-base-url", type=str, default="", help="Override Ollama base URL.")
-    parser.add_argument("--openai-base-url", type=str, default="", help="Override OpenAI base URL.")
-    parser.add_argument("--openai-api-key", type=str, default="", help="OpenAI API key override.")
-    parser.add_argument(
-        "--openai-api-key-env",
-        type=str,
-        default="",
-        help="Environment variable name for OpenAI API key.",
-    )
-
-
-def build_parser() -> argparse.ArgumentParser:
-    parser = argparse.ArgumentParser(prog="osint-env")
-    sub = parser.add_subparsers(dest="cmd", required=True)
-
-    d = sub.add_parser("demo", help="Run one episode and print debug info.")
-    _add_common_args(d)
-
-    e = sub.add_parser("eval", help="Run multiple episodes and show aggregate metrics.")
-    _add_common_args(e)
-    e.add_argument("--episodes", type=int, default=0)
-
-    b = sub.add_parser("benchmark", help="Run eval, update leaderboard, and export interactive dashboard.")
-    _add_common_args(b)
-    b.add_argument("--episodes", type=int, default=0)
-    b.add_argument("--name", type=str, default="")
-    b.add_argument("--leaderboard", type=str, default="")
-    b.add_argument("--dashboard", type=str, default="")
-
-    l = sub.add_parser("leaderboard", help="Print ranked benchmark leaderboard.")
-    _add_common_args(l)
-    l.add_argument("--leaderboard", type=str, default="")
-    l.add_argument("--top", type=int, default=20)
-    l.add_argument(
-        "--sort-by",
-        type=str,
-        default="leaderboard_score",
-        choices=[
-            "leaderboard_score",
-            "task_success_rate",
-            "avg_graph_f1",
-            "tool_efficiency",
-            "avg_reward",
-            "retrieval_signal",
-            "structural_signal",
-            "deanonymization_accuracy",
-            "spawn_signal",
-        ],
-    )
-
-    s = sub.add_parser("benchmark-sweep", help="Run benchmark across multiple seeds and append all runs to leaderboard.")
-    _add_common_args(s)
-    s.add_argument("--episodes", type=int, default=0)
-    s.add_argument("--seeds", type=str, default="7,11,17,23,31")
-    s.add_argument("--name-prefix", type=str, default="sweep")
-    s.add_argument("--leaderboard", type=str, default="")
-    s.add_argument("--dashboard-dir", type=str, default="")
-
-    v = sub.add_parser("viz", help="Export an interactive graph/database explorer.")
-    _add_common_args(v)
-    v.add_argument("--output", type=str, default="artifacts/osint_explorer.html")
-    v.add_argument("--with-demo", action="store_true")
-    v.add_argument("--leaderboard", type=str, default="")
-    v.add_argument(
-        "--evaluation",
-        type=str,
-        default=DEFAULT_EVALUATION_PATH,
-        help="Path to a saved evaluation payload with episode details.",
-    )
-    return parser
-
-
-def _resolve_environment_config(args: argparse.Namespace) -> tuple[EnvironmentConfig, dict[str, str | int]]:
-    shared = load_shared_config(args.config)
-    env_cfg = clone_environment_config(shared.environment)
-
-    if args.seed_file:
-        env_cfg.seeding = load_seeding_config(args.seed_file)
-
-    if args.llm_provider != "config":
-        env_cfg.llm.provider = args.llm_provider
-    if args.llm_model:
-        env_cfg.llm.model = args.llm_model
-    if int(args.llm_timeout_seconds) > 0:
-        env_cfg.llm.timeout_seconds = int(args.llm_timeout_seconds)
-    if args.ollama_base_url:
-        env_cfg.llm.ollama_base_url = args.ollama_base_url
-    if args.openai_base_url:
-        env_cfg.llm.openai_base_url = args.openai_base_url
-    if args.openai_api_key:
-        env_cfg.llm.openai_api_key = args.openai_api_key
-    if args.openai_api_key_env:
-        env_cfg.llm.openai_api_key_env = args.openai_api_key_env
-
-    if args.agent_mode == "single":
-        env_cfg.swarm.enabled = False
-    elif args.agent_mode == "swarm":
-        env_cfg.swarm.enabled = True
-
-    runtime = {
-        "default_episodes": shared.runtime.default_episodes,
-        "leaderboard_path": shared.runtime.leaderboard_path,
-        "dashboard_path": shared.runtime.dashboard_path,
-        "sweep_dashboard_dir": shared.runtime.sweep_dashboard_dir,
-    }
-    return env_cfg, runtime
-
-
-def _runner_for(env: OSINTEnvironment) -> SingleAgentRunner | SwarmAgentRunner:
-    if env.config.swarm.enabled:
-        return SwarmAgentRunner(env, llm=build_llm_client(env.config.llm))
-    return SingleAgentRunner(env, llm=build_llm_client(env.config.llm))
-
-
-def main() -> None:
-    args = build_parser().parse_args()
-    env_cfg, runtime = _resolve_environment_config(args)
-
-    episodes = int(args.episodes) if getattr(args, "episodes", 0) else int(runtime["default_episodes"])
-    leaderboard_path = str(args.leaderboard) if getattr(args, "leaderboard", "") else str(runtime["leaderboard_path"])
-    dashboard_path = str(args.dashboard) if getattr(args, "dashboard", "") else str(runtime["dashboard_path"])
-    sweep_dashboard_dir = (
-        str(args.dashboard_dir) if getattr(args, "dashboard_dir", "") else str(runtime["sweep_dashboard_dir"])
-    )
-    evaluation_path = str(getattr(args, "evaluation", "") or DEFAULT_EVALUATION_PATH)
-
-    if args.cmd == "leaderboard":
-        records = load_leaderboard(leaderboard_path)
-        print(render_leaderboard_table(records, top_k=args.top, sort_by=args.sort_by))
-        return
-
-    if args.cmd == "benchmark-sweep":
-        seed_values = [int(x.strip()) for x in args.seeds.split(",") if x.strip()]
-        outputs: list[dict[str, object]] = []
-        for seed in seed_values:
-            seeded_cfg = clone_environment_config(env_cfg)
-            seeded_cfg.seed = seed
-            env = OSINTEnvironment(seeded_cfg, llm=build_llm_client(seeded_cfg.llm))
-            evaluation = run_evaluation(env, episodes=episodes, return_details=True, llm=build_llm_client(seeded_cfg.llm))
-            summary = evaluation["summary"]
-            run_name = f"{args.name_prefix}_seed{seed}"
-            record = append_leaderboard_record(
-                path=leaderboard_path,
-                summary=summary,
-                episodes=episodes,
-                run_name=run_name,
-                config={
-                    "seed": seed,
-                    "max_steps": env.config.max_steps,
-                    "swarm_enabled": env.config.swarm.enabled,
-                    "max_agents": env.config.swarm.max_agents,
-                    "max_breadth": env.config.swarm.max_breadth,
-                    "max_width": env.config.swarm.max_width,
-                    "max_depth": env.config.swarm.max_depth,
-                    "seeded_questions": len(env.config.seeding.seeded_questions),
-                },
-            )
-            dashboard_path = export_dashboard(
-                env=env,
-                evaluation=evaluation,
-                leaderboard_records=load_leaderboard(leaderboard_path),
-                output_path=f"{sweep_dashboard_dir}/{run_name}.html",
-            )
-            _save_evaluation(DEFAULT_EVALUATION_PATH, evaluation)
-            outputs.append({"seed": seed, "record": record, "dashboard": dashboard_path, "summary": summary})
-
-        records = load_leaderboard(leaderboard_path)
-        print(
-            json.dumps(
-                {
-                    "runs": outputs,
-                    "leaderboard_preview": render_leaderboard_table(records, top_k=min(10, len(records))),
-                },
-                indent=2,
-                sort_keys=True,
-            )
-        )
-        return
-
-    llm_client = build_llm_client(env_cfg.llm)
-    env = OSINTEnvironment(env_cfg, llm=llm_client)
-    if args.cmd == "demo":
-        info = _runner_for(env).run_episode()
-        print(json.dumps(info, indent=2, sort_keys=True))
-    elif args.cmd == "eval":
-        metrics = run_evaluation(env, episodes=episodes, llm=llm_client)
-        print(json.dumps(metrics, indent=2, sort_keys=True))
-    elif args.cmd == "benchmark":
-        evaluation = run_evaluation(env, episodes=episodes, return_details=True, llm=llm_client)
-        summary = evaluation["summary"]
-        record = append_leaderboard_record(
-            path=leaderboard_path,
-            summary=summary,
-            episodes=episodes,
-            run_name=args.name or None,
-            config={
-                "seed": env.config.seed,
-                "max_steps": env.config.max_steps,
-                "swarm_enabled": env.config.swarm.enabled,
-                "max_agents": env.config.swarm.max_agents,
-                "max_breadth": env.config.swarm.max_breadth,
-                "max_width": env.config.swarm.max_width,
-                "max_depth": env.config.swarm.max_depth,
-                "seeded_questions": len(env.config.seeding.seeded_questions),
-            },
-        )
-        leaderboard = load_leaderboard(leaderboard_path)
-        dashboard_path = export_dashboard(
-            env=env,
-            evaluation=evaluation,
-            leaderboard_records=leaderboard,
-            output_path=dashboard_path,
-        )
-        _save_evaluation(DEFAULT_EVALUATION_PATH, evaluation)
-        payload = {
-            "record": record,
-            "summary": summary,
-            "dashboard": dashboard_path,
-        }
-        print(json.dumps(payload, indent=2, sort_keys=True))
-    elif args.cmd == "viz":
-        evaluation: dict | None = _load_evaluation(evaluation_path)
-        if args.with_demo:
-            _runner_for(env).run_episode()
-            info = {
-                "agent_answer": env.state.answer if env.state else "",
-                "task_answer": env.state.task.answer if env.state else "",
-                "total_reward": env.state.total_reward if env.state else 0.0,
-                "step_count": env.state.step_count if env.state else 0,
-                "tool_calls": env.state.tool_calls if env.state else 0,
-            }
-            evaluation = {
-                "summary": {
-                    "task_success_rate": float(info["agent_answer"] == info["task_answer"]),
-                    "tool_efficiency": 0.0,
-                    "avg_graph_f1": 0.0,
-                    "avg_steps_to_solution": float(info["step_count"]),
-                    "deanonymization_accuracy": 0.0,
-                    "avg_reward": float(info["total_reward"]),
-                    "leaderboard_score": 0.0,
-                },
-                "episodes": [
-                    {
-                        "task_id": env.state.task.task_id if env.state else "n/a",
-                        "task_type": env.state.task.task_type if env.state else "n/a",
-                        "question": env.state.task.question if env.state else "n/a",
-                        "task_answer": str(info["task_answer"]),
-                        "agent_answer": str(info["agent_answer"]),
-                        "graph_f1": 0.0,
-                        "reward": float(info["total_reward"]),
-                        "steps": int(info["step_count"]),
-                        "tool_calls": int(info["tool_calls"]),
-                        "success": int(info["agent_answer"] == info["task_answer"]),
-                    }
-                ],
-            }
-
-        graph_f1 = 0.0
-        if env.state is not None:
-            graph_f1 = compute_graph_f1(env.memory_graph.edges, env.state.task.supporting_edges)
-
-        if evaluation is None:
-            summary = {
-                "task_success_rate": 0.0,
-                "tool_efficiency": 0.0,
-                "avg_graph_f1": graph_f1,
-                "avg_steps_to_solution": float(env.state.step_count) if env.state else 0.0,
-                "deanonymization_accuracy": 0.0,
-                "avg_reward": float(env.state.total_reward) if env.state else 0.0,
-                "leaderboard_score": 0.0,
-            }
-            evaluation = {"summary": summary, "episodes": []}
-
-        leaderboard = load_leaderboard(leaderboard_path)
-        out = export_dashboard(env=env, evaluation=evaluation, leaderboard_records=leaderboard, output_path=args.output)
-        print(json.dumps({"dashboard": out, "evaluation": evaluation_path}, indent=2, sort_keys=True))
-
-
-if __name__ == "__main__":
-    main()

scratch/osint/src/osint_env/config/__init__.py

@@ -1,9 +0,0 @@
-from osint_env.config.shared import RuntimeDefaults, SharedConfig, clone_environment_config, load_seeding_config, load_shared_config
-
-__all__ = [
-    "RuntimeDefaults",
-    "SharedConfig",
-    "clone_environment_config",
-    "load_seeding_config",
-    "load_shared_config",
-]

scratch/osint/src/osint_env/config/shared.py

@@ -1,248 +0,0 @@
-from __future__ import annotations
-
-import copy
-import json
-from dataclasses import dataclass, field
-from pathlib import Path
-from typing import Any
-
-from osint_env.domain.models import (
-    EnvironmentConfig,
-    LLMConfig,
-    NodeType,
-    SeedingConfig,
-    SeedEdgeSpec,
-    SeedNodeSpec,
-    SeedQuestionSpec,
-    SpawnRewardConfig,
-    SwarmConfig,
-)
-
-
-@dataclass(slots=True)
-class RuntimeDefaults:
-    default_episodes: int = 20
-    leaderboard_path: str = "artifacts/leaderboard.json"
-    dashboard_path: str = "artifacts/osint_dashboard.html"
-    sweep_dashboard_dir: str = "artifacts/sweep_dashboards"
-
-
-@dataclass(slots=True)
-class SharedConfig:
-    environment: EnvironmentConfig = field(default_factory=EnvironmentConfig)
-    runtime: RuntimeDefaults = field(default_factory=RuntimeDefaults)
-
-
-def clone_environment_config(config: EnvironmentConfig) -> EnvironmentConfig:
-    return copy.deepcopy(config)
-
-
-def _as_dict(value: Any) -> dict[str, Any]:
-    return value if isinstance(value, dict) else {}
-
-
-def _parse_int(value: Any, default: int) -> int:
-    try:
-        return int(value)
-    except (TypeError, ValueError):
-        return default
-
-
-def _parse_float(value: Any, default: float) -> float:
-    try:
-        return float(value)
-    except (TypeError, ValueError):
-        return default
-
-
-def _parse_bool(value: Any, default: bool) -> bool:
-    if isinstance(value, bool):
-        return value
-    if isinstance(value, str):
-        lowered = value.strip().lower()
-        if lowered in {"1", "true", "yes", "y", "on"}:
-            return True
-        if lowered in {"0", "false", "no", "n", "off"}:
-            return False
-    return default
-
-
-def _infer_node_type(node_id: str) -> NodeType:
-    prefix = str(node_id).split("_", 1)[0].lower()
-    mapping = {
-        "user": NodeType.USER,
-        "alias": NodeType.ALIAS,
-        "org": NodeType.ORG,
-        "loc": NodeType.LOCATION,
-        "location": NodeType.LOCATION,
-        "post": NodeType.POST,
-        "thr": NodeType.THREAD,
-        "thread": NodeType.THREAD,
-        "event": NodeType.EVENT,
-    }
-    return mapping.get(prefix, NodeType.USER)
-
-
-def _parse_node_type(value: Any, node_id: str) -> NodeType:
-    if isinstance(value, NodeType):
-        return value
-    if isinstance(value, str):
-        raw = value.strip().lower()
-        try:
-            return NodeType(raw)
-        except ValueError:
-            return _infer_node_type(node_id)
-    return _infer_node_type(node_id)
-
-
-def _parse_seed_edge(item: dict[str, Any]) -> SeedEdgeSpec | None:
-    src = str(item.get("src", "")).strip()
-    rel = str(item.get("rel", "")).strip()
-    dst = str(item.get("dst", "")).strip()
-    if not src or not rel or not dst:
-        return None
-    confidence = _parse_float(item.get("confidence", 1.0), 1.0)
-    return SeedEdgeSpec(src=src, rel=rel, dst=dst, confidence=confidence)
-
-
-def _parse_seeding(data: dict[str, Any]) -> SeedingConfig:
-    seeded_nodes: list[SeedNodeSpec] = []
-    for item in data.get("seeded_nodes", []):
-        row = _as_dict(item)
-        node_id = str(row.get("node_id", "")).strip()
-        if not node_id:
-            continue
-        node_type = _parse_node_type(row.get("node_type"), node_id)
-        attrs = _as_dict(row.get("attrs"))
-        seeded_nodes.append(SeedNodeSpec(node_id=node_id, node_type=node_type, attrs=attrs))
-
-    seeded_edges: list[SeedEdgeSpec] = []
-    for item in data.get("seeded_edges", []):
-        edge = _parse_seed_edge(_as_dict(item))
-        if edge is not None:
-            seeded_edges.append(edge)
-
-    seeded_questions: list[SeedQuestionSpec] = []
-    for item in data.get("seeded_questions", []):
-        row = _as_dict(item)
-        question = str(row.get("question", "")).strip()
-        if not question:
-            continue
-        answer_val = row.get("answer")
-        answer = str(answer_val).strip() if answer_val is not None and str(answer_val).strip() else None
-        task_type = str(row.get("task_type", "seeded")).strip() or "seeded"
-        support_edges: list[SeedEdgeSpec] = []
-        for edge_item in row.get("supporting_edges", []):
-            edge = _parse_seed_edge(_as_dict(edge_item))
-            if edge is not None:
-                support_edges.append(edge)
-        metadata = _as_dict(row.get("metadata"))
-        seeded_questions.append(
-            SeedQuestionSpec(
-                question=question,
-                answer=answer,
-                task_type=task_type,
-                supporting_edges=support_edges,
-                metadata=metadata,
-            )
-        )
-
-    return SeedingConfig(
-        seeded_nodes=seeded_nodes,
-        seeded_edges=seeded_edges,
-        seeded_questions=seeded_questions,
-        llm_generate_remaining_graph=_parse_bool(data.get("llm_generate_remaining_graph"), True),
-        llm_generate_remaining_tasks=_parse_bool(data.get("llm_generate_remaining_tasks"), True),
-        llm_generated_edge_budget=max(0, _parse_int(data.get("llm_generated_edge_budget"), 6)),
-        llm_generated_task_budget=max(0, _parse_int(data.get("llm_generated_task_budget"), 8)),
-        llm_generation_parallel=_parse_bool(data.get("llm_generation_parallel"), True),
-        llm_generation_workers=max(1, _parse_int(data.get("llm_generation_workers"), 3)),
-        llm_generation_retries=max(1, _parse_int(data.get("llm_generation_retries"), 2)),
-        allow_template_fallback_on_llm_failure=_parse_bool(
-            data.get("allow_template_fallback_on_llm_failure"),
-            False,
-        ),
-    )
-
-
-def load_seeding_config(path: str | Path) -> SeedingConfig:
-    payload = json.loads(Path(path).read_text(encoding="utf-8"))
-    if not isinstance(payload, dict):
-        raise ValueError("Seed file must contain a JSON object.")
-    source = _as_dict(payload.get("seeding", payload))
-    return _parse_seeding(source)
-
-
-def _parse_environment(payload: dict[str, Any]) -> EnvironmentConfig:
-    env_data = _as_dict(payload.get("environment", payload))
-    swarm_data = _as_dict(payload.get("swarm", env_data.get("swarm", {})))
-    spawn_data = _as_dict(payload.get("spawn_reward", env_data.get("spawn_reward", {})))
-    seeding_data = _as_dict(payload.get("seeding", env_data.get("seeding", {})))
-    llm_data = _as_dict(payload.get("llm", env_data.get("llm", {})))
-
-    env = EnvironmentConfig(
-        n_users=max(4, _parse_int(env_data.get("n_users"), 40)),
-        alias_density=max(0.0, min(1.0, _parse_float(env_data.get("alias_density"), 0.35))),
-        noise_level=max(0.0, min(1.0, _parse_float(env_data.get("noise_level"), 0.15))),
-        red_herring_rate=max(0.0, min(1.0, _parse_float(env_data.get("red_herring_rate"), 0.1))),
-        max_steps=max(2, _parse_int(env_data.get("max_steps"), 18)),
-        seed=_parse_int(env_data.get("seed"), 7),
-    )
-
-    env.swarm = SwarmConfig(
-        enabled=_parse_bool(swarm_data.get("enabled"), False),
-        max_agents=max(1, _parse_int(swarm_data.get("max_agents"), 3)),
-        max_breadth=max(1, _parse_int(swarm_data.get("max_breadth"), 2)),
-        max_width=max(1, _parse_int(swarm_data.get("max_width"), 2)),
-        max_depth=max(1, _parse_int(swarm_data.get("max_depth"), 2)),
-        planner_rounds=max(1, _parse_int(swarm_data.get("planner_rounds"), 2)),
-        tools_per_agent=max(1, _parse_int(swarm_data.get("tools_per_agent"), 1)),
-    )
-
-    env.spawn_reward = SpawnRewardConfig(
-        lambda_parallel=max(0.0, _parse_float(spawn_data.get("lambda_parallel"), 0.15)),
-        lambda_finish=max(0.0, _parse_float(spawn_data.get("lambda_finish"), 0.2)),
-        anneal=max(0.0, min(1.0, _parse_float(spawn_data.get("anneal"), 1.0))),
-        max_parallel_hint=max(1, _parse_int(spawn_data.get("max_parallel_hint"), 3)),
-    )
-
-    env.seeding = _parse_seeding(seeding_data)
-    env.llm = LLMConfig(
-        provider=str(llm_data.get("provider", "mock")).strip() or "mock",
-        model=str(llm_data.get("model", "qwen3:2b")).strip() or "qwen3:2b",
-        temperature=_parse_float(llm_data.get("temperature"), 0.1),
-        max_tokens=max(1, _parse_int(llm_data.get("max_tokens"), 256)),
-        timeout_seconds=max(1, _parse_int(llm_data.get("timeout_seconds"), 240)),
-        ollama_base_url=str(llm_data.get("ollama_base_url", "http://127.0.0.1:11434")).strip()
-        or "http://127.0.0.1:11434",
-        openai_base_url=str(llm_data.get("openai_base_url", "https://api.openai.com/v1")).strip()
-        or "https://api.openai.com/v1",
-        openai_api_key_env=str(llm_data.get("openai_api_key_env", "OPENAI_API_KEY")).strip() or "OPENAI_API_KEY",
-        openai_api_key=str(llm_data.get("openai_api_key", "")).strip(),
-    )
-    return env
-
-
-def _parse_runtime(payload: dict[str, Any]) -> RuntimeDefaults:
-    runtime = _as_dict(payload.get("runtime", {}))
-    return RuntimeDefaults(
-        default_episodes=max(1, _parse_int(runtime.get("default_episodes"), 20)),
-        leaderboard_path=str(runtime.get("leaderboard_path", "artifacts/leaderboard.json")),
-        dashboard_path=str(runtime.get("dashboard_path", "artifacts/osint_dashboard.html")),
-        sweep_dashboard_dir=str(runtime.get("sweep_dashboard_dir", "artifacts/sweep_dashboards")),
-    )
-
-
-def load_shared_config(path: str | Path | None) -> SharedConfig:
-    if not path:
-        return SharedConfig()
-
-    file_path = Path(path)
-    if not file_path.exists():
-        return SharedConfig()
-
-    payload = json.loads(file_path.read_text(encoding="utf-8"))
-    if not isinstance(payload, dict):
-        raise ValueError("Shared config file must contain a JSON object.")
-
-    return SharedConfig(environment=_parse_environment(payload), runtime=_parse_runtime(payload))

scratch/osint/src/osint_env/data/__init__.py

@@ -1,2 +0,0 @@
-"""Dataset generation package."""
-

scratch/osint/src/osint_env/data/generator.py

@@ -1,819 +0,0 @@
-from __future__ import annotations
-
-from concurrent.futures import ThreadPoolExecutor, as_completed
-import json
-import random
-import re
-from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any
-
-from osint_env.domain.models import (
-    CanonicalGraph,
-    Edge,
-    EnvironmentConfig,
-    Node,
-    NodeType,
-    SeedEdgeSpec,
-    SeedQuestionSpec,
-    TaskInstance,
-)
-
-if TYPE_CHECKING:
-    from osint_env.llm.interface import LLMClient
-
-
-@dataclass(slots=True)
-class PlatformViews:
-    microblog_posts: list[dict]
-    forum_threads: list[dict]
-    profiles: list[dict]
-    alias_lookup: dict[str, str]
-
-
-class DatasetGenerator:
-    def __init__(self, config: EnvironmentConfig, llm: LLMClient | None = None):
-        self.config = config
-        self.rng = random.Random(config.seed)
-        self.llm = llm
-
-    @staticmethod
-    def _edge_key(edge: Edge) -> tuple[str, str, str]:
-        return (edge.src, edge.rel, edge.dst)
-
-    @staticmethod
-    def _infer_node_type(node_id: str) -> NodeType:
-        prefix = str(node_id).split("_", 1)[0].lower()
-        mapping = {
-            "user": NodeType.USER,
-            "alias": NodeType.ALIAS,
-            "org": NodeType.ORG,
-            "loc": NodeType.LOCATION,
-            "location": NodeType.LOCATION,
-            "post": NodeType.POST,
-            "thr": NodeType.THREAD,
-            "thread": NodeType.THREAD,
-            "event": NodeType.EVENT,
-        }
-        return mapping.get(prefix, NodeType.USER)
-
-    def _ensure_node(self, graph: CanonicalGraph, node_id: str) -> None:
-        if node_id in graph.nodes:
-            return
-        node_type = self._infer_node_type(node_id)
-        attrs: dict[str, Any] = {}
-        if node_type == NodeType.USER:
-            attrs = {"name": node_id, "org": "Unknown", "location": "Unknown"}
-        if node_type == NodeType.ALIAS:
-            attrs = {"handle": f"@{node_id}"}
-        graph.nodes[node_id] = Node(node_id=node_id, node_type=node_type, attrs=attrs)
-
-    def _add_edge_if_missing(self, graph: CanonicalGraph, edge: Edge) -> None:
-        key = self._edge_key(edge)
-        if any(self._edge_key(existing) == key for existing in graph.edges):
-            return
-        self._ensure_node(graph, edge.src)
-        self._ensure_node(graph, edge.dst)
-        graph.edges.append(edge)
-
-    @staticmethod
-    def _extract_json_blob(text: str) -> Any:
-        text = str(text).strip()
-        if not text:
-            return None
-        for start, end in (("{", "}"), ("[", "]")):
-            left = text.find(start)
-            right = text.rfind(end)
-            if left >= 0 and right > left:
-                snippet = text[left : right + 1]
-                try:
-                    return json.loads(snippet)
-                except json.JSONDecodeError:
-                    continue
-        return None
-
-    def _apply_seed_nodes(self, graph: CanonicalGraph) -> None:
-        for node_spec in self.config.seeding.seeded_nodes:
-            node_type = (
-                node_spec.node_type
-                if isinstance(node_spec.node_type, NodeType)
-                else self._infer_node_type(node_spec.node_id)
-            )
-            existing = graph.nodes.get(node_spec.node_id)
-            attrs = dict(existing.attrs) if existing else {}
-            attrs.update(node_spec.attrs)
-            graph.nodes[node_spec.node_id] = Node(node_spec.node_id, node_type, attrs)
-
-    def _apply_seed_edges(self, graph: CanonicalGraph) -> None:
-        for edge_spec in self.config.seeding.seeded_edges:
-            self._add_edge_if_missing(
-                graph,
-                Edge(
-                    src=edge_spec.src,
-                    rel=edge_spec.rel,
-                    dst=edge_spec.dst,
-                    confidence=float(edge_spec.confidence),
-                ),
-            )
-
-    @staticmethod
-    def _normalize_edge_candidates(value: Any) -> list[SeedEdgeSpec]:
-        items: list[SeedEdgeSpec] = []
-        if not isinstance(value, list):
-            return items
-        for row in value:
-            if not isinstance(row, dict):
-                continue
-            src = str(row.get("src", "")).strip()
-            rel = str(row.get("rel", "")).strip()
-            dst = str(row.get("dst", "")).strip()
-            if not src or not rel or not dst:
-                continue
-            try:
-                confidence = float(row.get("confidence", 1.0))
-            except (TypeError, ValueError):
-                confidence = 1.0
-            items.append(SeedEdgeSpec(src=src, rel=rel, dst=dst, confidence=confidence))
-        return items
-
-    @staticmethod
-    def _split_budget(total: int, parts: int) -> list[int]:
-        if total <= 0:
-            return []
-        slots = max(1, parts)
-        base = total // slots
-        remainder = total % slots
-        chunks = [base + (1 if i < remainder else 0) for i in range(slots)]
-        return [chunk for chunk in chunks if chunk > 0]
-
-    @staticmethod
-    def _shared_context_blob(graph: CanonicalGraph, node_limit: int = 100, edge_limit: int = 80) -> str:
-        payload = {
-            "known_nodes": sorted(graph.nodes.keys())[:node_limit],
-            "known_edges": [
-                {"src": edge.src, "rel": edge.rel, "dst": edge.dst}
-                for edge in graph.edges[: min(edge_limit, len(graph.edges))]
-            ],
-        }
-        return json.dumps(payload)
-
-    def _llm_generate_json_with_retry(self, prompt: str) -> Any:
-        if self.llm is None:
-            return None
-
-        attempts = max(1, int(self.config.seeding.llm_generation_retries))
-        for _ in range(attempts):
-            try:
-                response = self.llm.generate([{"role": "system", "content": prompt}], tools=[])
-            except Exception:
-                continue
-            parsed = self._extract_json_blob(response.content)
-            if parsed is not None:
-                return parsed
-        return None
-
-    def _run_generation_workers(self, prompts: list[str]) -> list[Any]:
-        if not prompts:
-            return []
-
-        max_workers = max(1, min(self.config.seeding.llm_generation_workers, len(prompts)))
-        if not self.config.seeding.llm_generation_parallel or max_workers == 1:
-            output: list[Any] = []
-            for prompt in prompts:
-                parsed = self._llm_generate_json_with_retry(prompt)
-                if parsed is not None:
-                    output.append(parsed)
-            return output
-
-        output = []
-        with ThreadPoolExecutor(max_workers=max_workers) as executor:
-            futures = [executor.submit(self._llm_generate_json_with_retry, prompt) for prompt in prompts]
-            for future in as_completed(futures):
-                try:
-                    parsed = future.result()
-                except Exception:
-                    parsed = None
-                if parsed is not None:
-                    output.append(parsed)
-        return output
-
-    def _template_fallback_allowed(self) -> bool:
-        if self.llm is None:
-            return True
-        return bool(self.config.seeding.allow_template_fallback_on_llm_failure)
-
-    def _template_generated_edges(self, graph: CanonicalGraph, budget: int) -> list[Edge]:
-        if budget <= 0:
-            return []
-        users = [n.node_id for n in graph.nodes.values() if n.node_type == NodeType.USER]
-        aliases = [n.node_id for n in graph.nodes.values() if n.node_type == NodeType.ALIAS]
-        if len(users) < 2:
-            return []
-
-        generated: list[Edge] = []
-        rels = ["connected_to", "mentions", "co_occurs_with"]
-        for _ in range(budget * 3):
-            if len(generated) >= budget:
-                break
-            roll = self.rng.random()
-            if aliases and roll < 0.2:
-                src = self.rng.choice(aliases)
-                dst = self.rng.choice(users)
-                rel = "alias_of"
-            elif roll < 0.75:
-                src, dst = self.rng.sample(users, 2)
-                rel = self.rng.choice(rels)
-            else:
-                src = self.rng.choice(users)
-                dst = self.rng.choice([u for u in users if u != src])
-                rel = "connected_to"
-            generated.append(Edge(src=src, rel=rel, dst=dst, confidence=0.7))
-        return generated[:budget]
-
-    def _llm_expand_graph(self, graph: CanonicalGraph, budget: int) -> list[Edge]:
-        if budget <= 0:
-            return []
-
-        if self.llm is None:
-            return self._template_generated_edges(graph, budget)
-
-        shared_context = self._shared_context_blob(graph)
-        workers = max(1, min(self.config.seeding.llm_generation_workers, budget))
-        chunks = self._split_budget(budget, workers)
-        focus_tracks = ["entity_linking", "network_expansion", "org_location", "event_trace"]
-
-        prompts: list[str] = []
-        for idx, chunk_budget in enumerate(chunks):
-            focus = focus_tracks[idx % len(focus_tracks)]
-            prompts.append(
-                (
-                    "SEED_GRAPH_EXPANSION_AGENT\n"
-                    "SHARED_CONTEXT\n"
-                    f"{shared_context}\n"
-                    f"worker_id: {idx}\n"
-                    f"focus: {focus}\n"
-                    f"budget: {chunk_budget}\n"
-                    "Generate plausible graph edges for OSINT retrieval.\n"
-                    "Return STRICT JSON object: {\"edges\": [{\"src\": str, \"rel\": str, \"dst\": str, \"confidence\": float}]}.\n"
-                    "Prefer known nodes from SHARED_CONTEXT and avoid duplicates."
-                )
-            )
-
-        generated: list[Edge] = []
-        seen: set[tuple[str, str, str]] = set()
-        for payload in self._run_generation_workers(prompts):
-            raw_edges: Any = None
-            if isinstance(payload, dict):
-                raw_edges = payload.get("edges")
-            elif isinstance(payload, list):
-                raw_edges = payload
-            for edge_spec in self._normalize_edge_candidates(raw_edges):
-                key = (edge_spec.src, edge_spec.rel, edge_spec.dst)
-                if key in seen:
-                    continue
-                seen.add(key)
-                generated.append(Edge(edge_spec.src, edge_spec.rel, edge_spec.dst, float(edge_spec.confidence)))
-                if len(generated) >= budget:
-                    break
-            if len(generated) >= budget:
-                break
-
-        if len(generated) < budget:
-            residual = budget - len(generated)
-            residual_prompt = (
-                "SEED_GRAPH_EXPANSION_AGENT\n"
-                "SHARED_CONTEXT\n"
-                f"{shared_context}\n"
-                f"budget: {residual}\n"
-                "Generate any remaining high-utility edges.\n"
-                "Return STRICT JSON object: {\"edges\": [{\"src\": str, \"rel\": str, \"dst\": str, \"confidence\": float}]}."
-            )
-            payload = self._llm_generate_json_with_retry(residual_prompt)
-            raw_edges: Any = payload.get("edges") if isinstance(payload, dict) else payload
-            for edge_spec in self._normalize_edge_candidates(raw_edges):
-                key = (edge_spec.src, edge_spec.rel, edge_spec.dst)
-                if key in seen:
-                    continue
-                seen.add(key)
-                generated.append(Edge(edge_spec.src, edge_spec.rel, edge_spec.dst, float(edge_spec.confidence)))
-                if len(generated) >= budget:
-                    break
-
-        if len(generated) < budget and self._template_fallback_allowed():
-            for edge in self._template_generated_edges(graph, budget - len(generated)):
-                key = (edge.src, edge.rel, edge.dst)
-                if key in seen:
-                    continue
-                seen.add(key)
-                generated.append(edge)
-                if len(generated) >= budget:
-                    break
-
-        return generated[:budget]
-
-    @staticmethod
-    def _extract_entity_tokens(question: str) -> list[str]:
-        return re.findall(r"\b(?:alias|user|org|loc|post|thr|thread|event)_[a-zA-Z0-9_]+\b", question)
-
-    @staticmethod
-    def _normalize_difficulty(value: str, index: int) -> str:
-        token = str(value or "").strip().lower()
-        if token in {"easy", "e"}:
-            return "easy"
-        if token in {"mid", "medium", "m"}:
-            return "medium"
-        if token in {"high", "hard", "h"}:
-            return "hard"
-        if index < 10:
-            return "easy"
-        if index < 20:
-            return "medium"
-        return "hard"
-
-    @staticmethod
-    def _task_type_for_difficulty(base_task_type: str, difficulty: str) -> str:
-        token = str(base_task_type or "").strip().lower()
-        if token and token != "fixed_trace":
-            return token
-        if difficulty == "easy":
-            return "easy_trace"
-        if difficulty == "medium":
-            return "medium_trace"
-        return "hard_trace"
-
-    @staticmethod
-    def _grader_for_difficulty(difficulty: str) -> dict[str, Any]:
-        return {
-            "type": "difficulty_exact_match",
-            "answer_type": "node_id",
-            "case_sensitive": True,
-            "reward_profile": difficulty,
-            "logic": {
-                "easy": "single_agent_simplified",
-                "medium": "reduced_components",
-                "hard": "full_reward",
-            }.get(difficulty, "full_reward"),
-        }
-
-    def _task_metadata(self, index: int, base_task_type: str, metadata: dict[str, Any] | None = None) -> dict[str, Any]:
-        out = dict(metadata or {})
-        difficulty = self._normalize_difficulty(out.get("difficulty", ""), index)
-        out["difficulty"] = difficulty
-        out.setdefault("grader", self._grader_for_difficulty(difficulty))
-        out.setdefault("scenario", self._task_type_for_difficulty(base_task_type, difficulty))
-        return out
-
-    def _infer_answer_from_question(self, question: str, graph: CanonicalGraph) -> str:
-        entities = self._extract_entity_tokens(question)
-        question_l = question.lower()
-
-        alias_tokens = [token for token in entities if token.startswith("alias_")]
-        if alias_tokens:
-            alias = alias_tokens[0]
-            for edge in graph.edges:
-                if edge.rel == "alias_of" and edge.src == alias:
-                    return edge.dst
-
-        if "connected" in question_l:
-            user_tokens = [token for token in entities if token.startswith("user_")]
-            if user_tokens:
-                source = user_tokens[0]
-                for edge in graph.edges:
-                    if edge.rel == "connected_to" and edge.src == source:
-                        return edge.dst
-
-        if "works at" in question_l:
-            for edge in graph.edges:
-                if edge.rel != "works_at":
-                    continue
-                org = graph.nodes.get(edge.dst)
-                org_name = str((org.attrs or {}).get("name", "")).lower() if org else ""
-                if org_name and org_name in question_l:
-                    return edge.src
-
-        return entities[0] if entities else "unknown"
-
-    def _infer_support_edges(self, question: str, answer: str, graph: CanonicalGraph) -> list[Edge]:
-        if answer:
-            for edge in graph.edges:
-                if edge.dst == answer or edge.src == answer:
-                    if edge.src in question or edge.dst in question or edge.rel in question.lower():
-                        return [edge]
-
-        entities = self._extract_entity_tokens(question)
-        for edge in graph.edges:
-            if edge.src in entities or edge.dst in entities:
-                return [edge]
-        return []
-
-    def _seeded_tasks(self, graph: CanonicalGraph) -> list[TaskInstance]:
-        tasks: list[TaskInstance] = []
-        for idx, question_spec in enumerate(self.config.seeding.seeded_questions):
-            answer = question_spec.answer or self._infer_answer_from_question(question_spec.question, graph)
-            metadata = self._task_metadata(idx, question_spec.task_type, dict(question_spec.metadata))
-            difficulty = str(metadata.get("difficulty", "hard"))
-            if question_spec.supporting_edges:
-                support = [
-                    Edge(src=e.src, rel=e.rel, dst=e.dst, confidence=float(e.confidence))
-                    for e in question_spec.supporting_edges
-                ]
-            else:
-                support = self._infer_support_edges(question_spec.question, answer, graph)
-
-            tasks.append(
-                TaskInstance(
-                    task_id=f"seed_task_{idx}",
-                    task_type=self._task_type_for_difficulty(question_spec.task_type, difficulty),
-                    question=question_spec.question,
-                    answer=answer,
-                    supporting_edges=support,
-                    metadata=metadata,
-                )
-            )
-        return tasks
-
-    def _template_tasks(self, graph: CanonicalGraph, count: int, start_idx: int = 0) -> list[TaskInstance]:
-        alias_edges = [e for e in graph.edges if e.rel == "alias_of"]
-        conn_edges = [e for e in graph.edges if e.rel == "connected_to"]
-        work_edges = [e for e in graph.edges if e.rel == "works_at"]
-        tasks: list[TaskInstance] = []
-
-        for i in range(count):
-            mode = self.rng.choice(["identity_resolution", "network_discovery", "event_tracing"])
-            if mode == "identity_resolution" and alias_edges:
-                edge = self.rng.choice(alias_edges)
-                q = f"Which canonical user owns alias {edge.src}?"
-                a = edge.dst
-                support = [edge]
-            elif mode == "network_discovery" and conn_edges:
-                edge = self.rng.choice(conn_edges)
-                q = f"Who is connected to {edge.src}?"
-                a = edge.dst
-                support = [edge]
-            else:
-                edge = self.rng.choice(work_edges)
-                org_node = graph.nodes.get(edge.dst)
-                org_name = (org_node.attrs or {}).get("name", edge.dst) if org_node else edge.dst
-                q = f"Which user works at {org_name}?"
-                a = edge.src
-                support = [edge]
-            tasks.append(
-                TaskInstance(
-                    task_id=f"task_{start_idx + i}",
-                    task_type=mode,
-                    question=q,
-                    answer=a,
-                    supporting_edges=support,
-                    metadata=self._task_metadata(start_idx + i, mode),
-                )
-            )
-        return tasks
-
-    def _llm_generated_tasks(self, graph: CanonicalGraph, count: int, start_idx: int) -> list[TaskInstance]:
-        if count <= 0:
-            return []
-        if self.llm is None:
-            return self._template_tasks(graph, count=count, start_idx=start_idx)
-
-        candidate_edges = [
-            {"src": edge.src, "rel": edge.rel, "dst": edge.dst}
-            for edge in graph.edges
-            if edge.rel in {"alias_of", "connected_to", "works_at"}
-        ][:60]
-        shared_context = json.dumps(
-            {
-                "known_nodes": sorted(graph.nodes.keys())[:100],
-                "edge_sample": candidate_edges,
-            }
-        )
-        workers = max(1, min(self.config.seeding.llm_generation_workers, count))
-        chunks = self._split_budget(count, workers)
-        focus_tracks = ["identity_resolution", "network_discovery", "event_tracing", "deanonymization"]
-
-        prompts: list[str] = []
-        for idx, chunk_budget in enumerate(chunks):
-            focus = focus_tracks[idx % len(focus_tracks)]
-            prompts.append(
-                (
-                    "SEED_TASK_EXPANSION_AGENT\n"
-                    "SHARED_CONTEXT\n"
-                    f"{shared_context}\n"
-                    f"worker_id: {idx}\n"
-                    f"focus: {focus}\n"
-                    f"task_budget: {chunk_budget}\n"
-                    "Generate OSINT QA tasks with answers and support edges.\n"
-                    "Return STRICT JSON object: {\"tasks\": [{\"task_type\": str, \"question\": str, \"answer\": str, \"supporting_edges\": [{\"src\": str, \"rel\": str, \"dst\": str, \"confidence\": float}]}]}."
-                )
-            )
-
-        llm_tasks: list[TaskInstance] = []
-        seen_questions: set[str] = set()
-        for payload in self._run_generation_workers(prompts):
-            raw_tasks: Any = None
-            if isinstance(payload, dict):
-                raw_tasks = payload.get("tasks")
-            elif isinstance(payload, list):
-                raw_tasks = payload
-            if not isinstance(raw_tasks, list):
-                continue
-
-            for row in raw_tasks:
-                if not isinstance(row, dict):
-                    continue
-                question = str(row.get("question", "")).strip()
-                if not question:
-                    continue
-                key = question.lower()
-                if key in seen_questions:
-                    continue
-                seen_questions.add(key)
-                answer = str(row.get("answer", "")).strip() or self._infer_answer_from_question(question, graph)
-                task_type = str(row.get("task_type", "llm_generated")).strip() or "llm_generated"
-                support_specs = self._normalize_edge_candidates(row.get("supporting_edges"))
-                if support_specs:
-                    support = [Edge(e.src, e.rel, e.dst, e.confidence) for e in support_specs]
-                else:
-                    support = self._infer_support_edges(question, answer, graph)
-                llm_tasks.append(
-                    TaskInstance(
-                        task_id=f"task_{start_idx + len(llm_tasks)}",
-                        task_type=task_type,
-                        question=question,
-                        answer=answer,
-                        supporting_edges=support,
-                        metadata=self._task_metadata(
-                            start_idx + len(llm_tasks),
-                            task_type,
-                            {"generated_by": "llm", "shared_context": True},
-                        ),
-                    )
-                )
-                if len(llm_tasks) >= count:
-                    break
-            if len(llm_tasks) >= count:
-                break
-
-        if len(llm_tasks) < count:
-            residual = count - len(llm_tasks)
-            residual_prompt = (
-                "SEED_TASK_EXPANSION_AGENT\n"
-                "SHARED_CONTEXT\n"
-                f"{shared_context}\n"
-                f"task_budget: {residual}\n"
-                "Generate additional tasks not already present in SHARED_CONTEXT.\n"
-                "Return STRICT JSON object: {\"tasks\": [{\"task_type\": str, \"question\": str, \"answer\": str, \"supporting_edges\": [{\"src\": str, \"rel\": str, \"dst\": str, \"confidence\": float}]}]}."
-            )
-            payload = self._llm_generate_json_with_retry(residual_prompt)
-            raw_tasks: Any = payload.get("tasks") if isinstance(payload, dict) else payload
-            if isinstance(raw_tasks, list):
-                for row in raw_tasks:
-                    if not isinstance(row, dict):
-                        continue
-                    question = str(row.get("question", "")).strip()
-                    if not question:
-                        continue
-                    key = question.lower()
-                    if key in seen_questions:
-                        continue
-                    seen_questions.add(key)
-                    answer = str(row.get("answer", "")).strip() or self._infer_answer_from_question(question, graph)
-                    task_type = str(row.get("task_type", "llm_generated")).strip() or "llm_generated"
-                    support_specs = self._normalize_edge_candidates(row.get("supporting_edges"))
-                    if support_specs:
-                        support = [Edge(e.src, e.rel, e.dst, e.confidence) for e in support_specs]
-                    else:
-                        support = self._infer_support_edges(question, answer, graph)
-                    llm_tasks.append(
-                        TaskInstance(
-                            task_id=f"task_{start_idx + len(llm_tasks)}",
-                            task_type=task_type,
-                            question=question,
-                            answer=answer,
-                            supporting_edges=support,
-                            metadata=self._task_metadata(
-                                start_idx + len(llm_tasks),
-                                task_type,
-                                {"generated_by": "llm", "shared_context": True},
-                            ),
-                        )
-                    )
-                    if len(llm_tasks) >= count:
-                        break
-
-        if len(llm_tasks) < count and self._template_fallback_allowed():
-            llm_tasks.extend(
-                self._template_tasks(
-                    graph,
-                    count=count - len(llm_tasks),
-                    start_idx=start_idx + len(llm_tasks),
-                )
-            )
-        return llm_tasks[:count]
-
-    def build_canonical_graph(self) -> CanonicalGraph:
-        graph = CanonicalGraph()
-        orgs = ["Apex Dynamics", "Helios Labs", "Northbridge"]
-        locations = ["Bengaluru", "Pune", "Hyderabad", "Delhi"]
-
-        for i in range(self.config.n_users):
-            uid = f"user_{i}"
-            org = self.rng.choice(orgs)
-            loc = self.rng.choice(locations)
-            graph.nodes[uid] = Node(uid, NodeType.USER, {"name": f"Person {i}", "org": org, "location": loc})
-            org_id = f"org_{org.lower().replace(' ', '_')}"
-            loc_id = f"loc_{loc.lower()}"
-            graph.nodes.setdefault(org_id, Node(org_id, NodeType.ORG, {"name": org}))
-            graph.nodes.setdefault(loc_id, Node(loc_id, NodeType.LOCATION, {"name": loc}))
-            graph.edges.append(Edge(uid, "works_at", org_id))
-            graph.edges.append(Edge(uid, "located_in", loc_id))
-
-            if self.rng.random() < self.config.alias_density:
-                alias = f"alias_{i}_{self.rng.randint(100,999)}"
-                graph.nodes[alias] = Node(alias, NodeType.ALIAS, {"handle": f"@{alias}"})
-                graph.edges.append(Edge(alias, "alias_of", uid))
-
-        users = [n for n in graph.nodes.values() if n.node_type == NodeType.USER]
-        for _ in range(max(1, self.config.n_users // 2)):
-            a, b = self.rng.sample(users, 2)
-            graph.edges.append(Edge(a.node_id, "connected_to", b.node_id, confidence=0.8))
-
-        self._apply_seed_nodes(graph)
-        self._apply_seed_edges(graph)
-
-        if self.config.seeding.llm_generate_remaining_graph:
-            llm_edges = self._llm_expand_graph(graph, self.config.seeding.llm_generated_edge_budget)
-            for edge in llm_edges:
-                self._add_edge_if_missing(graph, edge)
-        return graph
-
-    def build_platform_views(self, graph: CanonicalGraph) -> PlatformViews:
-        users = [n for n in graph.nodes.values() if n.node_type == NodeType.USER]
-        aliases = [n for n in graph.nodes.values() if n.node_type == NodeType.ALIAS]
-        alias_owner = {e.src: e.dst for e in graph.edges if e.rel == "alias_of"}
-        user_aliases: dict[str, list[str]] = {}
-        for alias_id, user_id in alias_owner.items():
-            user_aliases.setdefault(user_id, []).append(alias_id)
-        node_names = {
-            node_id: str((node.attrs or {}).get("name") or (node.attrs or {}).get("handle") or node_id)
-            for node_id, node in graph.nodes.items()
-        }
-
-        microblog_posts: list[dict] = []
-        for i, user in enumerate(users):
-            poster = user.node_id
-            if aliases and self.rng.random() < 0.45:
-                candidate = self.rng.choice(aliases).node_id
-                poster = candidate
-            text = f"Update {i} from {user.attrs['org']} #{user.attrs['location'].lower()}"
-            if self.rng.random() < self.config.noise_level:
-                text = f"Rumor: {text} maybe fake"
-            microblog_posts.append(
-                {
-                    "post_id": f"post_{i}",
-                    "user_id": poster,
-                    "canonical_user": alias_owner.get(poster, user.node_id),
-                    "text": text,
-                    "references": [],
-                    "reference_names": [],
-                    "mentions": [f"user_{self.rng.randint(0, self.config.n_users - 1)}"],
-                    "timestamp": 1000 + i,
-                }
-            )
-
-        authored_posts: dict[str, str] = {}
-        post_references: dict[str, list[str]] = {}
-        for edge in graph.edges:
-            if edge.rel == "authored_post":
-                authored_posts[edge.dst] = edge.src
-            elif edge.rel == "references" and edge.src.startswith("post_"):
-                post_references.setdefault(edge.src, []).append(edge.dst)
-
-        for post_id, author_id in authored_posts.items():
-            refs = post_references.get(post_id, [])
-            ref_names = [node_names.get(ref, ref) for ref in refs]
-            author_label = node_names.get(author_id, author_id)
-            text_parts = [f"{post_id} update from {author_label}"]
-            if ref_names:
-                text_parts.append("references " + ", ".join(ref_names))
-            if refs:
-                text_parts.append("ids " + ", ".join(refs))
-            post_payload = {
-                "post_id": post_id,
-                "user_id": author_id,
-                "canonical_user": alias_owner.get(author_id, author_id),
-                "text": ". ".join(text_parts),
-                "references": refs,
-                "reference_names": ref_names,
-                "mentions": [],
-                "timestamp": 5000 + len(microblog_posts),
-            }
-            existing_idx = next((idx for idx, row in enumerate(microblog_posts) if row["post_id"] == post_id), None)
-            if existing_idx is None:
-                microblog_posts.append(post_payload)
-            else:
-                microblog_posts[existing_idx] = post_payload
-
-        forum_threads: list[dict] = []
-        for i in range(max(8, self.config.n_users // 3)):
-            author = self.rng.choice(users).node_id
-            forum_threads.append(
-                {
-                    "thread_id": f"thr_{i}",
-                    "topic": self.rng.choice(["security", "startup", "ai", "infra"]),
-                    "author_id": author,
-                    "comments": [
-                        {"user_id": self.rng.choice(users).node_id, "text": "Following this."},
-                        {"user_id": self.rng.choice(users).node_id, "text": "Interesting link."},
-                    ],
-                    "references": [],
-                    "discusses": [],
-                }
-            )
-
-        authored_threads: dict[str, str] = {}
-        thread_refs: dict[str, list[str]] = {}
-        thread_discusses: dict[str, list[str]] = {}
-        for edge in graph.edges:
-            if edge.rel == "authored_thread":
-                authored_threads[edge.dst] = edge.src
-            elif edge.rel == "references" and edge.src.startswith(("thr_", "thread_")):
-                thread_refs.setdefault(edge.src, []).append(edge.dst)
-            elif edge.rel == "discusses" and edge.src.startswith(("thr_", "thread_")):
-                thread_discusses.setdefault(edge.src, []).append(edge.dst)
-
-        for thread_id, author_id in authored_threads.items():
-            node = graph.nodes.get(thread_id)
-            refs = thread_refs.get(thread_id, [])
-            discussed = thread_discusses.get(thread_id, [])
-            comments = []
-            for ref in refs:
-                comments.append({"user_id": author_id, "text": f"Reference: {node_names.get(ref, ref)} ({ref})"})
-            for item in discussed:
-                comments.append({"user_id": author_id, "text": f"Discusses: {node_names.get(item, item)} ({item})"})
-            thread_payload = {
-                "thread_id": thread_id,
-                "topic": str((node.attrs or {}).get("topic", "seeded")) if node else "seeded",
-                "author_id": author_id,
-                "title": node_names.get(thread_id, thread_id),
-                "comments": comments,
-                "references": refs,
-                "discusses": discussed,
-            }
-            existing_idx = next((idx for idx, row in enumerate(forum_threads) if row["thread_id"] == thread_id), None)
-            if existing_idx is None:
-                forum_threads.append(thread_payload)
-            else:
-                forum_threads[existing_idx] = thread_payload
-
-        profiles: list[dict] = []
-        for user in users:
-            conns = [e.dst for e in graph.edges if e.src == user.node_id and e.rel == "connected_to"][:5]
-            org_id = next((e.dst for e in graph.edges if e.src == user.node_id and e.rel == "works_at"), "")
-            location_id = next((e.dst for e in graph.edges if e.src == user.node_id and e.rel == "located_in"), "")
-            profiles.append(
-                {
-                    "user_id": user.node_id,
-                    "name": user.attrs["name"],
-                    "org": user.attrs["org"],
-                    "org_id": org_id,
-                    "location": user.attrs["location"],
-                    "location_id": location_id,
-                    "alias_ids": sorted(user_aliases.get(user.node_id, [])),
-                    "connections": conns,
-                    "work_history": [user.attrs["org"]],
-                }
-            )
-
-        for i in range(int(len(users) * self.config.red_herring_rate)):
-            profiles.append(
-                {
-                    "user_id": f"noise_{i}",
-                    "name": f"P{self.rng.randint(100,999)}",
-                    "org": self.rng.choice(["Stealth Co", "Unknown Ventures"]),
-                    "org_id": "",
-                    "location": self.rng.choice(["Remote", "Unknown"]),
-                    "location_id": "",
-                    "alias_ids": [],
-                    "connections": [],
-                    "work_history": [],
-                }
-            )
-        return PlatformViews(microblog_posts, forum_threads, profiles, alias_lookup=alias_owner)
-
-    def generate_tasks(self, graph: CanonicalGraph, views: PlatformViews, count: int = 12) -> list[TaskInstance]:
-        tasks = self._seeded_tasks(graph)
-        target_count = max(1, count, len(tasks))
-
-        llm_budget = min(
-            max(0, self.config.seeding.llm_generated_task_budget),
-            max(0, target_count - len(tasks)),
-        )
-        if self.config.seeding.llm_generate_remaining_tasks and llm_budget > 0:
-            tasks.extend(self._llm_generated_tasks(graph, count=llm_budget, start_idx=len(tasks)))
-
-        if len(tasks) < target_count and self._template_fallback_allowed():
-            tasks.extend(self._template_tasks(graph, count=target_count - len(tasks), start_idx=len(tasks)))
-
-        if not tasks:
-            tasks.extend(self._template_tasks(graph, count=target_count, start_idx=0))
-
-        return tasks[:target_count]

scratch/osint/src/osint_env/domain/__init__.py

@@ -1,2 +0,0 @@
-"""Core domain models."""
-

scratch/osint/src/osint_env/domain/models.py

@@ -1,185 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass, field
-from enum import Enum
-from typing import Any
-
-from pydantic import BaseModel, ConfigDict, Field
-
-
-class NodeType(str, Enum):
-    USER = "user"
-    ALIAS = "alias"
-    ORG = "org"
-    LOCATION = "location"
-    POST = "post"
-    THREAD = "thread"
-    EVENT = "event"
-
-
-class ActionType(str, Enum):
-    CALL_TOOL = "CALL_TOOL"
-    ADD_EDGE = "ADD_EDGE"
-    ANSWER = "ANSWER"
-
-
-@dataclass(slots=True)
-class Node:
-    node_id: str
-    node_type: NodeType
-    attrs: dict[str, Any] = field(default_factory=dict)
-
-
-@dataclass(slots=True)
-class Edge:
-    src: str
-    rel: str
-    dst: str
-    confidence: float = 1.0
-
-
-@dataclass(slots=True)
-class CanonicalGraph:
-    nodes: dict[str, Node] = field(default_factory=dict)
-    edges: list[Edge] = field(default_factory=list)
-
-
-@dataclass(slots=True)
-class ToolCall:
-    tool_name: str
-    args: dict[str, Any]
-
-
-class Action(BaseModel):
-    """Structured action payload used by OpenEnv step()."""
-
-    model_config = ConfigDict(extra="forbid")
-
-    action_type: ActionType
-    payload: dict[str, Any] = Field(default_factory=dict)
-
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        # Backward-compatible positional form: Action(action_type, payload)
-        if args:
-            if len(args) != 2:
-                raise TypeError("Action() accepts either keyword fields or 2 positional args")
-            if "action_type" in kwargs or "payload" in kwargs:
-                raise TypeError("Action() cannot mix positional and keyword fields")
-            kwargs["action_type"] = args[0]
-            kwargs["payload"] = args[1]
-        super().__init__(**kwargs)
-
-
-class Observation(BaseModel):
-    """Typed observation payload returned by reset()/step()/state()."""
-
-    model_config = ConfigDict(extra="forbid")
-
-    tool_outputs: list[dict[str, Any]] = Field(default_factory=list)
-    graph_snapshot: dict[str, Any] = Field(default_factory=dict)
-    action_history: list[dict[str, Any]] = Field(default_factory=list)
-    task: dict[str, Any] = Field(default_factory=dict)
-
-
-class Reward(BaseModel):
-    """Typed reward payload for structured reward accounting."""
-
-    model_config = ConfigDict(extra="forbid")
-
-    value: float = 0.0
-    components: dict[str, float] = Field(default_factory=dict)
-
-
-@dataclass(slots=True)
-class TaskInstance:
-    task_id: str
-    task_type: str
-    question: str
-    answer: str
-    supporting_edges: list[Edge]
-    metadata: dict[str, Any] = field(default_factory=dict)
-
-
-@dataclass(slots=True)
-class SeedNodeSpec:
-    node_id: str
-    node_type: NodeType | str
-    attrs: dict[str, Any] = field(default_factory=dict)
-
-
-@dataclass(slots=True)
-class SeedEdgeSpec:
-    src: str
-    rel: str
-    dst: str
-    confidence: float = 1.0
-
-
-@dataclass(slots=True)
-class SeedQuestionSpec:
-    question: str
-    answer: str | None = None
-    task_type: str = "seeded"
-    supporting_edges: list[SeedEdgeSpec] = field(default_factory=list)
-    metadata: dict[str, Any] = field(default_factory=dict)
-
-
-@dataclass(slots=True)
-class SeedingConfig:
-    seeded_nodes: list[SeedNodeSpec] = field(default_factory=list)
-    seeded_edges: list[SeedEdgeSpec] = field(default_factory=list)
-    seeded_questions: list[SeedQuestionSpec] = field(default_factory=list)
-    llm_generate_remaining_graph: bool = True
-    llm_generate_remaining_tasks: bool = True
-    llm_generated_edge_budget: int = 6
-    llm_generated_task_budget: int = 8
-    llm_generation_parallel: bool = True
-    llm_generation_workers: int = 3
-    llm_generation_retries: int = 2
-    allow_template_fallback_on_llm_failure: bool = False
-
-
-@dataclass(slots=True)
-class SwarmConfig:
-    enabled: bool = False
-    max_agents: int = 3
-    max_breadth: int = 2
-    max_width: int = 2
-    max_depth: int = 2
-    planner_rounds: int = 2
-    tools_per_agent: int = 1
-
-
-@dataclass(slots=True)
-class SpawnRewardConfig:
-    lambda_parallel: float = 0.15
-    lambda_finish: float = 0.20
-    anneal: float = 1.0
-    max_parallel_hint: int = 3
-
-
-@dataclass(slots=True)
-class LLMConfig:
-    provider: str = "mock"
-    model: str = "qwen3:2b"
-    temperature: float = 0.1
-    max_tokens: int = 256
-    timeout_seconds: int = 240
-    ollama_base_url: str = "http://127.0.0.1:11434"
-    openai_base_url: str = "https://api.openai.com/v1"
-    openai_api_key_env: str = "OPENAI_API_KEY"
-    openai_api_key: str = ""
-
-
-@dataclass(slots=True)
-class EnvironmentConfig:
-    n_users: int = 40
-    alias_density: float = 0.35
-    noise_level: float = 0.15
-    red_herring_rate: float = 0.1
-    max_steps: int = 18
-    seed: int = 7
-    seeding: SeedingConfig = field(default_factory=SeedingConfig)
-    swarm: SwarmConfig = field(default_factory=SwarmConfig)
-    spawn_reward: SpawnRewardConfig = field(default_factory=SpawnRewardConfig)
-    llm: LLMConfig = field(default_factory=LLMConfig)

scratch/osint/src/osint_env/env/__init__.py

@@ -1,2 +0,0 @@
-"""Environment package."""
-

scratch/osint/src/osint_env/env/environment.py

@@ -1,260 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass, field
-from typing import TYPE_CHECKING, Any
-
-from osint_env.data.generator import DatasetGenerator
-from osint_env.domain.models import Action, ActionType, Edge, EnvironmentConfig, Observation, TaskInstance
-from osint_env.env.openenv_compat import Env
-from osint_env.env.reward import (
-    build_reward_model,
-    compute_answer_reward,
-    compute_edge_reward,
-    compute_graph_f1,
-)
-from osint_env.memory.store import MemoryGraph, SemanticMemory
-from osint_env.platforms.tools import ToolRegistry
-
-if TYPE_CHECKING:
-    from osint_env.llm.interface import LLMClient
-
-
-@dataclass(slots=True)
-class EpisodeState:
-    task: TaskInstance
-    task_index: int = 0
-    difficulty: str = "hard"
-    step_count: int = 0
-    done: bool = False
-    total_reward: float = 0.0
-    tool_calls: int = 0
-    redundant_tool_calls: int = 0
-    action_history: list[dict[str, Any]] = field(default_factory=list)
-    tool_outputs: list[dict[str, Any]] = field(default_factory=list)
-    answer: str | None = None
-    call_fingerprints: set[str] = field(default_factory=set)
-    reward_components: dict[str, float] = field(default_factory=dict)
-
-
-class OSINTEnvironment(Env):
-    def __init__(self, config: EnvironmentConfig, llm: "LLMClient | None" = None):
-        super().__init__(
-            name="OSINTEnvironment",
-            state_space="json-observation",
-            action_space=["CALL_TOOL", "ADD_EDGE", "ANSWER"],
-            episode_max_length=config.max_steps,
-        )
-        self.config = config
-        self.generator = DatasetGenerator(config, llm=llm)
-        self.graph = self.generator.build_canonical_graph()
-        self.views = self.generator.build_platform_views(self.graph)
-        self.tasks = self.generator.generate_tasks(self.graph, self.views, count=24)
-        self.reward_model = build_reward_model(self.graph)
-        self.tools = ToolRegistry(self.views)
-        self.memory_graph = MemoryGraph()
-        self.semantic_memory = SemanticMemory()
-        self._task_idx = 0
-        self.state: EpisodeState | None = None
-
-    @staticmethod
-    def _normalize_difficulty(value: str) -> str:
-        token = str(value or "").strip().lower()
-        if token in {"easy", "e"}:
-            return "easy"
-        if token in {"mid", "medium", "m"}:
-            return "medium"
-        if token in {"high", "hard", "h"}:
-            return "hard"
-        return "hard"
-
-    def _resolve_task_difficulty(self, task: TaskInstance, task_index: int) -> str:
-        metadata = dict(task.metadata or {})
-        if "difficulty" in metadata:
-            return self._normalize_difficulty(str(metadata.get("difficulty", "")))
-        if task_index < 10:
-            return "easy"
-        if task_index < 20:
-            return "medium"
-        return "hard"
-
-    def reset(self) -> Observation:
-        task_index = self._task_idx % len(self.tasks)
-        task = self.tasks[task_index]
-        self._task_idx += 1
-        self.state = EpisodeState(
-            task=task,
-            task_index=task_index,
-            difficulty=self._resolve_task_difficulty(task, task_index),
-        )
-        self.memory_graph = MemoryGraph()
-        self.semantic_memory = SemanticMemory()
-        return self._observation()
-
-    def step(self, action: Action) -> tuple[Observation, float, bool, dict[str, Any]]:
-        if self.state is None:
-            raise RuntimeError("Call reset() before step().")
-        if self.state.done:
-            return self._observation(), 0.0, True, self._info()
-
-        self.state.step_count += 1
-        reward = 0.0
-
-        if action.action_type == ActionType.CALL_TOOL:
-            reward += self._handle_tool(action.payload)
-        elif action.action_type == ActionType.ADD_EDGE:
-            reward += self._handle_add_edge(action.payload)
-        elif action.action_type == ActionType.ANSWER:
-            reward += self._handle_answer(action.payload)
-        else:
-            reward -= 0.5
-
-        if self.state.step_count >= self.config.max_steps and not self.state.done:
-            self.state.done = True
-            reward -= 0.3
-
-        self.state.total_reward += reward
-        self.state.action_history.append({"type": action.action_type.value, "payload": action.payload, "reward": reward})
-        return self._observation(), reward, self.state.done, self._info()
-
-    def _handle_tool(self, payload: dict[str, Any]) -> float:
-        if self.state is None:
-            return 0.0
-        tool_name = payload["tool_name"]
-        args = payload.get("args", {})
-        fp = f"{tool_name}:{sorted(args.items())}"
-        self.state.tool_calls += 1
-        if fp in self.state.call_fingerprints:
-            self.state.redundant_tool_calls += 1
-            penalty = -0.2
-        else:
-            penalty = 0.05
-            self.state.call_fingerprints.add(fp)
-
-        invalid_tool_penalty = 0.0
-        try:
-            if tool_name == "search_memory":
-                query = str(args.get("query", "")).strip()
-                top_k = int(args.get("k", 5)) if str(args.get("k", "")).strip() else 5
-                results = self.semantic_memory.search(query=query, k=max(1, top_k)) if query else []
-                output = {"results": results, "count": len(results)}
-            else:
-                output = self.tools.call(tool_name, args)
-        except Exception as exc:
-            output = {"error": str(exc)}
-            invalid_tool_penalty = -0.25
-        self.state.tool_outputs.append({"tool": tool_name, "args": args, "output": output})
-        self.semantic_memory.add(f"{tool_name} {args} {output}", {"tool": tool_name})
-        relevance_bonus = 0.08 * self._tool_relevance(self.state.task, output)
-        total = penalty + relevance_bonus + invalid_tool_penalty
-        self._accumulate_reward_components(
-            {
-                "tool_novelty": penalty,
-                "tool_relevance": relevance_bonus,
-                "invalid_tool_penalty": invalid_tool_penalty,
-            }
-        )
-        return total
-
-    def _handle_add_edge(self, payload: dict[str, Any]) -> float:
-        if self.state is None:
-            return 0.0
-        edge = Edge(payload["src"], payload["rel"], payload["dst"], float(payload.get("confidence", 1.0)))
-        existing_edges = list(self.memory_graph.edges)
-        added = self.memory_graph.add_edge(edge)
-        if not added:
-            self._accumulate_reward_components({"duplicate_edge_penalty": -0.15})
-            return -0.15
-
-        breakdown = compute_edge_reward(
-            edge=edge,
-            task=self.state.task,
-            existing_edges=existing_edges,
-            step_count=self.state.step_count,
-            model=self.reward_model,
-            graph=self.graph,
-            difficulty=self.state.difficulty,
-        )
-        self._accumulate_reward_components(breakdown.to_dict())
-        return breakdown.total
-
-    def _handle_answer(self, payload: dict[str, Any]) -> float:
-        if self.state is None:
-            return 0.0
-        proposed = str(payload.get("answer", "")).strip()
-        self.state.answer = proposed
-        self.state.done = True
-        breakdown = compute_answer_reward(
-            proposed_answer=proposed,
-            task=self.state.task,
-            pred_edges=self.memory_graph.edges,
-            tool_outputs=self.state.tool_outputs,
-            step_count=self.state.step_count,
-            model=self.reward_model,
-            difficulty=self.state.difficulty,
-        )
-        self._accumulate_reward_components(breakdown.to_dict())
-        return breakdown.total
-
-    def _tool_relevance(self, task: TaskInstance, output: dict[str, Any]) -> float:
-        haystack = str(output).lower()
-        clues = {task.answer.lower()}
-        for edge in task.supporting_edges:
-            clues.add(edge.src.lower())
-            clues.add(edge.dst.lower())
-            clues.add(edge.rel.lower())
-        if not clues:
-            return 0.0
-        matches = sum(1 for token in clues if token in haystack)
-        return matches / len(clues)
-
-    def _accumulate_reward_components(self, values: dict[str, float]) -> None:
-        if self.state is None:
-            return
-        for key, value in values.items():
-            self.state.reward_components[key] = self.state.reward_components.get(key, 0.0) + float(value)
-
-    def _observation(self) -> Observation:
-        if self.state is None:
-            raise RuntimeError("State is not initialized.")
-        metadata = dict(self.state.task.metadata or {})
-        grader = metadata.get("grader") if isinstance(metadata.get("grader"), dict) else None
-        task_payload = {
-            "task_id": self.state.task.task_id,
-            "task_type": self.state.task.task_type,
-            "question": self.state.task.question,
-            "difficulty": self.state.difficulty,
-            "grader": (
-                dict(grader)
-                if grader is not None
-                else {
-                    "type": "difficulty_exact_match",
-                    "answer_type": "node_id",
-                    "case_sensitive": True,
-                    "reward_profile": self.state.difficulty,
-                }
-            ),
-        }
-        if "scenario" in metadata:
-            task_payload["scenario"] = str(metadata.get("scenario", ""))
-        return Observation(
-            tool_outputs=self.state.tool_outputs[-5:],
-            graph_snapshot=self.memory_graph.to_snapshot(),
-            action_history=self.state.action_history[-10:],
-            task=task_payload,
-        )
-
-    def _info(self) -> dict[str, Any]:
-        if self.state is None:
-            return {}
-        return {
-            "step_count": self.state.step_count,
-            "difficulty": self.state.difficulty,
-            "task_index": self.state.task_index,
-            "total_reward": self.state.total_reward,
-            "tool_calls": self.state.tool_calls,
-            "redundant_tool_calls": self.state.redundant_tool_calls,
-            "task_answer": self.state.task.answer,
-            "agent_answer": self.state.answer,
-            "graph_f1": compute_graph_f1(self.memory_graph.edges, self.state.task.supporting_edges),
-            "reward_components": dict(self.state.reward_components),
-        }

scratch/osint/src/osint_env/env/openenv_compat.py

@@ -1,20 +0,0 @@
-from __future__ import annotations
-
-try:
-    from openenv.env import Env
-except ImportError:
-    class Env:
-        """Minimal fallback used when openenv is not installed locally."""
-
-        def __init__(
-            self,
-            name: str,
-            state_space: str,
-            action_space: list[str],
-            episode_max_length: int,
-        ) -> None:
-            self.name = name
-            self.state_space = state_space
-            self.action_space = action_space
-            self.episode_max_length = episode_max_length
-

scratch/osint/src/osint_env/env/reward.py

@@ -1,483 +0,0 @@
-from __future__ import annotations
-
-import json
-import math
-import re
-from collections import Counter
-from dataclasses import asdict, dataclass
-
-from osint_env.domain.models import CanonicalGraph, Edge, TaskInstance
-
-
-@dataclass(slots=True)
-class RewardModel:
-    relation_idf: dict[str, float]
-    max_relation_idf: float
-    hub_penalty: dict[str, float]
-    max_hub_penalty: float
-    type_priors: dict[tuple[str, str, str], float]
-
-
-@dataclass(slots=True)
-class EdgeRewardBreakdown:
-    total: float
-    global_accuracy: float
-    soft_shaping: float
-    efficiency: float
-    diversity: float
-    relation_informativeness: float
-    entity_informativeness: float
-    connectivity_gain: float
-
-    def to_dict(self) -> dict[str, float]:
-        return asdict(self)
-
-
-@dataclass(slots=True)
-class AnswerRewardBreakdown:
-    total: float
-    format_reward: float
-    correctness: float
-    knowledge_carrier: float
-    knowledge_indexing: float
-    connectivity: float
-    graph_f1: float
-    efficiency: float
-    compactness: float
-    relation_informativeness: float
-    entity_informativeness: float
-    repetition_penalty: float
-
-    def to_dict(self) -> dict[str, float]:
-        return asdict(self)
-
-
-def _normalize_difficulty(value: str) -> str:
-    token = str(value or "").strip().lower()
-    if token in {"easy", "e"}:
-        return "easy"
-    if token in {"mid", "medium", "m"}:
-        return "medium"
-    if token in {"high", "hard", "h"}:
-        return "hard"
-    return "hard"
-
-
-def build_reward_model(graph: CanonicalGraph) -> RewardModel:
-    relation_freq: Counter[str] = Counter(e.rel for e in graph.edges)
-    total_edges = max(1, len(graph.edges))
-    relation_idf = {
-        rel: math.log((1.0 + total_edges) / (1.0 + freq)) + 1.0 for rel, freq in relation_freq.items()
-    }
-    max_relation_idf = max(relation_idf.values()) if relation_idf else 1.0
-
-    degree: Counter[str] = Counter()
-    for edge in graph.edges:
-        degree[edge.src] += 1
-        degree[edge.dst] += 1
-    hub_penalty = {node_id: math.log(1.0 + deg) for node_id, deg in degree.items()}
-    max_hub_penalty = max(hub_penalty.values()) if hub_penalty else 1.0
-
-    type_counts: Counter[tuple[str, str, str]] = Counter()
-    rel_counts: Counter[str] = Counter()
-    for edge in graph.edges:
-        src = graph.nodes.get(edge.src)
-        dst = graph.nodes.get(edge.dst)
-        if src is None or dst is None:
-            continue
-        key = (str(src.node_type.value), edge.rel, str(dst.node_type.value))
-        type_counts[key] += 1
-        rel_counts[edge.rel] += 1
-    type_priors = {
-        key: count / max(1, rel_counts[key[1]]) for key, count in type_counts.items()
-    }
-
-    return RewardModel(
-        relation_idf=relation_idf,
-        max_relation_idf=max_relation_idf,
-        hub_penalty=hub_penalty,
-        max_hub_penalty=max_hub_penalty,
-        type_priors=type_priors,
-    )
-
-
-def edge_in_truth(edge: Edge, task: TaskInstance) -> bool:
-    return any(e.src == edge.src and e.rel == edge.rel and e.dst == edge.dst for e in task.supporting_edges)
-
-
-def _cosine(a: Counter[str], b: Counter[str]) -> float:
-    common = set(a) & set(b)
-    num = sum(a[t] * b[t] for t in common)
-    den = math.sqrt(sum(v * v for v in a.values())) * math.sqrt(sum(v * v for v in b.values()))
-    return (num / den) if den else 0.0
-
-
-def _edge_signature(edge: Edge) -> Counter[str]:
-    # Approximate path/edge embedding using relation and endpoint prefixes.
-    src_prefix = edge.src.split("_", 1)[0]
-    dst_prefix = edge.dst.split("_", 1)[0]
-    return Counter({f"rel:{edge.rel}": 2, f"src:{src_prefix}": 1, f"dst:{dst_prefix}": 1})
-
-
-def _soft_fact_score(edge: Edge, model: RewardModel, graph: CanonicalGraph) -> float:
-    if any(e.src == edge.src and e.rel == edge.rel and e.dst == edge.dst for e in graph.edges):
-        return 1.0
-
-    src = graph.nodes.get(edge.src)
-    dst = graph.nodes.get(edge.dst)
-    if src is None or dst is None:
-        return 0.0
-
-    type_key = (str(src.node_type.value), edge.rel, str(dst.node_type.value))
-    prior = model.type_priors.get(type_key, 0.0)
-
-    # A tiny domain heuristic: alias links are common and worth soft credit even without exact support edge.
-    alias_bias = 0.2 if (edge.rel == "alias_of" and edge.src.startswith("alias_") and edge.dst.startswith("user_")) else 0.0
-    relation_exists = any(e.rel == edge.rel for e in graph.edges)
-    relation_bonus = 0.1 if relation_exists else 0.0
-    return max(0.0, min(1.0, 0.1 + (0.65 * prior) + alias_bias + relation_bonus))
-
-
-def _normalized_relation_info(rel: str, model: RewardModel) -> float:
-    idf = model.relation_idf.get(rel, 1.0)
-    return idf / max(1e-6, model.max_relation_idf)
-
-
-def _normalized_entity_info(src: str, dst: str, model: RewardModel) -> float:
-    src_h = model.hub_penalty.get(src, 0.0)
-    dst_h = model.hub_penalty.get(dst, 0.0)
-    mean_hub = (src_h + dst_h) / 2.0
-    # UniRel-style preference for low-degree intermediates: lower hub penalty -> higher informativeness.
-    return 1.0 - (mean_hub / max(1e-6, model.max_hub_penalty))
-
-
-def _is_reachable_undirected(edges: list[Edge], src: str, dst: str) -> bool:
-    if src == dst:
-        return True
-    adj: dict[str, set[str]] = {}
-    for edge in edges:
-        adj.setdefault(edge.src, set()).add(edge.dst)
-        adj.setdefault(edge.dst, set()).add(edge.src)
-    seen = {src}
-    stack = [src]
-    while stack:
-        node = stack.pop()
-        for nxt in adj.get(node, set()):
-            if nxt == dst:
-                return True
-            if nxt not in seen:
-                seen.add(nxt)
-                stack.append(nxt)
-    return False
-
-
-def _connectivity_gain(edge: Edge, existing_edges: list[Edge]) -> float:
-    # Reward edges that bridge disconnected regions and penalize already-connected shortcuts.
-    if edge.src == edge.dst:
-        return -0.06
-    already_connected = _is_reachable_undirected(existing_edges, edge.src, edge.dst)
-    if already_connected:
-        return -0.03
-    return 0.10
-
-
-def _sigmoid_temperature(value: float, temperature: float = 2.0) -> float:
-    scaled = float(value) / max(1e-6, float(temperature))
-    if scaled >= 0:
-        z = math.exp(-scaled)
-        return 1.0 / (1.0 + z)
-    z = math.exp(scaled)
-    return z / (1.0 + z)
-
-
-def compute_edge_reward(
-    edge: Edge,
-    task: TaskInstance,
-    existing_edges: list[Edge],
-    step_count: int,
-    model: RewardModel,
-    graph: CanonicalGraph,
-    difficulty: str = "hard",
-) -> EdgeRewardBreakdown:
-    in_truth = edge_in_truth(edge, task)
-    difficulty_level = _normalize_difficulty(difficulty)
-
-    # DeepPath-inspired global accuracy term.
-    global_accuracy = 0.85 if in_truth else -0.55
-
-    # D18 reward shaping: R = Rb + (1 - Rb) * f, where f is a soft fact plausibility score.
-    base_reward = 1.0 if in_truth else 0.0
-    shaped = base_reward + ((1.0 - base_reward) * _soft_fact_score(edge, model, graph))
-    soft_shaping = 0.30 * (shaped - 0.5)
-
-    # DeepPath-inspired efficiency term: earlier useful edges are better.
-    efficiency = 0.10 * (1.0 / max(1, step_count))
-
-    # DeepPath-inspired diversity term: discourage repeated edge patterns.
-    if not existing_edges:
-        diversity = 0.08
-    else:
-        new_sig = _edge_signature(edge)
-        avg_similarity = sum(_cosine(new_sig, _edge_signature(e)) for e in existing_edges) / len(existing_edges)
-        novelty = 1.0 - avg_similarity
-        diversity = 0.14 * (novelty - 0.5)
-
-    # UniRel-style informativeness terms.
-    relation_informativeness = 0.12 * (_normalized_relation_info(edge.rel, model) - 0.5)
-    entity_informativeness = 0.12 * (_normalized_entity_info(edge.src, edge.dst, model) - 0.5)
-
-    # Additional structural utility shaping for KG construction.
-    connectivity_gain = _connectivity_gain(edge, existing_edges)
-
-    if difficulty_level == "easy":
-        global_accuracy = 0.75 if in_truth else -0.45
-        soft_shaping = 0.0
-        diversity = 0.0
-        relation_informativeness = 0.0
-        entity_informativeness = 0.0
-        connectivity_gain = 0.0
-        efficiency = 0.15 * (1.0 / max(1, step_count))
-    elif difficulty_level == "medium":
-        diversity = 0.0
-        relation_informativeness = 0.0
-        entity_informativeness = 0.0
-
-    raw_total = (
-        global_accuracy
-        + soft_shaping
-        + efficiency
-        + diversity
-        + relation_informativeness
-        + entity_informativeness
-        + connectivity_gain
-    )
-    total = _sigmoid_temperature(raw_total, temperature=2.0)
-    return EdgeRewardBreakdown(
-        total=total,
-        global_accuracy=global_accuracy,
-        soft_shaping=soft_shaping,
-        efficiency=efficiency,
-        diversity=diversity,
-        relation_informativeness=relation_informativeness,
-        entity_informativeness=entity_informativeness,
-        connectivity_gain=connectivity_gain,
-    )
-
-
-def _connectivity_ratio(pred_edges: list[Edge], task: TaskInstance) -> float:
-    nodes = {e.src for e in task.supporting_edges} | {e.dst for e in task.supporting_edges}
-    if len(nodes) <= 1:
-        return 1.0
-
-    adj: dict[str, set[str]] = {}
-    for edge in pred_edges:
-        adj.setdefault(edge.src, set()).add(edge.dst)
-        adj.setdefault(edge.dst, set()).add(edge.src)
-
-    start = next(iter(nodes))
-    seen = {start}
-    stack = [start]
-    while stack:
-        cur = stack.pop()
-        for nxt in adj.get(cur, set()):
-            if nxt not in seen:
-                seen.add(nxt)
-                stack.append(nxt)
-    return len(seen & nodes) / max(1, len(nodes))
-
-
-def _knowledge_indexing_recall(task: TaskInstance, tool_outputs: list[dict[str, object]]) -> float:
-    gold_terms = {task.answer.lower()}
-    for edge in task.supporting_edges:
-        gold_terms.add(edge.src.lower())
-        gold_terms.add(edge.dst.lower())
-        gold_terms.add(edge.rel.lower())
-
-    serialized = json.dumps(tool_outputs).lower()
-    covered = sum(1 for term in gold_terms if term and term in serialized)
-    return covered / max(1, len(gold_terms))
-
-
-def _knowledge_carrier_reward(pred_edges: list[Edge], task: TaskInstance) -> float:
-    pred = {(e.src, e.rel, e.dst) for e in pred_edges}
-    truth = {(e.src, e.rel, e.dst) for e in task.supporting_edges}
-    deducible = bool(truth & pred)
-    return 0.4 if deducible else -0.2
-
-
-def _extract_query_entities(question: str) -> set[str]:
-    pattern = r"\b(?:alias|user|org|loc|post|thr|thread|event)_[a-zA-Z0-9_]+\b"
-    return set(re.findall(pattern, question))
-
-
-def _max_connected_seed_count(pred_edges: list[Edge], seeds: set[str]) -> int:
-    if not seeds:
-        return 0
-    adj: dict[str, set[str]] = {}
-    for edge in pred_edges:
-        adj.setdefault(edge.src, set()).add(edge.dst)
-        adj.setdefault(edge.dst, set()).add(edge.src)
-
-    best = 1
-    for seed in seeds:
-        seen = {seed}
-        stack = [seed]
-        while stack:
-            cur = stack.pop()
-            for nxt in adj.get(cur, set()):
-                if nxt not in seen:
-                    seen.add(nxt)
-                    stack.append(nxt)
-        connected_seed_count = len(seeds & seen)
-        best = max(best, connected_seed_count)
-    return best
-
-
-def _unirel_connectivity_score(pred_edges: list[Edge], seeds: set[str]) -> float:
-    # UniRel-style discrete connectivity range projected to [-1, 1] for stable weighting.
-    n = len(seeds)
-    if n <= 1:
-        return 0.0
-
-    connected = _max_connected_seed_count(pred_edges, seeds)
-    raw = -math.floor(n / 2) + max(0, connected - 1)
-    lo = -math.floor(n / 2)
-    hi = math.ceil(n / 2) - 1
-    if hi <= lo:
-        return 0.0
-    return ((raw - lo) / (hi - lo)) * 2.0 - 1.0
-
-
-def _subgraph_relation_informativeness(pred_edges: list[Edge], model: RewardModel | None) -> float:
-    if not pred_edges or model is None:
-        return 0.0
-    avg = sum(_normalized_relation_info(edge.rel, model) for edge in pred_edges) / len(pred_edges)
-    return avg - 0.5
-
-
-def _subgraph_entity_informativeness(pred_edges: list[Edge], model: RewardModel | None) -> float:
-    if not pred_edges or model is None:
-        return 0.0
-    avg = sum(_normalized_entity_info(edge.src, edge.dst, model) for edge in pred_edges) / len(pred_edges)
-    return avg - 0.5
-
-
-def _relation_repetition_ratio(pred_edges: list[Edge]) -> float:
-    if len(pred_edges) <= 1:
-        return 0.0
-    rels = [edge.rel for edge in pred_edges]
-    unique = len(set(rels))
-    return 1.0 - (unique / len(rels))
-
-
-def _deducible_answer(proposed_answer: str, task: TaskInstance, pred_edges: list[Edge]) -> bool:
-    if proposed_answer != task.answer:
-        return False
-    truth = {(edge.src, edge.rel, edge.dst) for edge in task.supporting_edges}
-    pred = {(edge.src, edge.rel, edge.dst) for edge in pred_edges}
-    if truth & pred:
-        return True
-
-    seeds = _extract_query_entities(task.question)
-    if not seeds:
-        return False
-    for seed in seeds:
-        if _is_reachable_undirected(pred_edges, seed, proposed_answer):
-            return True
-    return False
-
-
-def compute_answer_reward(
-    proposed_answer: str,
-    task: TaskInstance,
-    pred_edges: list[Edge],
-    tool_outputs: list[dict[str, object]],
-    step_count: int,
-    model: RewardModel | None = None,
-    difficulty: str = "hard",
-) -> AnswerRewardBreakdown:
-    difficulty_level = _normalize_difficulty(difficulty)
-
-    format_reward = 0.15 if proposed_answer else -0.55
-    correctness = 1.15 if proposed_answer == task.answer else -1.0
-
-    # AutoGraph-R1 style task utility decomposition.
-    knowledge_carrier = 0.50 if _deducible_answer(proposed_answer, task, pred_edges) else -0.25
-    knowledge_indexing = 0.45 * _knowledge_indexing_recall(task, tool_outputs)
-
-    # UniRel-style connectivity over seed entities.
-    seed_entities = _extract_query_entities(task.question)
-    seed_entities.add(task.answer)
-    connectivity = 0.30 * _unirel_connectivity_score(pred_edges, seed_entities)
-
-    graph_f1 = 0.55 * compute_graph_f1(pred_edges, task.supporting_edges)
-    efficiency = 0.12 * (1.0 / max(1, step_count))
-
-    extra_edges = max(0, len(pred_edges) - len(task.supporting_edges))
-    compactness = -0.05 * extra_edges
-
-    relation_informativeness = 0.12 * _subgraph_relation_informativeness(pred_edges, model)
-    entity_informativeness = 0.12 * _subgraph_entity_informativeness(pred_edges, model)
-
-    # AutoGraph-R1 repetition control variant used in larger models.
-    repetition_penalty = -0.10 * _relation_repetition_ratio(pred_edges)
-
-    if difficulty_level == "easy":
-        knowledge_carrier = 0.0
-        knowledge_indexing = 0.25 * _knowledge_indexing_recall(task, tool_outputs)
-        connectivity = 0.0
-        graph_f1 = 0.0
-        efficiency = 0.18 * (1.0 / max(1, step_count))
-        compactness = 0.0
-        relation_informativeness = 0.0
-        entity_informativeness = 0.0
-        repetition_penalty = 0.0
-    elif difficulty_level == "medium":
-        connectivity = 0.18 * _unirel_connectivity_score(pred_edges, seed_entities)
-        graph_f1 = 0.35 * compute_graph_f1(pred_edges, task.supporting_edges)
-        compactness = -0.04 * extra_edges
-        relation_informativeness = 0.0
-        entity_informativeness = 0.0
-        repetition_penalty = 0.0
-
-    raw_total = (
-        format_reward
-        + correctness
-        + knowledge_carrier
-        + knowledge_indexing
-        + connectivity
-        + graph_f1
-        + efficiency
-        + compactness
-        + relation_informativeness
-        + entity_informativeness
-        + repetition_penalty
-    )
-    total = _sigmoid_temperature(raw_total, temperature=2.0)
-    return AnswerRewardBreakdown(
-        total=total,
-        format_reward=format_reward,
-        correctness=correctness,
-        knowledge_carrier=knowledge_carrier,
-        knowledge_indexing=knowledge_indexing,
-        connectivity=connectivity,
-        graph_f1=graph_f1,
-        efficiency=efficiency,
-        compactness=compactness,
-        relation_informativeness=relation_informativeness,
-        entity_informativeness=entity_informativeness,
-        repetition_penalty=repetition_penalty,
-    )
-
-
-def compute_graph_f1(pred_edges: list[Edge], truth_edges: list[Edge]) -> float:
-    pred = {(e.src, e.rel, e.dst) for e in pred_edges}
-    truth = {(e.src, e.rel, e.dst) for e in truth_edges}
-    if not pred and not truth:
-        return 1.0
-    if not pred or not truth:
-        return 0.0
-    tp = len(pred & truth)
-    p = tp / len(pred) if pred else 0.0
-    r = tp / len(truth) if truth else 0.0
-    return (2 * p * r / (p + r)) if (p + r) else 0.0

scratch/osint/src/osint_env/env/spawn_reward_hooks.py

@@ -1,93 +0,0 @@
-from __future__ import annotations
-
-import math
-
-
-def critical_steps(main_steps: list[int], parallel_subagent_steps: list[list[int]]) -> int:
-    """Compute critical-step latency proxy used in Kimi-style PARL shaping.
-
-    For each stage t, we add:
-      Smain(t) + max_i Ssub,i(t)
-    where Ssub,i(t) is the i-th sub-agent step count for that stage.
-    """
-    if len(main_steps) != len(parallel_subagent_steps):
-        raise ValueError("main_steps and parallel_subagent_steps must have the same length")
-
-    total = 0
-    for stage_main, stage_sub in zip(main_steps, parallel_subagent_steps):
-        main = max(0, int(stage_main))
-        longest_sub = max((max(0, int(v)) for v in stage_sub), default=0)
-        total += main + longest_sub
-    return total
-
-
-def parl_style_spawn_reward(
-    task_outcome_reward: float,
-    spawn_count: int,
-    finished_subtasks: int,
-    critical_steps: int,
-    lambda_parallel: float = 0.15,
-    lambda_finish: float = 0.20,
-    anneal: float = 1.0,
-    breadth: int | None = None,
-    depth: int | None = None,
-    max_parallel_hint: int | None = None,
-) -> float:
-    """Kimi K2.5 inspired PARL reward utility for future multi-agent branches.
-
-    This helper intentionally does not orchestrate agents. It only exposes the reward shape:
-
-      r_parl = r_perf + a * (lambda_parallel * r_parallel + lambda_finish * r_finish + r_latency)
-
-    where:
-    - r_parallel encourages non-zero agent spawning (avoids serial collapse)
-    - r_finish rewards meaningful completion, preventing spawn-only reward hacking
-    - r_latency favors lower critical-step execution paths
-
-    The optional breadth/depth controls are small shaping terms for future branches where
-    orchestration state includes tree shape telemetry.
-    """
-    spawn_count = max(0, int(spawn_count))
-    finished_subtasks = max(0, int(finished_subtasks))
-    critical_steps = max(1, int(critical_steps))
-    anneal = max(0.0, min(1.0, anneal))
-    lambda_parallel = max(0.0, float(lambda_parallel))
-    lambda_finish = max(0.0, float(lambda_finish))
-    breadth = max(0, int(breadth or 0))
-    depth = max(0, int(depth or 0))
-    max_parallel_hint = max(0, int(max_parallel_hint or 0))
-
-    if spawn_count == 0:
-        r_parallel = 0.0
-        r_finish = 0.0
-    else:
-        # Saturating incentive for parallelism so reward cannot grow unbounded with spawns.
-        r_parallel = math.tanh(spawn_count / 4.0)
-        if max_parallel_hint > 0:
-            utilization = min(1.0, spawn_count / max_parallel_hint)
-            r_parallel *= (0.7 + (0.3 * utilization))
-
-        r_finish = min(1.0, finished_subtasks / spawn_count)
-
-    if breadth > 0:
-        breadth_bonus = 0.04 * math.tanh(breadth / 6.0)
-    else:
-        breadth_bonus = 0.0
-
-    if depth > 0:
-        # Mild depth penalty discourages brittle over-decomposition chains.
-        depth_penalty = -0.03 * math.tanh(max(0, depth - 1) / 4.0)
-    else:
-        depth_penalty = 0.0
-
-    # Optional latency shaping hook using critical steps (higher is worse).
-    r_latency = 0.05 * (1.0 / critical_steps)
-
-    auxiliary = (
-        (lambda_parallel * r_parallel)
-        + (lambda_finish * r_finish)
-        + r_latency
-        + breadth_bonus
-        + depth_penalty
-    )
-    return float(task_outcome_reward) + (anneal * auxiliary)

scratch/osint/src/osint_env/eval/__init__.py

@@ -1,2 +0,0 @@
-"""Evaluation package."""
-

scratch/osint/src/osint_env/eval/leaderboard.py

@@ -1,83 +0,0 @@
-from __future__ import annotations
-
-import json
-from datetime import datetime, timezone
-from pathlib import Path
-from typing import Any
-
-
-def _utc_now() -> str:
-    return datetime.now(tz=timezone.utc).replace(microsecond=0).isoformat()
-
-
-def load_leaderboard(path: str | Path) -> list[dict[str, Any]]:
-    file_path = Path(path)
-    if not file_path.exists():
-        return []
-    with file_path.open("r", encoding="utf-8") as f:
-        data = json.load(f)
-    if not isinstance(data, list):
-        return []
-    return data
-
-
-def save_leaderboard(path: str | Path, records: list[dict[str, Any]]) -> None:
-    file_path = Path(path)
-    file_path.parent.mkdir(parents=True, exist_ok=True)
-    with file_path.open("w", encoding="utf-8") as f:
-        json.dump(records, f, indent=2, sort_keys=True)
-
-
-def _metric_value(record: dict[str, Any], sort_by: str) -> float:
-    metrics = record.get("metrics", {})
-    return float(metrics.get(sort_by, 0.0))
-
-
-def sorted_leaderboard(records: list[dict[str, Any]], sort_by: str = "leaderboard_score") -> list[dict[str, Any]]:
-    return sorted(records, key=lambda r: _metric_value(r, sort_by), reverse=True)
-
-
-def append_leaderboard_record(
-    path: str | Path,
-    summary: dict[str, Any],
-    episodes: int,
-    run_name: str | None = None,
-    config: dict[str, Any] | None = None,
-) -> dict[str, Any]:
-    records = load_leaderboard(path)
-    run_id = f"run_{len(records) + 1:04d}"
-    record = {
-        "run_id": run_id,
-        "run_name": run_name or run_id,
-        "created_at": _utc_now(),
-        "episodes": int(episodes),
-        "config": config or {},
-        "metrics": summary,
-    }
-    records.append(record)
-    save_leaderboard(path, records)
-    return record
-
-
-def render_leaderboard_table(records: list[dict[str, Any]], top_k: int = 20, sort_by: str = "leaderboard_score") -> str:
-    ranked = sorted_leaderboard(records, sort_by=sort_by)[:top_k]
-    header = "| rank | run | score | success | graph_f1 | retrieval | structural | spawn | reward | tool_eff |\n"
-    sep = "|---|---|---:|---:|---:|---:|---:|---:|---:|---:|\n"
-    rows: list[str] = []
-    for idx, rec in enumerate(ranked, start=1):
-        m = rec.get("metrics", {})
-        rows.append(
-            "| {rank} | {run} | {score:.4f} | {succ:.3f} | {f1:.3f} | {retrieval:.3f} | {structural:.3f} | {spawn:.3f} | {reward:.3f} | {tool:.3f} |".format(
-                rank=idx,
-                run=rec.get("run_name", rec.get("run_id", "run")),
-                score=float(m.get("leaderboard_score", 0.0)),
-                succ=float(m.get("task_success_rate", 0.0)),
-                f1=float(m.get("avg_graph_f1", 0.0)),
-                retrieval=float(m.get("retrieval_signal", 0.0)),
-                structural=float(m.get("structural_signal", 0.0)),
-                spawn=float(m.get("spawn_signal", 0.0)),
-                reward=float(m.get("avg_reward", 0.0)),
-                tool=float(m.get("tool_efficiency", 0.0)),
-            )
-        )
-    return header + sep + "\n".join(rows)