Skip to content

Commit c82dc15

Browse files
feat: Add per-agent AWS Bedrock OIDC inference role (backend)
1 parent bc889e1 commit c82dc15

6 files changed

Lines changed: 383 additions & 5 deletions

File tree

.stats.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 23
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-d662aff074fd108261bcfc38540e31dbbbd86c4e94a798ebe58714650c21cb87.yml
3-
openapi_spec_hash: 7065b44212d19f637200a4b5da97c2e2
4-
config_hash: 5a6e285f6e3a958a887b31b972a3f49c
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-b719dd35d7850ee303cdebf54fa3dfddb492a6f578344c2060cfae013b61541c.yml
3+
openapi_spec_hash: 4c21e0d940ef5fc42767be5380571c5d
4+
config_hash: 236823a4936c76818117c16aa5c188df

src/resources/agent/agent.ts

Lines changed: 57 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,11 +245,21 @@ export interface AmbientAgentConfig {
245245
*/
246246
idle_timeout_minutes?: number;
247247

248+
/**
249+
* Inference provider settings used for LLM calls.
250+
*/
251+
inference_providers?: AmbientAgentConfig.InferenceProviders;
252+
248253
/**
249254
* Map of MCP server configurations by name
250255
*/
251256
mcp_servers?: { [key: string]: McpServerConfig };
252257

258+
/**
259+
* Memory stores to attach to this run.
260+
*/
261+
memory_stores?: Array<AmbientAgentConfig.MemoryStore>;
262+
253263
/**
254264
* LLM model to use (uses team default if not specified)
255265
*/
@@ -318,6 +328,53 @@ export namespace AmbientAgentConfig {
318328
claude_auth_secret_name?: string;
319329
}
320330

331+
/**
332+
* Inference provider settings used for LLM calls.
333+
*/
334+
export interface InferenceProviders {
335+
/**
336+
* Configures AWS Bedrock as the LLM inference provider for this agent or run.
337+
*/
338+
aws?: InferenceProviders.Aws;
339+
}
340+
341+
export namespace InferenceProviders {
342+
/**
343+
* Configures AWS Bedrock as the LLM inference provider for this agent or run.
344+
*/
345+
export interface Aws {
346+
/**
347+
* If true, opt out of Bedrock at this layer.
348+
*/
349+
disabled?: boolean;
350+
351+
/**
352+
* IAM role ARN to assume when calling Bedrock.
353+
*/
354+
role_arn?: string;
355+
}
356+
}
357+
358+
/**
359+
* Reference to a memory store to attach to an agent.
360+
*/
361+
export interface MemoryStore {
362+
/**
363+
* Access level for the store.
364+
*/
365+
access: 'read_write' | 'read_only';
366+
367+
/**
368+
* Instructions for how the agent should use this memory store. Must not be empty.
369+
*/
370+
instructions: string;
371+
372+
/**
373+
* UID of the memory store.
374+
*/
375+
uid: string;
376+
}
377+
321378
/**
322379
* Configures sharing behavior for the run's shared session. When set, the worker
323380
* emits `--share public:<level>` and the bundled Warp client applies an

0 commit comments

Comments
 (0)