feat: Update public API and graphql to support creating multiple service accounts and API keys

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 15
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-11fd464a4cf5b849a565d00039093093ef93acdb99679ae94b2908543a55097b.yml
-openapi_spec_hash: 12840099351c5309d6cf8c4b92f028b3
-config_hash: 42666caf45e82675e80982d51e3b7727
+configured_endpoints: 19
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/warp-bnavetta%2Fwarp-api-3226a0810ae5de980509e31910138990d93d2843918730249caed1436e677ec3.yml
+openapi_spec_hash: 776a9087e0c342e9fcbd140e75936d75
+config_hash: c5fc921cc04f541a85f92299f365eba6

api.md

@@ -60,6 +60,22 @@ Methods:
 - <code title="post /agent/schedules/{scheduleId}/pause">client.agent.schedules.<a href="./src/resources/agent/schedules.ts">pause</a>(scheduleID) -> ScheduledAgentItem</code>
 - <code title="post /agent/schedules/{scheduleId}/resume">client.agent.schedules.<a href="./src/resources/agent/schedules.ts">resume</a>(scheduleID) -> ScheduledAgentItem</code>
 
+## Agent
+
+Types:
+
+- <code><a href="./src/resources/agent/agent_.ts">AgentResponse</a></code>
+- <code><a href="./src/resources/agent/agent_.ts">CreateAgentRequest</a></code>
+- <code><a href="./src/resources/agent/agent_.ts">ListAgentIdentitiesResponse</a></code>
+- <code><a href="./src/resources/agent/agent_.ts">UpdateAgentRequest</a></code>
+
+Methods:
+
+- <code title="post /agent/identities">client.agent.agent.<a href="./src/resources/agent/agent_.ts">create</a>({ ...params }) -> AgentResponse</code>
+- <code title="put /agent/identities/{uid}">client.agent.agent.<a href="./src/resources/agent/agent_.ts">update</a>(uid, { ...params }) -> AgentResponse</code>
+- <code title="get /agent/identities">client.agent.agent.<a href="./src/resources/agent/agent_.ts">list</a>() -> ListAgentIdentitiesResponse</code>
+- <code title="delete /agent/identities/{uid}">client.agent.agent.<a href="./src/resources/agent/agent_.ts">delete</a>(uid) -> void</code>
+
 ## Sessions
 
 Types:

src/resources/agent/agent.ts

@@ -2,6 +2,16 @@
 
 import { APIResource } from '../../core/resource';
 import * as AgentAPI from './agent';
