Skip to content

Commit 5ef1cd3

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Allow both camelCase and snake_case in _StreamRunRequest
PiperOrigin-RevId: 823217390
1 parent 57d2709 commit 5ef1cd3

File tree

2 files changed

+6
-2
lines changed
  • vertexai
    • agent_engines/templates
    • preview/reasoning_engines/templates

2 files changed

+6
-2
lines changed

vertexai/agent_engines/templates/adk.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,9 @@ def __init__(self, **kwargs):
169169
self.user_id: Optional[str] = kwargs.get("user_id", _DEFAULT_USER_ID)
170170
# The user ID.
171171

172-
self.session_id: Optional[str] = kwargs.get("session_id")
172+
self.session_id: Optional[str] = kwargs.get("session_id") or kwargs.get(
173+
"sessionId"
174+
)
173175
# The session ID.
174176

175177

vertexai/preview/reasoning_engines/templates/adk.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,9 @@ def __init__(self, **kwargs):
183183
self.user_id: Optional[str] = kwargs.get("user_id", _DEFAULT_USER_ID)
184184
# The user ID.
185185

186-
self.session_id: Optional[str] = kwargs.get("session_id")
186+
self.session_id: Optional[str] = kwargs.get("session_id") or kwargs.get(
187+
"sessionId"
188+
)
187189
# The session ID.
188190

189191

0 commit comments

Comments
 (0)