diff --git a/google/genai/_interactions/types/interaction.py b/google/genai/_interactions/types/interaction.py index 41c241f2b..725d88e72 100644 --- a/google/genai/_interactions/types/interaction.py +++ b/google/genai/_interactions/types/interaction.py @@ -114,6 +114,12 @@ class AgentConfigFindRequest(BaseModel): validation on a single vulnerability. """ + model: Optional[str] = None + """The name of the model to use for the CodeMender agent. + + One CodeMender session will only use one model. + """ + session_config: Optional[AgentConfigFindRequestSessionConfig] = None """Optional session-specific configurations to override default agent behavior.""" @@ -182,6 +188,12 @@ class AgentConfigFixRequest(BaseModel): This ID maps to a previously discovered vulnerability. """ + model: Optional[str] = None + """The name of the model to use for the CodeMender agent. + + One CodeMender session will only use one model. + """ + session_config: Optional[AgentConfigFixRequestSessionConfig] = None """Optional session-specific configurations to override default agent behavior.""" diff --git a/google/genai/_interactions/types/interaction_create_params.py b/google/genai/_interactions/types/interaction_create_params.py index 12d35cc51..c82396355 100644 --- a/google/genai/_interactions/types/interaction_create_params.py +++ b/google/genai/_interactions/types/interaction_create_params.py @@ -273,6 +273,12 @@ class AgentConfigFindRequest(TypedDict, total=False): validation on a single vulnerability. """ + model: str + """The name of the model to use for the CodeMender agent. + + One CodeMender session will only use one model. + """ + session_config: AgentConfigFindRequestSessionConfig """Optional session-specific configurations to override default agent behavior.""" @@ -341,6 +347,12 @@ class AgentConfigFixRequest(TypedDict, total=False): This ID maps to a previously discovered vulnerability. """ + model: str + """The name of the model to use for the CodeMender agent. + + One CodeMender session will only use one model. + """ + session_config: AgentConfigFixRequestSessionConfig """Optional session-specific configurations to override default agent behavior."""