From f0d07499d65164b5a0cb1e7fb08a047f2b314721 Mon Sep 17 00:00:00 2001
From: Paula Wesselmann <38981219+paulaWesselmann@users.noreply.github.com>
Date: Thu, 10 Jul 2025 16:48:49 +0200
Subject: [PATCH 1/2] Fix StateGraph params
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Fix deprecated parameter names in StateGraph

Replace deprecated parameter names with their updated equivalents:
- `input` → `input_schema`
- `output` → `output_schema`

This change ensures compatibility with current LangGraph conventions. (langgraph = "0.4.9")
---
 src/enrichment_agent/graph.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/enrichment_agent/graph.py b/src/enrichment_agent/graph.py
index 64ef445..a59f749 100644
--- a/src/enrichment_agent/graph.py
+++ b/src/enrichment_agent/graph.py
@@ -178,7 +178,7 @@ def route_after_agent(
     # it ensures the system doesn't crash but instead tries to recover by calling the agent model again.
     if not isinstance(last_message, AIMessage):
         return "call_agent_model"
-    # If the "Into" tool was called, then the model provided its extraction output. Reflect on the result
+    # If the "Info" tool was called, then the model provided its extraction output. Reflect on the result
     if last_message.tool_calls and last_message.tool_calls[0]["name"] == "Info":
         return "reflect"
     # The last message is a tool call that is not "Info" (extraction output)
@@ -215,7 +215,7 @@ def route_after_checker(
 
 # Create the graph
 workflow = StateGraph(
-    State, input=InputState, output=OutputState, config_schema=Configuration
+    State, input_schema=InputState, output_schema=OutputState, config_schema=Configuration
 )
 workflow.add_node(call_agent_model)
 workflow.add_node(reflect)

From 3d563461c868ccf25d73cfe22ff81828e9d1d50e Mon Sep 17 00:00:00 2001
From: Paula Wesselmann <38981219+paulaWesselmann@users.noreply.github.com>
Date: Thu, 10 Jul 2025 16:57:02 +0200
Subject: [PATCH 2/2] match toml version

---
 src/enrichment_agent/graph.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/enrichment_agent/graph.py b/src/enrichment_agent/graph.py
index a59f749..1b3c504 100644
--- a/src/enrichment_agent/graph.py
+++ b/src/enrichment_agent/graph.py
@@ -215,7 +215,7 @@ def route_after_checker(
 
 # Create the graph
 workflow = StateGraph(
-    State, input_schema=InputState, output_schema=OutputState, config_schema=Configuration
+    State, input=InputState, output=OutputState, config_schema=Configuration
 )
 workflow.add_node(call_agent_model)
 workflow.add_node(reflect)