From 24874f99d4be6eb72e95316324afae67bd81999b Mon Sep 17 00:00:00 2001
From: Mahmut CAVDAR <4072246+mcavdar@users.noreply.github.com>
Date: Tue, 28 Oct 2025 09:55:51 +0100
Subject: [PATCH] Change context_schema from ContextSchema to Context

---
 src/oss/python/migrate/langchain-v1.mdx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/oss/python/migrate/langchain-v1.mdx b/src/oss/python/migrate/langchain-v1.mdx
index b013aa3e05..bdec3b0509 100644
--- a/src/oss/python/migrate/langchain-v1.mdx
+++ b/src/oss/python/migrate/langchain-v1.mdx
@@ -690,7 +690,7 @@ class Context:
 agent = create_agent(
     model=model,
     tools=tools,
-    context_schema=ContextSchema  # [!code highlight]
+    context_schema=Context  # [!code highlight]
 )
 
 result = agent.invoke(