+import * as AgentAgentAPI from './agent_';
+import {
+  Agent as AgentAPIAgent,
+  AgentCreateParams,
+  AgentResponse,
+  AgentUpdateParams,
+  CreateAgentRequest,
+  ListAgentIdentitiesResponse,
+  UpdateAgentRequest,
+} from './agent_';
 import * as RunsAPI from './runs';
 import {
   ArtifactItem,
@@ -35,6 +45,7 @@ import { path } from '../../internal/utils/path';
 export class Agent extends APIResource {
   runs: RunsAPI.Runs = new RunsAPI.Runs(this._client);
   schedules: SchedulesAPI.Schedules = new SchedulesAPI.Schedules(this._client);
+  agent: AgentAgentAPI.Agent = new AgentAgentAPI.Agent(this._client);
   sessions: SessionsAPI.Sessions = new SessionsAPI.Sessions(this._client);
 
   /**
@@ -1019,6 +1030,7 @@ export namespace AgentRunParams {
 
 Agent.Runs = Runs;
 Agent.Schedules = Schedules;
+Agent.Agent = AgentAPIAgent;
 Agent.Sessions = Sessions;
 
 export declare namespace Agent {
@@ -1064,5 +1076,15 @@ export declare namespace Agent {
     type ScheduleUpdateParams as ScheduleUpdateParams,
   };
 
+  export {
+    AgentAPIAgent as Agent,
+    type AgentResponse as AgentResponse,
+    type CreateAgentRequest as CreateAgentRequest,
+    type ListAgentIdentitiesResponse as ListAgentIdentitiesResponse,
+    type UpdateAgentRequest as UpdateAgentRequest,
+    type AgentCreateParams as AgentCreateParams,
+    type AgentUpdateParams as AgentUpdateParams,
+  };
+
   export { Sessions as Sessions, type SessionCheckRedirectResponse as SessionCheckRedirectResponse };
 }

src/resources/agent/agent_.ts

@@ -0,0 +1,135 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../../core/resource';
+import { APIPromise } from '../../core/api-promise';
+import { buildHeaders } from '../../internal/headers';
+import { RequestOptions } from '../../internal/request-options';
+import { path } from '../../internal/utils/path';
+
+/**
+ * Operations for running and managing cloud agents
+ */
+export class Agent extends APIResource {
+  /**
+   * Create a new agent for the caller's team. Agents can be used as the execution
+   * principal for team-owned runs.
+   *
+   * @example
+   * ```ts
+   * const agentResponse = await client.agent.agent.create({
+   *   name: 'name',
+   * });
+   * ```
+   */
+  create(body: AgentCreateParams, options?: RequestOptions): APIPromise<AgentResponse> {
+    return this._client.post('/agent/identities', { body, ...options });
+  }
+
+  /**
+   * Update an existing agent.
+   *
+   * @example
+   * ```ts
+   * const agentResponse = await client.agent.agent.update(
+   *   'uid',
+   * );
+   * ```
+   */
+  update(uid: string, body: AgentUpdateParams, options?: RequestOptions): APIPromise<AgentResponse> {
+    return this._client.put(path`/agent/identities/${uid}`, { body, ...options });
+  }
+
+  /**
+   * List all agents for the caller's team. Each agent includes an `available` flag
+   * indicating whether it is within the team's plan limit and may be used for runs.
+   *
+   * @example
+   * ```ts
+   * const listAgentIdentitiesResponse =
+   *   await client.agent.agent.list();
+   * ```
+   */
+  list(options?: RequestOptions): APIPromise<ListAgentIdentitiesResponse> {
+    return this._client.get('/agent/identities', options);
+  }
+
+  /**
+   * Delete an agent. All API keys associated with the agent are deleted atomically.
+   *
+   * @example
+   * ```ts
+   * await client.agent.agent.delete('uid');
+   * ```
+   */
+  delete(uid: string, options?: RequestOptions): APIPromise<void> {
+    return this._client.delete(path`/agent/identities/${uid}`, {
+      ...options,
+      headers: buildHeaders([{ Accept: '*/*' }, options?.headers]),
+    });
+  }
+}
+
+export interface AgentResponse {
+  /**
+   * Whether this agent is within the team's plan limit and can be used for runs
+   */
+  available: boolean;
+
+  /**
+   * When the agent was created (RFC3339)
+   */
+  created_at: string;
+
+  /**
+   * Name of the agent
+   */
+  name: string;
+
+  /**
+   * Unique identifier for the agent
+   */
+  uid: string;
+}
+
+export interface CreateAgentRequest {
+  /**
+   * A name for the agent
+   */
+  name: string;
+}
+
+export interface ListAgentIdentitiesResponse {
+  agents: Array<AgentResponse>;
+}
+
+export interface UpdateAgentRequest {
+  /**
+   * The new name for the agent
+   */
+  name?: string;
+}
+
+export interface AgentCreateParams {
+  /**
+   * A name for the agent
+   */
+  name: string;
+}
+
+export interface AgentUpdateParams {
+  /**
+   * The new name for the agent
+   */
+  name?: string;
+}
+
+export declare namespace Agent {
+  export {
+    type AgentResponse as AgentResponse,
+    type CreateAgentRequest as CreateAgentRequest,
+    type ListAgentIdentitiesResponse as ListAgentIdentitiesResponse,
+    type UpdateAgentRequest as UpdateAgentRequest,
+    type AgentCreateParams as AgentCreateParams,
+    type AgentUpdateParams as AgentUpdateParams,
+  };
+}

src/resources/agent/index.ts

@@ -41,3 +41,11 @@ export {
   type ScheduleUpdateParams,
 } from './schedules';
 export { Sessions, type SessionCheckRedirectResponse } from './sessions';
+export {
+  type AgentResponse,
+  type CreateAgentRequest,
+  type ListAgentIdentitiesResponse,
+  type UpdateAgentRequest,
+  type AgentCreateParams,
+  type AgentUpdateParams,
+} from './agent_';

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

@@ -0,0 +1,63 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import OzAPI from 'oz-agent-sdk';
+
+const client = new OzAPI({
+  apiKey: 'My API Key',
+  baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010',
+});
+
+describe('resource agent', () => {
+  // Mock server tests are disabled
+  test.skip('create: only required params', async () => {
+    const responsePromise = client.agent.agent.create({ name: 'name' });
+    const rawResponse = await responsePromise.asResponse();
+    expect(rawResponse).toBeInstanceOf(Response);
+    const response = await responsePromise;
+    expect(response).not.toBeInstanceOf(Response);
+    const dataAndResponse = await responsePromise.withResponse();
+    expect(dataAndResponse.data).toBe(response);
+    expect(dataAndResponse.response).toBe(rawResponse);
+  });
+
+  // Mock server tests are disabled
+  test.skip('create: required and optional params', async () => {
+    const response = await client.agent.agent.create({ name: 'name' });
+  });
+
+  // Mock server tests are disabled
+  test.skip('update', async () => {
+    const responsePromise = client.agent.agent.update('uid', {});
+    const rawResponse = await responsePromise.asResponse();
+    expect(rawResponse).toBeInstanceOf(Response);
+    const response = await responsePromise;
+    expect(response).not.toBeInstanceOf(Response);
+    const dataAndResponse = await responsePromise.withResponse();
+    expect(dataAndResponse.data).toBe(response);
+    expect(dataAndResponse.response).toBe(rawResponse);
+  });
+
+  // Mock server tests are disabled
+  test.skip('list', async () => {
+    const responsePromise = client.agent.agent.list();
+    const rawResponse = await responsePromise.asResponse();
+    expect(rawResponse).toBeInstanceOf(Response);
+    const response = await responsePromise;
+    expect(response).not.toBeInstanceOf(Response);
+    const dataAndResponse = await responsePromise.withResponse();
+    expect(dataAndResponse.data).toBe(response);
+    expect(dataAndResponse.response).toBe(rawResponse);
+  });
+
+  // Mock server tests are disabled
+  test.skip('delete', async () => {
+    const responsePromise = client.agent.agent.delete('uid');
+    const rawResponse = await responsePromise.asResponse();
+    expect(rawResponse).toBeInstanceOf(Response);
+    const response = await responsePromise;
+    expect(response).not.toBeInstanceOf(Response);
+    const dataAndResponse = await responsePromise.withResponse();
+    expect(dataAndResponse.data).toBe(response);
+    expect(dataAndResponse.response).toBe(rawResponse);
+  });
+});