From d33c5a20e417aa8f60edc170b7a7d46b55362c99 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Thu, 1 May 2025 12:14:09 -0700
Subject: [PATCH] Fix

---
 libs/langgraph/langgraph/func/__init__.py | 2 +-
 libs/langgraph/langgraph/pregel/call.py   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/libs/langgraph/langgraph/func/__init__.py b/libs/langgraph/langgraph/func/__init__.py
index 689ac05ab..dd6f5f4a5 100644
--- a/libs/langgraph/langgraph/func/__init__.py
+++ b/libs/langgraph/langgraph/func/__init__.py
@@ -145,7 +145,7 @@ def task(
                 func.__name__ = name
 
         call_func = functools.partial(
-            call, func, retry=retry_policies, cache=cache_policy
+            call, func, retry=retry_policies, cache_policy=cache_policy
         )
         object.__setattr__(call_func, "_is_pregel_task", True)
         return functools.update_wrapper(call_func, func)
diff --git a/libs/langgraph/langgraph/pregel/call.py b/libs/langgraph/langgraph/pregel/call.py
index bc41eafb5..d9abc6cf3 100644
--- a/libs/langgraph/langgraph/pregel/call.py
+++ b/libs/langgraph/langgraph/pregel/call.py
@@ -251,7 +251,7 @@ def call(
         func,
         (args, kwargs),
         retry=retry,
-        cache=cache_policy,
+        cache_policy=cache_policy,
         callbacks=config["callbacks"],
     )
     return fut