mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-15 06:00:48 +00:00
moving inline
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
48fc88faff
commit
b484c7ec08
1 changed files with 5 additions and 18 deletions
|
@ -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.
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue