From c036eda8ecde38a1ed746beddd870120f3e78b7e Mon Sep 17 00:00:00 2001
From: Pamela Fox <pamela.fox@gmail.com>
Date: Mon, 11 Aug 2025 17:07:07 -0700
Subject: [PATCH] Move state into the same file

---
 src/agent/graph.py | 11 ++++++++++-
 src/agent/state.py | 12 ------------
 2 files changed, 10 insertions(+), 13 deletions(-)
 delete mode 100644 src/agent/state.py

diff --git a/src/agent/graph.py b/src/agent/graph.py
index d160e9d..2f9d72c 100644
--- a/src/agent/graph.py
+++ b/src/agent/graph.py
@@ -1,5 +1,8 @@
 """Module for defining the agent's workflow graph and human interaction nodes."""
 
+from __future__ import annotations
+
+from dataclasses import dataclass
 from typing import Any, Dict
 
 from langchain_core.runnables import RunnableConfig
@@ -12,7 +15,13 @@
 )
 from langgraph.types import interrupt
 
-from agent.state import State
+
+@dataclass
+class State:
+    """Class representing the state of the agent interaction."""
+
+    interrupt_response: str = "example"
+
 
 
 async def human_node(state: State, config: RunnableConfig) -> Dict[str, Any]:
diff --git a/src/agent/state.py b/src/agent/state.py
deleted file mode 100644
index 76e43a4..0000000
--- a/src/agent/state.py
+++ /dev/null
@@ -1,12 +0,0 @@
-"""State module for managing agent state."""
-
-from __future__ import annotations
-
-from dataclasses import dataclass
-
-
-@dataclass
-class State:
-    """Class representing the state of the agent interaction."""
-
-    interrupt_response: str = "example"