feat(memory): wire memory stores into run pipeline and add listing endpoint

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 23
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-2783b07289e9f71f07550497c5180824f3dbc92477b5d1d8a80fa6dce4e3b8b6.yml
-openapi_spec_hash: e1d4cff965beed1ec255bf387d55d940
-config_hash: 5a6e285f6e3a958a887b31b972a3f49c
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta/warp-api-8c84aaafca600b8a8f64e590997958a0b9dcb6172fdf3220dd243c20bd0ee3dc.yml
+openapi_spec_hash: 82d67b5080e55941b619875c222b94b8
+config_hash: 236823a4936c76818117c16aa5c188df

src/resources/agent/agent.ts

@@ -250,6 +250,11 @@ export interface AmbientAgentConfig {
    */
   mcp_servers?: { [key: string]: McpServerConfig };
 
+  /**
+   * Memory stores to attach to this run.
+   */
+  memory_stores?: Array<AmbientAgentConfig.MemoryStore>;
+
   /**
    * LLM model to use (uses team default if not specified)
    */
@@ -318,6 +323,26 @@ export namespace AmbientAgentConfig {
     claude_auth_secret_name?: string;
   }
 
+  /**
+   * Reference to a memory store to attach to an agent.
+   */
+  export interface MemoryStore {
+    /**
+     * Access level for the store.
+     */
+    access: 'read_write' | 'read_only';
+
+    /**
+     * Instructions for how the agent should use this memory store. Must not be empty.
+     */
+    instructions: string;
+
+    /**
+     * UID of the memory store.
+     */
+    uid: string;
+  }
+
   /**
    * Configures sharing behavior for the run's shared session. When set, the worker
    * emits `--share public:<level>` and the bundled Warp client applies an

src/resources/agent/schedules.ts

@@ -178,6 +178,11 @@ export interface ScheduledAgentItem {
    */
   agent_config?: AgentAPI.AmbientAgentConfig;
 
+  /**
+   * UID of the agent that this schedule runs as
+   */
+  agent_uid?: string;
+
   created_by?: AgentAPI.UserProfile;
 
   /**

tests/api-resources/agent/schedules.test.ts

@@ -45,13 +45,20 @@ describe('resource schedules', () => {
             warp_id: 'warp_id',
           },
         },
+        memory_stores: [
+          {
+            access: 'read_write',
+            instructions: 'instructions',
+            uid: 'uid',
+          },
+        ],
         model_id: 'model_id',
         name: 'name',
         session_sharing: { public_access: 'VIEWER' },
         skill_spec: 'skill_spec',
         worker_host: 'worker_host',
       },
-      agent_uid: 'agent_uid',
+      agent_uid: '182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e',
       enabled: true,
       mode: 'normal',
       prompt: 'Review open pull requests and provide feedback',
@@ -110,13 +117,20 @@ describe('resource schedules', () => {
             warp_id: 'warp_id',
           },
         },
+        memory_stores: [
+          {
+            access: 'read_write',
+            instructions: 'instructions',
+            uid: 'uid',
+          },
+        ],
         model_id: 'model_id',
         name: 'name',
         session_sharing: { public_access: 'VIEWER' },
         skill_spec: 'skill_spec',
         worker_host: 'worker_host',
       },
-      agent_uid: 'agent_uid',
+      agent_uid: '182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e',
       mode: 'normal',
       prompt: 'prompt',
     });