Skip to content

Commit c01e04f

Browse files
google-genai-botcopybara-github
authored andcommitted
feat(interaction-api): Add model field and CodeMenderAgentData response message
PiperOrigin-RevId: 928812896
1 parent ca97c58 commit c01e04f

2 files changed

Lines changed: 24 additions & 0 deletions

File tree

google/genai/_interactions/types/interaction.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,12 @@ class AgentConfigFindRequest(BaseModel):
114114
validation on a single vulnerability.
115115
"""
116116

117+
model: Optional[str] = None
118+
"""The name of the model to use for the CodeMender agent.
119+
120+
One CodeMender session will only use one model.
121+
"""
122+
117123
session_config: Optional[AgentConfigFindRequestSessionConfig] = None
118124
"""Optional session-specific configurations to override default agent behavior."""
119125

@@ -182,6 +188,12 @@ class AgentConfigFixRequest(BaseModel):
182188
This ID maps to a previously discovered vulnerability.
183189
"""
184190

191+
model: Optional[str] = None
192+
"""The name of the model to use for the CodeMender agent.
193+
194+
One CodeMender session will only use one model.
195+
"""
196+
185197
session_config: Optional[AgentConfigFixRequestSessionConfig] = None
186198
"""Optional session-specific configurations to override default agent behavior."""
187199

google/genai/_interactions/types/interaction_create_params.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -273,6 +273,12 @@ class AgentConfigFindRequest(TypedDict, total=False):
273273
validation on a single vulnerability.
274274
"""
275275

276+
model: str
277+
"""The name of the model to use for the CodeMender agent.
278+
279+
One CodeMender session will only use one model.
280+
"""
281+
276282
session_config: AgentConfigFindRequestSessionConfig
277283
"""Optional session-specific configurations to override default agent behavior."""
278284

@@ -341,6 +347,12 @@ class AgentConfigFixRequest(TypedDict, total=False):
341347
This ID maps to a previously discovered vulnerability.
342348
"""
343349

350+
model: str
351+
"""The name of the model to use for the CodeMender agent.
352+
353+
One CodeMender session will only use one model.
354+
"""
355+
344356
session_config: AgentConfigFixRequestSessionConfig
345357
"""Optional session-specific configurations to override default agent behavior."""
346358

0 commit comments

Comments
 (0)