From c2e8011175ae178ff21f597e741de98f7fc56b9e Mon Sep 17 00:00:00 2001 From: Sarthak Deshpande Date: Fri, 18 Oct 2024 12:52:28 +0530 Subject: [PATCH] Added a check on turn_ids --- .../impls/meta_reference/agents/agents.py | 31 ++++++++++--------- 1 file changed, 16 insertions(+), 15 deletions(-) diff --git a/llama_stack/providers/impls/meta_reference/agents/agents.py b/llama_stack/providers/impls/meta_reference/agents/agents.py index c938886b1..14db8ab20 100644 --- a/llama_stack/providers/impls/meta_reference/agents/agents.py +++ b/llama_stack/providers/impls/meta_reference/agents/agents.py @@ -160,21 +160,22 @@ class MetaReferenceAgentsImpl(Agents): f"Could not JSON decode session for {agent_id} and {session_id}" ) from e turns = [] - for turn_id in turn_ids: - turn = await self.persistence_store.get(f"session:{agent_id}:{session_id}:{turn_id}") - try: - turn = json.loads(turn) - except json.JSONDecodeError as e: - raise ValueError( - f"Could not JSON decode session for {agent_id} and {session_id}" - ) from e - try: - turn = Turn(**turn) - except Exception as e: - raise ValueError( - f"Could not validate(?) Turns for {turn_id}" - ) from e - turns.append(turn) + if turn_ids: + for turn_id in turn_ids: + turn = await self.persistence_store.get(f"session:{agent_id}:{session_id}:{turn_id}") + try: + turn = json.loads(turn) + except json.JSONDecodeError as e: + raise ValueError( + f"Could not JSON decode session for {agent_id} and {session_id}" + ) from e + try: + turn = Turn(**turn) + except Exception as e: + raise ValueError( + f"Could not validate(?) Turns for {turn_id}" + ) from e + turns.append(turn) return Session( session_name=session.session_name, session_id=session_id,