fix: Explictly throw execption when custom id is set for vertex ai session service.

PiperOrigin-RevId: 757815469
This commit is contained in:
Shangjie Chen
2025-05-12 10:18:48 -07:00
committed by Copybara-Service
parent 4c4cfb74ae
commit 70a6bfae90
2 changed files with 18 additions and 0 deletions

View File

@@ -261,3 +261,15 @@ def test_create_session():
assert session == session_service.get_session(
app_name='123', user_id='user', session_id=session_id
)
def test_create_session_with_custom_session_id():
session_service = mock_vertex_ai_session_service()
with pytest.raises(ValueError) as excinfo:
session_service.create_session(
app_name='123', user_id='user', session_id='1'
)
assert str(excinfo.value) == (
'User-provided Session id is not supported for VertexAISessionService.'
)