Skip to content

Commit 6a6674d

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Allow both camelCase and snake_case in _StreamRunRequest
PiperOrigin-RevId: 823613001
1 parent 5dd51a2 commit 6a6674d

File tree

2 files changed

+12
-4
lines changed
  • vertexai
    • agent_engines/templates
    • preview/reasoning_engines/templates

2 files changed

+12
-4
lines changed

vertexai/agent_engines/templates/adk.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,10 +166,14 @@ def __init__(self, **kwargs):
166166
)
167167
# The authorizations of the user, keyed by authorization ID.
168168

169-
self.user_id: Optional[str] = kwargs.get("user_id", _DEFAULT_USER_ID)
169+
self.user_id: Optional[str] = kwargs.get("user_id") or kwargs.get(
170+
"userId", _DEFAULT_USER_ID
171+
)
170172
# The user ID.
171173

172-
self.session_id: Optional[str] = kwargs.get("session_id")
174+
self.session_id: Optional[str] = kwargs.get("session_id") or kwargs.get(
175+
"sessionId"
176+
)
173177
# The session ID.
174178

175179

vertexai/preview/reasoning_engines/templates/adk.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,10 +180,14 @@ def __init__(self, **kwargs):
180180
)
181181
# The authorizations of the user, keyed by authorization ID.
182182

183-
self.user_id: Optional[str] = kwargs.get("user_id", _DEFAULT_USER_ID)
183+
self.user_id: Optional[str] = kwargs.get("user_id") or kwargs.get(
184+
"userId", _DEFAULT_USER_ID
185+
)
184186
# The user ID.
185187

186-
self.session_id: Optional[str] = kwargs.get("session_id")
188+
self.session_id: Optional[str] = kwargs.get("session_id") or kwargs.get(
189+
"sessionId"
190+
)
187191
# The session ID.
188192

189193

0 commit comments

Comments
 (0)