From 9daf14fd42190f700547e3152a3ad008346a788f Mon Sep 17 00:00:00 2001
From: Yiorgis Gozadinos <ggozadinos@gmail.com>
Date: Fri, 19 Sep 2025 16:28:03 +0300
Subject: [PATCH] Research graph diagram

---
 .pre-commit-config.yaml | 10 ----------
 docs/agents.md          | 14 +++++++++++++-
 mkdocs.yml              |  6 +++++-
 3 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 57298115..19c80c7c 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -20,13 +20,3 @@ repos:
     rev: v1.1.399
     hooks:
       - id: pyright
-
-  - repo: https://github.com/RodrigoGonzalez/check-mkdocs
-    rev: v1.2.0
-    hooks:
-      - id: check-mkdocs
-        name: check-mkdocs
-        args: ["--config", "mkdocs.yml"] # Optional, mkdocs.yml is the default
-        # If you have additional plugins or libraries that are not included in
-        # check-mkdocs, add them here
-        additional_dependencies: ["mkdocs-material"]
diff --git a/docs/agents.md b/docs/agents.md
index 2d018701..4d2d82c7 100644
--- a/docs/agents.md
+++ b/docs/agents.md
@@ -38,7 +38,19 @@ print(answer)
 
 ### Research Graph
 
-The research workflow is now implemented as a typed pydantic‑graph. It plans, searches (in parallel batches), evaluates, and synthesizes into a final report — with clear stop conditions and shared state.
+The research workflow is implemented as a typed pydantic‑graph. It plans, searches (in parallel batches), evaluates, and synthesizes into a final report — with clear stop conditions and shared state.
+
+```mermaid
+---
+title: Research graph
+---
+stateDiagram-v2
+  PlanNode --> SearchDispatchNode
+  SearchDispatchNode --> EvaluateNode
+  EvaluateNode --> SearchDispatchNode
+  EvaluateNode --> SynthesizeNode
+  SynthesizeNode --> [*]
+```
 
 Key nodes:
 
diff --git a/mkdocs.yml b/mkdocs.yml
index 943f2576..2edd9719 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -76,4 +76,8 @@ markdown_extensions:
       use_pygments: true
   - pymdownx.inlinehilite
   - pymdownx.snippets
-  - pymdownx.superfences
+  - pymdownx.superfences:
+      custom_fences:
+        - name: mermaid
+          class: mermaid
+          format: !!python/name:pymdownx.superfences.fence_code_format