chore: Handle exception in preload_memory_tool to not fail the llm request

Resolves https://github.com/google/adk-python/issues/3069

PiperOrigin-RevId: 814391260
This commit is contained in:
Shangjie Chen
2025-10-02 14:41:27 -07:00
committed by Copybara-Service
parent 5b8d523a4b
commit b170a84279
@@ -14,6 +14,7 @@
from __future__ import annotations
import logging
from typing import TYPE_CHECKING
from typing_extensions import override
@@ -25,6 +26,8 @@ from .tool_context import ToolContext
if TYPE_CHECKING:
from ..models import LlmRequest
logger = logging.getLogger('google_adk.' + __name__)
class PreloadMemoryTool(BaseTool):
"""A tool that preloads the memory for the current user.
@@ -56,7 +59,12 @@ class PreloadMemoryTool(BaseTool):
return
user_query: str = user_content.parts[0].text
try:
response = await tool_context.search_memory(user_query)
except Exception:
logging.warning('Failed to preload memory for query: %s', user_query)
return
if not response.memories:
return