mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-15 14:08:00 +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
|
return int(value) if value else None
|
||||||
|
|
||||||
async def _get_sessions_from_values(self, values: list[str]) -> list[Session]:
|
async def list_sessions(self) -> list[Session]:
|
||||||
"""
|
values = await self.kvstore.values_in_range(
|
||||||
Helper method to process values from the KV store and extract session objects.
|
start_key=f"session:{self.agent_id}:",
|
||||||
|
end_key=f"session:{self.agent_id}:\xff\xff\xff\xff",
|
||||||
Args:
|
)
|
||||||
values: List of serialized JSON values from the KV store
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
List of parsed Session objects
|
|
||||||
"""
|
|
||||||
sessions = []
|
sessions = []
|
||||||
for value in values:
|
for value in values:
|
||||||
try:
|
try:
|
||||||
|
@ -207,14 +202,6 @@ class AgentPersistence:
|
||||||
continue
|
continue
|
||||||
return sessions
|
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:
|
async def delete_session_turns(self, session_id: str) -> None:
|
||||||
"""Delete all turns and their associated data for a session.
|
"""Delete all turns and their associated data for a session.
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue