Move state into the same file as graph

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]: