diff --git a/packages/api/src/endpoints/bedrock/initialize.spec.ts b/packages/api/src/endpoints/bedrock/initialize.spec.ts index 158650017e89..b890b3a262b5 100644 --- a/packages/api/src/endpoints/bedrock/initialize.spec.ts +++ b/packages/api/src/endpoints/bedrock/initialize.spec.ts @@ -123,6 +123,7 @@ describe('initializeBedrock', () => { guardrailIdentifier: 'test-guardrail-id', guardrailVersion: '1', trace: 'enabled' as const, + streamProcessingMode: 'async', }; const params = createMockParams({ diff --git a/packages/api/src/types/bedrock.ts b/packages/api/src/types/bedrock.ts index 8f34b2864d19..6608c52fcf59 100644 --- a/packages/api/src/types/bedrock.ts +++ b/packages/api/src/types/bedrock.ts @@ -19,6 +19,8 @@ export interface GuardrailConfiguration { guardrailVersion: string; /** The trace behavior for the guardrail */ trace?: 'enabled' | 'disabled' | 'enabled_full'; + /** The processing mode for guardrail; 'sync' is the default guardrail behavior if unset */ + streamProcessingMode?: 'sync' | 'async'; } /**