Skip to content

Commit 62496c3

Browse files
committed
test: adapt test_retrieve_session_no_session to unified SessionInfo schema
1 parent 1c3ba20 commit 62496c3

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/praisonai-agents/tests/unit/test_managed_backend.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -511,11 +511,14 @@ def test_retrieve_session(self):
511511
assert info["usage"]["output_tokens"] == 200
512512

513513
def test_retrieve_session_no_session(self):
514-
"""retrieve_session with no session returns empty dict."""
514+
"""retrieve_session with no session returns SessionInfo defaults (per #1429 unified schema)."""
515515
from praisonai.integrations.managed_agents import ManagedAgent
516516

517517
m = ManagedAgent(api_key="test-key")
518-
assert m.retrieve_session() == {}
518+
info = m.retrieve_session()
519+
assert info["id"] == ""
520+
assert info["status"] == "unknown"
521+
assert info["usage"] == {"input_tokens": 0, "output_tokens": 0}
519522

520523
def test_list_sessions(self):
521524
"""list_sessions should return list of session dicts."""

0 commit comments

Comments
 (0)