@agent-os-sdk/client 0.4.3 → 0.4.4

package/src/modules/agents.ts

@@ -11,7 +11,11 @@ import type { RawClient, APIResponse, components } from "../client/raw.js";
 import type { PaginationParams, PaginatedResponse } from "../client/helpers.js";
 
 // Type aliases for this module
+
 type AgentSchema = components["schemas"]["AgentExportAgent"];
+// Bundle types from schema
+type AgentBundleSchema = components["schemas"]["AgentExportBundle"];
+
 // Response types
 export interface Agent {
     id: string;
@@ -21,57 +25,12 @@ export interface Agent {
     tenant_id: string;
     created_at: string;
     updated_at: string;
-    live_version_id?: string | null;
-}
-
-export interface AgentVersion {
-    id: string;
-    agent_id: string;
-    version_tag: string;
-    artifact_uri?: string;
-    artifact_hash?: string;
-    format?: string;
-    graph_spec?: unknown;
-    created_at: string;
-    draft_state?: string | null;
-    base_version_id?: string | null;
-}
-
-export interface CreateDraftRequest {
-    /** Base type: "live" (current live version) or "version" (specific version) */
-    base?: "live" | "version";
-    /** Required when base="version" */
-    base_version_id?: string;
-}
-
-export interface CreateDraftResponse {
-    draft_version_id: string;
-    agent_id: string;
-    base_version_id?: string | null;
-    created_at: string;
+    live_bundle_id?: string | null;
 }
 
-export interface AgentBundle {
-    schema_version: string;
-    exported_at: string;
-    agent?: AgentBundleAgent;
-    versions?: AgentBundleVersion[];
-}
 
-export interface AgentBundleAgent {
-    id: string;
-    name: string;
-    live_version_id?: string | null;
-}
 
-export interface AgentBundleVersion {
-    id: string;
-    version_tag: string;
-    artifact_uri: string;
-    artifact_hash?: string;
-    format?: string;
-    created_at: string;
-}
+export interface AgentBundle extends AgentBundleSchema { }
 
 export interface ImportAgentResponse {
     agent_id: string;
@@ -79,30 +38,12 @@ export interface ImportAgentResponse {
     versions_imported: number;
 }
 
-export interface AgentVersionMetadata {
-    id: string;
+export interface ImportAgentResponse {
     agent_id: string;
-    version_tag: string;
-    created_at: string;
-}
-
-export interface AgentVersionBundleResponse {
-    version: AgentVersionMetadata;
-    graph_spec: unknown;
-    artifact_hash: string;
-    format: string;
-    resolved_at: string;
-}
-
-export interface UpdateAgentVersionRequest {
-    graph_spec?: unknown;
+    message: string;
+    versions_imported: number;
 }
 
-export interface PublishAgentVersionResponse {
-    agent_id: string;
-    live_version_id: string;
-    status: string;
-}
 
 export interface AgentGraphResponse {
     mermaid?: string;
@@ -116,7 +57,6 @@ export interface GraphValidationResponse {
 }
 
 export type AgentListResponse = PaginatedResponse<Agent>;
-export type AgentVersionListResponse = PaginatedResponse<AgentVersion>;
 
 export class AgentsModule {
     constructor(private client: RawClient, private headers: () => Record<string, string>) { }
@@ -181,7 +121,7 @@ export class AgentsModule {
     async update(agentId: string, body: {
         name?: string;
         description?: string;
-        live_version_id?: string;
+        live_bundle_id?: string;
     }): Promise<APIResponse<Agent>> {
         return this.client.PUT<Agent>("/v1/api/agents/{agentId}", {
             params: { path: { agentId } },
@@ -200,102 +140,6 @@ export class AgentsModule {
         });
     }
 
-    // ======================== Versions ========================
-
-    /**
-     * List all versions of an agent.
-     * @example
-     * ```ts
-     * const { data } = await client.agents.listVersions("agent-uuid");
-     * ```
-     */
-    async listVersions(agentId: string): Promise<APIResponse<AgentVersion[]>> {
-        return this.client.GET<AgentVersion[]>("/v1/api/agents/{agentId}/versions", {
-            params: { path: { agentId } },
-            headers: this.headers(),
-        });
-    }
-
-    /** Alias for listVersions (convenience) */
-    versions = (agentId: string) => this.listVersions(agentId);
-
-    /**
-     * Create a new version of an agent.
-     */
-    async createVersion(agentId: string, body: {
-        version_tag: string;
-        artifact_uri?: string;
-        artifact_hash?: string;
-        format?: string;
-        graph_spec?: unknown;
-    }): Promise<APIResponse<AgentVersion>> {
-        return this.client.POST<AgentVersion>("/v1/api/agents/{agentId}/versions", {
-            params: { path: { agentId } },
-            body,
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Get a specific version of an agent.
-     */
-    async getVersion(agentId: string, versionId: string): Promise<APIResponse<AgentVersion>> {
-        return this.client.GET<AgentVersion>("/v1/api/agents/{agentId}/versions/{versionId}", {
-            params: { path: { agentId, versionId } },
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Create or retrieve an active draft version for editing.
-     * This endpoint is idempotent - calling it multiple times returns the same draft.
-     * @example
-     * ```ts
-     * const { data } = await client.agents.createDraft("agent-uuid", { base: "live" });
-     * console.log(data?.draft_version_id);
-     * // Then call getBundle(draftVersionId) to load the graph
-     * ```
-     */
-    async createDraft(agentId: string, body?: CreateDraftRequest): Promise<APIResponse<CreateDraftResponse>> {
-        return this.client.POST<CreateDraftResponse>("/v1/api/agents/{agentId}/versions/draft", {
-            params: { path: { agentId } },
-            body: body ?? { base: "live" },
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Get the full bundle for a version (including resolved graph_spec).
-     */
-    async getBundle(agentId: string, versionId: string): Promise<APIResponse<AgentVersionBundleResponse>> {
-        return this.client.GET<AgentVersionBundleResponse>("/v1/api/agents/{agentId}/versions/{versionId}/bundle", {
-            params: { path: { agentId, versionId } },
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Update a draft version.
-     * Only allowed if draft_state="active".
-     */
-    async updateVersion(agentId: string, versionId: string, body: UpdateAgentVersionRequest): Promise<APIResponse<AgentVersion>> {
-        return this.client.PATCH<AgentVersion>("/v1/api/agents/{agentId}/versions/{versionId}", {
-            params: { path: { agentId, versionId } },
-            body,
-            headers: this.headers(),
-        });
-    }
-
-    /**
-     * Publish a draft version to live.
-     */
-    async publishVersion(agentId: string, versionId: string): Promise<APIResponse<PublishAgentVersionResponse>> {
-        return this.client.POST<PublishAgentVersionResponse>("/v1/api/agents/{agentId}/versions/{versionId}/publish", {
-            params: { path: { agentId, versionId } },
-            headers: this.headers(),
-        });
-    }
-
     // ======================== Portability ========================
 
     /**