From dbdc71d33a9961bed6da42fb5d8a6345ea0da3af Mon Sep 17 00:00:00 2001
From: Justin Li <justinlibc@gmail.com>
Date: Tue, 30 Dec 2025 17:55:59 -0800
Subject: [PATCH] fix: Fix type mismatch in ContextFilterPlugin.custom_filter
 signature

---
 src/google/adk/plugins/context_filter_plugin.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/google/adk/plugins/context_filter_plugin.py b/src/google/adk/plugins/context_filter_plugin.py
index b778de02ad..baf93e89fd 100644
--- a/src/google/adk/plugins/context_filter_plugin.py
+++ b/src/google/adk/plugins/context_filter_plugin.py
@@ -19,8 +19,9 @@
 from typing import List
 from typing import Optional
 
+from google.genai import types
+
 from ..agents.callback_context import CallbackContext
-from ..events.event import Event
 from ..models.llm_request import LlmRequest
 from ..models.llm_response import LlmResponse
 from .base_plugin import BasePlugin
@@ -34,7 +35,9 @@ class ContextFilterPlugin(BasePlugin):
   def __init__(
       self,
       num_invocations_to_keep: Optional[int] = None,
-      custom_filter: Optional[Callable[[List[Event]], List[Event]]] = None,
+      custom_filter: Optional[
+          Callable[[List[types.Content]], List[types.Content]]
+      ] = None,
       name: str = "context_filter_plugin",
   ):
     """Initializes the context management plugin.