diff --git a/src/google/adk/tools/load_memory_tool.py b/src/google/adk/tools/load_memory_tool.py
index 21d859a..dae961a 100644
--- a/src/google/adk/tools/load_memory_tool.py
+++ b/src/google/adk/tools/load_memory_tool.py
@@ -69,6 +69,7 @@ class LoadMemoryTool(FunctionTool):
                     type=types.Type.STRING,
                 )
             },
+            required=['query']
         ),
     )