Skip to content

Commit 9942988

Browse files
committed
fix: set apply_memory_context=True in TraceSession.rag/code paths
TraceSession injects memory_state into extra but bypasses GroundingClient's keyword arg path, so apply_memory_context was never set. Session-based callers would send memory_state without enabling hot_context augmentation on the server.
1 parent 355f9d7 commit 9942988

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/latence/client.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,7 @@ def rag(
388388
extra.setdefault("session_id", self.session_id)
389389
if self.memory_state:
390390
extra.setdefault("memory_state", self.memory_state)
391+
extra.setdefault("apply_memory_context", True)
391392
extra.setdefault("metadata", dict(self.metadata))
392393
result = self._owner.grounding.rag(
393394
extra=extra,
@@ -412,6 +413,7 @@ def code(
412413
extra.setdefault("session_id", self.session_id)
413414
if self.memory_state:
414415
extra.setdefault("memory_state", self.memory_state)
416+
extra.setdefault("apply_memory_context", True)
415417
extra.setdefault("metadata", dict(self.metadata))
416418
result = self._owner.grounding.code(
417419
extra=extra,

0 commit comments

Comments
 (0)