phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 07:14:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added a check on turn_ids

Browse source
This commit is contained in:
Sarthak Deshpande 2024-10-18 12:52:28 +05:30
parent 9bcd03fe2a
commit c2e8011175
1 changed files with 16 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

31
llama_stack/providers/impls/meta_reference/agents/agents.py
View file

@ -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,