From b484c7ec0852f08be825cc61a99bed3a50181db9 Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Tue, 12 Aug 2025 19:57:08 -0400 Subject: [PATCH] moving inline Signed-off-by: Francisco Javier Arceo --- .../agents/meta_reference/persistence.py | 23 ++++--------------- 1 file changed, 5 insertions(+), 18 deletions(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/persistence.py b/llama_stack/providers/inline/agents/meta_reference/persistence.py index 7053ee43c..0b234d96c 100644 --- a/llama_stack/providers/inline/agents/meta_reference/persistence.py +++ b/llama_stack/providers/inline/agents/meta_reference/persistence.py @@ -183,16 +183,11 @@ class AgentPersistence: ) return int(value) if value else None - async def _get_sessions_from_values(self, values: list[str]) -> list[Session]: - """ - Helper method to process values from the KV store and extract session objects. - - Args: - values: List of serialized JSON values from the KV store - - Returns: - List of parsed Session objects - """ + async def list_sessions(self) -> list[Session]: + values = await self.kvstore.values_in_range( + start_key=f"session:{self.agent_id}:", + end_key=f"session:{self.agent_id}:\xff\xff\xff\xff", + ) sessions = [] for value in values: try: @@ -207,14 +202,6 @@ class AgentPersistence: continue return sessions - async def list_sessions(self) -> list[Session]: - values = await self.kvstore.values_in_range( - start_key=f"session:{self.agent_id}:", - end_key=f"session:{self.agent_id}:\xff\xff\xff\xff", - ) - sessions = await self._get_sessions_from_values(values) - return sessions - async def delete_session_turns(self, session_id: str) -> None: """Delete all turns and their associated data for a session.