From dc5e3201da7ee67408f06dcf99690748788be887 Mon Sep 17 00:00:00 2001
From: Alex Clarke <alex.j.tusa@gmail.com>
Date: Thu, 10 Sep 2026 11:52:44 -0600
Subject: [PATCH] refactor(graph): move the cfg(test) OUTPUT_KEY import into
 llm's test module

PR #24 follow-up 7.
---
 src/graph/llm.rs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/graph/llm.rs b/src/graph/llm.rs
index 041d9fe..e07f927 100644
--- a/src/graph/llm.rs
+++ b/src/graph/llm.rs
@@ -1,7 +1,5 @@
 use super::state::StateManager;
 use super::state_updates;
-#[cfg(test)]
-use super::state_updates::OUTPUT_KEY;
 use super::structured;
 use super::types::LlmNode;
 use crate::client::{Model, ModelType, call_chat_completions};
@@ -475,6 +473,7 @@ fn format_schema_hint(schema: &Value) -> String {
 
 #[cfg(test)]
 mod tests {
+    use super::super::state_updates::OUTPUT_KEY;
     use super::super::types::*;
     use super::*;
     use serde_json::json;