Skip to content

Commit 267215a

Browse files
committed
fix: 🚑 remove thought_signature from events
1 parent 9bb0eb9 commit 267215a

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/google/adk/sessions/redis_memory_session_service.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,14 +137,22 @@ async def append_event(self, session: Session, event: Event) -> Event:
137137
self.cache.hset(
138138
f"{State.USER_PREFIX}{session.app_name}:{session.user_id}",
139139
key.removeprefix(State.USER_PREFIX),
140-
json.dumps(value),
140+
json.dumps(self.clean_event(event)),
141141
)
142142

143143
sessions = self._load_sessions(session.app_name, session.user_id)
144144
sessions[session.id] = session.to_dict()
145145
self._save_sessions(session.app_name, session.user_id, sessions)
146146

147147
return event
148+
149+
def clean_event(self,event):
150+
for event in event["events"]:
151+
if event["content"] is None:
152+
continue
153+
for part in event["content"]["parts"]:
154+
part.pop("thought_signature")
155+
return event
148156

149157
def _merge_state(
150158
self, app_name: str, user_id: str, session: Session

0 commit comments

Comments
 (0)