File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Original file line number Diff line number Diff line change 14
14
15
15
from __future__ import annotations
16
16
17
+ import logging
17
18
from typing import TYPE_CHECKING
18
19
19
20
from typing_extensions import override
25
26
if TYPE_CHECKING :
26
27
from ..models import LlmRequest
27
28
29
+ logger = logging .getLogger ('google_adk.' + __name__ )
30
+
28
31
29
32
class PreloadMemoryTool (BaseTool ):
30
33
"""A tool that preloads the memory for the current user.
@@ -56,7 +59,12 @@ async def process_llm_request(
56
59
return
57
60
58
61
user_query : str = user_content .parts [0 ].text
59
- response = await tool_context .search_memory (user_query )
62
+ try :
63
+ response = await tool_context .search_memory (user_query )
64
+ except Exception :
65
+ logging .warning ('Failed to preload memory for query: %s' , user_query )
66
+ return
67
+
60
68
if not response .memories :
61
69
return
62
70
You can’t perform that action at this time.
0 commit comments