@agent-os-sdk/client 0.4.4 → 0.4.6

package/src/modules/agents.ts

@@ -58,6 +58,10 @@ export interface GraphValidationResponse {
 
 export type AgentListResponse = PaginatedResponse<Agent>;
 
+// Draft types
+export type AgentDraftResponse = components["schemas"]["AgentDraftResponse"];
+export type UpdateDraftGraphRequest = components["schemas"]["UpdateDraftGraphRequest"];
+
 export class AgentsModule {
     constructor(private client: RawClient, private headers: () => Record<string, string>) { }
 
@@ -187,5 +191,37 @@ export class AgentsModule {
         });
     }
 
+    /**
+     * Update the draft graph for an agent (autosave).
+     * @param agentId The agent UUID
+     * @param graph_json The graph specification
+     * @param expected_ticks The expected ticks for optimistic concurrency (from If-Match)
+     */
+    async updateDraftGraph(
+        agentId: string,
+        graph_json: Record<string, unknown>,
+        expected_ticks: number | null
+    ): Promise<APIResponse<components["schemas"]["AgentDraftResponse"]>> {
+        // Construct If-Match header 
+        // If expected_ticks is null, we don't send header (or send something else? Logic says If-Match required if strictly following docs)
+        // But docs say: "Requires If-Match header with ETag from previous GET/PUT".
+        // If client passes 0, we send "agentId:0".
+        // If client passes null, maybe it's a force overwrite or initial save? UseCase handles logic.
+        // We will strictly set If-Match if expected_ticks is number.
+
+        const headers = this.headers();
+        if (expected_ticks !== null && expected_ticks !== undefined) {
+            headers["If-Match"] = `"${agentId}:${expected_ticks}"`;
+        }
+
+        return this.client.PUT<components["schemas"]["AgentDraftResponse"]>("/v1/api/agents/{id}/draft/graph", {
+            params: { path: { id: agentId } },
+            body: {
+                graph_json
+            } as components["schemas"]["UpdateDraftGraphRequest"],
+            headers
+        });
+    }
+
 
 }