Remove PatchedVertexAI since the upstream bug is fixed

Author: aabmass

samples/langgraph-sql-agent/agent.py

@@ -26,7 +26,7 @@
 from opentelemetry import trace
 from sqlalchemy import create_engine
 
-from patched_vertexai import PatchedChatVertexAI
+from langchain_google_vertexai import ChatVertexAI
 from utils import ask_prompt, console, print_markdown, render_messages
 
 SYSTEM_PROMPT = SystemMessage(
@@ -71,7 +71,7 @@
 
 
 def run_agent(*, model_name: str, recursion_limit: int = 50) -> None:
-    model = PatchedChatVertexAI(model=model_name)
+    model = ChatVertexAI(model=model_name)
     checkpointer = InMemorySaver()
 
     # Ephemeral sqlite database per run

samples/langgraph-sql-agent/patched_vertexai.py

@@ -20,8 +20,8 @@ readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
     "langchain-community>=0.3.16",
-    "langchain-google-vertexai>=2.0.7",
-    "langgraph>=0.4.3",
+    "langchain-google-vertexai>=2.0.27",
+    "langgraph>=0.5.1",
     "opentelemetry-exporter-gcp-logging>=1.9.0a0",
     "opentelemetry-exporter-gcp-monitoring>=1.9.0a0",
     "opentelemetry-exporter-otlp-proto-grpc>=1.33.0",