Skip to content

feat: [modelarmor] add streaming methods StreamSanitizeUserPrompt and StreamSanitizeModelResponse #8483

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
gcf-owl-bot wants to merge 2 commits into
base: main
Choose a base branch
from
Open

feat: [modelarmor] add streaming methods StreamSanitizeUserPrompt and StreamSanitizeModelResponse #8483

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
cbd6dbe
feat: add streaming methods StreamSanitizeUserPrompt and StreamSaniti…
gcf-owl-bot[bot] Jun 9, 2026
d5c49e9
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jun 9, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions packages/google-cloud-modelarmor/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,8 @@ Samples are in the [`samples/`][homepage_samples] directory. Each sample's `READ
| list templates | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1/model_armor.list_templates.js) |
| sanitize model response | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1/model_armor.sanitize_model_response.js) |
| sanitize user prompt | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1/model_armor.sanitize_user_prompt.js) |
| stream sanitize model response | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1/model_armor.stream_sanitize_model_response.js) |
| stream sanitize user prompt | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1/model_armor.stream_sanitize_user_prompt.js) |
| update floor setting | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1/model_armor.update_floor_setting.js) |
| update template | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1/model_armor.update_template.js) |
| create template | [source code](https://github.com/googleapis/google-cloud-node/blob/main/packages/google-cloud-modelarmor/samples/generated/v1beta/model_armor.create_template.js) |
Expand Down
35 changes: 34 additions & 1 deletion packages/google-cloud-modelarmor/protos/google/cloud/modelarmor/v1/service.proto
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
// Copyright 2025 Google LLC
// Copyright 2026 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
Expand All @@ -24,6 +24,7 @@ import "google/api/resource.proto";
import "google/protobuf/empty.proto";
import "google/protobuf/field_mask.proto";
import "google/protobuf/timestamp.proto";
import "google/type/date.proto";

option csharp_namespace = "Google.Cloud.ModelArmor.V1";
option go_package = "cloud.google.com/go/modelarmor/apiv1/modelarmorpb;modelarmorpb";
Expand Down Expand Up @@ -128,6 +129,14 @@ service ModelArmor {
body: "*"
};
}

// Streaming version of Sanitize User Prompt.
rpc StreamSanitizeUserPrompt(stream SanitizeUserPromptRequest)
returns (stream SanitizeUserPromptResponse) {}

// Streaming version of Sanitizes Model Response.
rpc StreamSanitizeModelResponse(stream SanitizeModelResponseRequest)
returns (stream SanitizeModelResponseResponse) {}
}

// Option to specify filter match state.
Expand Down Expand Up @@ -228,6 +237,18 @@ enum InvocationResult {
FAILURE = 3;
}

// Streaming Mode for Sanitize* API.
enum StreamingMode {
// Default value.
STREAMING_MODE_UNSPECIFIED = 0;

// Buffered Streaming mode.
STREAMING_MODE_BUFFERED = 1;

// Real Time Streaming mode.
STREAMING_MODE_REALTIME = 2;
}

// Message describing Template resource
message Template {
option (google.api.resource) = {
Expand Down Expand Up @@ -745,6 +766,10 @@ message SanitizeUserPromptRequest {
// Optional. Metadata related to Multi Language Detection.
MultiLanguageDetectionMetadata multi_language_detection_metadata = 6
[(google.api.field_behavior) = OPTIONAL];

// Optional. Streaming Mode for StreamSanitize* API.
optional StreamingMode streaming_mode = 7
[(google.api.field_behavior) = OPTIONAL];
}

// Sanitize Model Response request.
Expand All @@ -767,6 +792,10 @@ message SanitizeModelResponseRequest {
// Optional. Metadata related for multi language detection.
MultiLanguageDetectionMetadata multi_language_detection_metadata = 7
[(google.api.field_behavior) = OPTIONAL];

// Optional. Streaming Mode for StreamSanitize* API.
optional StreamingMode streaming_mode = 8
[(google.api.field_behavior) = OPTIONAL];
}

// Sanitized User Prompt Response.
Expand Down Expand Up @@ -796,6 +825,10 @@ message SanitizationResult {
// Passthrough field defined in TemplateMetadata to indicate whether to
// ignore partial invocation failures.
bool ignore_partial_invocation_failures = 3;

// Output only. The stream chunk processed by the Sanitization service.
DataItem stream_chunk_processed = 4
[(google.api.field_behavior) = OUTPUT_ONLY];
}

// Output only. Overall filter match state for Sanitization.
Expand Down
180 changes: 180 additions & 0 deletions packages/google-cloud-modelarmor/protos/protos.d.ts
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading
Loading