Skip to content

Add Model Armor config support to Python Gen-AI library #1870

@tanmha

Description

@tanmha

The Vertex AI REST API defines a ModelArmorConfig object with promptTemplateName and responseTemplateName fields to apply Model Armor templates during inference.

Please expose an equivalent ModelArmorConfig class in the Python Gen-AI SDK and allow it as an optional argument in model invocation calls.

Metadata

Metadata

Assignees

Labels

api: gemini-apipriority: p3Desirable enhancement or fix. May not be included in next release.type: feature request‘Nice-to-have’ improvement, new feature or different behavior or design.

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions