Plans and Goals

import ComposeSDK from "@compose-market/sdk";

const sdk = new ComposeSDK({
  userAddress: "0xYourWalletAddress",
  chainId: 43113,
});

const agentWallet = "0xa7abfd271130c3ee5c8f8862a123f3697e75af0d";
const threadId = "build-app";

const planStream = sdk.agent.stream({
  agentWallet,
  message: "/plan Build a todo app: set up the project, add CRUD endpoints, and write tests",
  threadId,
});

for await (const event of planStream) {
  if (event.type === "plan.proposed") {
    console.log(event.markdown);
    console.log(`State: ${event.state}`);
  }
}

const runStream = sdk.agent.stream({
  agentWallet,
  message: "/plan run",
  threadId,
});

for await (const event of runStream) {
  switch (event.type) {
    case "task.completed":
      console.log(`Done: ${event.taskId}`);
      break;
    case "task.blocked":
      console.log(`Blocked: ${event.taskId}`);
      break;
    case "task.failed":
      console.log(`Failed: ${event.taskId}`);
      break;
    case "action":
      console.log(`Plan status: ${event.status}`);
      break;
    case "done":
      console.log(`\nTokens: ${event.usage.total_tokens}`);
      break;
  }
}

type PlanState = {
  id: string;
  threadId: string;
  status: PlanStatus;
  tasks: TaskRecord[];
  goal: string;
  createdAt: number;
  updatedAt: number;
};

type PlanStatus =
  | "draft"
  | "proposed"
  | "awaiting_approval"
  | "running"
  | "checking"
  | "blocked"
  | "completed"
  | "failed"
  | "cancelled";

type TaskRecord = {
  id: string;
  title: string;
  description: string;
  status: "todo" | "doing" | "blocked" | "done" | "failed" | "cancelled";
  result?: string;
  error?: string;
  order: number;
};

await sdk.agent.stream({
  agentWallet,
  message: "/goal Get the todo app to production-ready quality",
  threadId,
});

await sdk.agent.stream({
  agentWallet,
  message: "Add input validation to the create endpoint",
  threadId,
});

await sdk.agent.stream({
  agentWallet,
  message: "Now add error handling and edge-case tests for all endpoints",
  threadId,
});

const statusStream = sdk.agent.stream({
  agentWallet,
  message: "/goal status",
  threadId,
});

for await (const event of statusStream) {
  if (event.type === "action" && event.action === "goal") {
    console.log(event.result.goal);
  }
}

type GoalState = {
  id: string;
  threadId: string;
  objective: string;
  status: "active" | "paused" | "completed" | "cleared";
  progress?: string;
  createdAt: number;
  updatedAt: number;
};