fmt: cleaned up graph implementation

src/config/agent.rs

@@ -802,6 +802,7 @@ fn resolve_document_paths(
             } else {
                 PathBuf::from(&resolved_path)
             };
+
             document_paths.push(format!("{}:{}", protocol, new_path.display()));
         } else if Path::new(&resolve_home_dir(path)).is_relative() {
             let new_path = safe_join_path(agent_data_dir, path)
@@ -829,6 +830,7 @@ async fn init_graph_rags(
     if info_flag {
         return Ok(rags);
     }
+
     for (node_id, node) in &graph.nodes {
         let NodeType::Rag(rag_node) = &node.node_type else {
             continue;
@@ -1059,6 +1061,7 @@ variables:
                 output: done
             "#};
         let graph: Graph = serde_yaml::from_str(&yaml).unwrap();
+
         let config = AgentConfig::from_graph("my-agent-dir", &graph);
 
         assert_eq!(config.name, "my-agent-dir");
@@ -1101,6 +1104,7 @@ variables:
     fn from_graph_keeps_defaults_for_llm_loop_fields() {
         let yaml = "name: g\nstart: x\nnodes:\n  x:\n    id: x\n    type: end\n    output: ok\n";
         let graph: Graph = serde_yaml::from_str(yaml).unwrap();
+
         let config = AgentConfig::from_graph("d", &graph);
 
         assert!(!config.auto_continue);

src/config/mod.rs

@@ -273,6 +273,7 @@ pub fn install_assets(category: AssetCategory) -> Result<()> {
     }
 
     println!("Reinstalled bundled {label} ({})", target.display());
+
     Ok(())
 }
 
@@ -297,6 +298,7 @@ fn confirm_asset_overwrite(category: AssetCategory, label: &str, target: &Path)
     };
     let prompt = format!("{} {body}\nContinue? [y/N] ", warning_text("WARNING:"));
     let answer = read_single_key(&['y', 'Y', 'n', 'N'], 'n', &prompt)?;
+
     Ok(matches!(answer, 'y' | 'Y'))
 }
 

src/config/request_context.rs

@@ -33,11 +33,11 @@ use indoc::formatdoc;
 use inquire::{Confirm, MultiSelect, Text, list_option::ListOption, validator::Validation};
 use parking_lot::RwLock;
 use std::collections::{HashMap, HashSet};
-use std::env;
 use std::fs::{File, OpenOptions, read_dir, read_to_string, remove_dir_all, remove_file};
 use std::io::Write;
 use std::path::{Path, PathBuf};
 use std::sync::Arc;
+use std::{env, fs};
 
 pub struct AutoContinueConfig {
     pub enabled: bool,
@@ -46,10 +46,6 @@ pub struct AutoContinueConfig {
     pub continuation_prompt: Option<String>,
 }
 
-/// Controls how LLM token streams are presented to the user. `Silent` is set
-/// on branch contexts during parallel graph super-steps so concurrent LLM
-/// calls don't interleave token-by-token on stdout — the full response still
-/// lands in graph state via the normal output_schema / state_updates pathway.
 #[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
 pub enum RenderMode {
     #[default]
@@ -166,27 +162,18 @@ impl RequestContext {
 
     /// Forks the context for one parallel branch of a graph super-step.
     ///
-    /// Each branch gets a fresh, owned clone — mutations (role swap,
+    /// Each branch gets a fresh, owned clone. Mutations (role swap,
     /// `before/after_chat_completion`, tool tracker, last_message, etc.) are
     /// scoped to the branch and discarded when the branch finishes. The
     /// user-visible state communication happens through the graph's
     /// `StateManager` (via `fork_for_branch_state` + `diff_against` +
-    /// `apply_branch_writes` reducers), NOT through `RequestContext`.
+    /// `apply_branch_writes` reducers), and not through `RequestContext`.
     ///
     /// Distinction from `new_for_child`: `new_for_child` builds a fresh context
-    /// for a SPAWNED SUB-AGENT (different agent identity, different supervisor
+    /// for a spawned sub-agent (different agent identity, different supervisor
     /// hierarchy, depth+1, fresh tool tracker). `fork_for_branch` keeps the
-    /// caller's identity and supervisor hierarchy — it's a sibling clone of the
-    /// SAME logical agent, running one of N parallel work items.
-    ///
-    /// Behavior of per-field cloning:
-    /// - `Arc`-wrapped fields (`app`, `rag`, `supervisor`, `parent_supervisor`,
-    ///   `inbox`, `escalation_queue`) — shared via Arc::clone
-    /// - Owned heap fields (`model`, `role`, `session`, `agent`, `tool_scope`,
-    ///   `todo_list`, etc.) — deep `.clone()` so the branch can mutate freely
-    /// - `auto_continue_count` reset to 0 (each branch starts a fresh
-    ///   continuation budget)
-    /// - `last_continuation_response` reset to None
+    /// caller's identity and supervisor hierarchy; it's a sibling clone of the
+    /// same logical agent, running one of N parallel work items.
     pub fn fork_for_branch(&self) -> Self {
         Self {
             app: Arc::clone(&self.app),
@@ -1419,6 +1406,7 @@ impl RequestContext {
             env!("CARGO_CRATE_NAME"),
             mcp_path.display(),
         );
+
         Ok(())
     }
 
@@ -1503,20 +1491,23 @@ impl RequestContext {
         } else {
             config_path
         };
+
         ensure_parent_exists(&target_path)?;
         if !target_path.exists() {
-            std::fs::write(
+            fs::write(
                 &target_path,
                 "# see https://github.com/Dark-Alex-17/loki/blob/main/config.agent.example.yaml\n",
             )
             .with_context(|| format!("Failed to write to '{}'", target_path.display()))?;
         }
+
         let editor = app.editor()?;
         edit_file(&editor, &target_path)?;
         println!(
             "NOTE: Remember to reload the agent if there are changes made to '{}'",
             target_path.display()
         );
+
         Ok(())
     }
 
@@ -2026,7 +2017,7 @@ impl RequestContext {
                 .collect();
         } else if cmd == ".agent" {
             if args.len() == 2 {
-                let dir = paths::agent_data_dir(args[0]).join(super::SESSIONS_DIR_NAME);
+                let dir = paths::agent_data_dir(args[0]).join(SESSIONS_DIR_NAME);
                 values = list_file_names(dir, ".yaml")
                     .into_iter()
                     .map(|v| (v, None))