@buildaureon/mcp 0.1.1 → 0.1.7

package/dist/index.js

@@ -1,7 +1,7 @@
 #!/usr/bin/env node
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import { DEFAULT_API_BASE_URL, createSessionTokenProvider, createAureonClient, isAureonError } from '@buildaureon/sdk';
+import { DEFAULT_API_BASE_URL, createSessionTokenProvider, createAureonClient, validateExecutionReceipt, isAureonError } from '@buildaureon/sdk';
 import { z } from 'zod';
 
 function loadConfig() {
@@ -68,6 +68,18 @@ function registerReadTools(server, client) {
       }
     }
   );
+  server.tool(
+    "aureon_get_allocation_vs_target",
+    "Objective vs actual portfolio \u2014 current weight vs policy target per objective, plus green-book/off-plan paradox flag",
+    {},
+    async () => {
+      try {
+        return ok(await client.getAllocationVsTarget());
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
   server.tool(
     "aureon_get_portfolio",
     "Current capital book snapshot \u2014 positions, marks, weights",
@@ -151,7 +163,7 @@ function registerCompassTools(server, client) {
   );
   server.tool(
     "aureon_run_execution",
-    "Run restorative execution for an objective currently outside policy",
+    "Run restorative execution for an objective outside policy. Read receipt.settlement and receipt.verifiedOnChain \u2014 staged = never on-chain; vault + verifiedOnChain = chain-observed; vault without verified = submitted but not yet observed.",
     { objectiveId: z.string().describe("The objective ID") },
     async ({ objectiveId }) => {
       try {
@@ -163,7 +175,7 @@ function registerCompassTools(server, client) {
   );
   server.tool(
     "aureon_restore_objective",
-    "Run vault-backed restorative execution for an objective outside policy",
+    "Run restorative execution for an objective outside policy. Not always vault-backed: read settlement and verifiedOnChain. staged is never on-chain. vault without verifiedOnChain is submitted, not proven. Returns ExecutionReceipt.",
     { objectiveId: z.string().describe("The objective ID") },
     async ({ objectiveId }) => {
       try {
@@ -175,7 +187,7 @@ function registerCompassTools(server, client) {
   );
   server.tool(
     "aureon_list_executions",
-    "Recent execution receipts \u2014 past restorative actions taken",
+    "Recent execution receipts \u2014 read settlement, verifiedOnChain, settlementRecord, explorerUrl, registryRef.",
     { objectiveId: z.string().optional().describe("Filter by objective ID (omit for all)") },
     async ({ objectiveId }) => {
       try {
@@ -327,6 +339,176 @@ var objectiveKind = z.enum([
   "reward_reinvestment"
 ]);
 var objectivePriority = z.enum(["low", "medium", "high", "critical"]);
+function registerIntentTools(server, client) {
+  server.tool(
+    "aureon_apply_financial_intent",
+    "Register user/agent intent as an Automatic objective and return AI \u2192 objective \u2192 portfolio flow",
+    {
+      brief: z.string().describe("What the user wants their money to do \u2014 agent-extracted wording"),
+      kind: objectiveKind.describe("Objective kind"),
+      targetWeight: z.number().describe("Target portfolio weight (0\u20131)"),
+      tolerance: z.number().describe("Allowed drift band (0\u20131)"),
+      targetSymbol: z.string().optional().describe("Asset symbol for balanced_portfolio (required for that kind)"),
+      name: z.string().optional().describe("Objective display name override"),
+      priority: objectivePriority.optional().describe("Evaluation priority")
+    },
+    async (input) => {
+      try {
+        return ok(await client.applyFinancialIntent(input));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_get_objective_portfolio_flow",
+    "Read AI \u2192 objective \u2192 portfolio flow for active objectives",
+    {
+      objectiveId: z.string().optional().describe("Filter to one objective (omit for all active)")
+    },
+    async ({ objectiveId }) => {
+      try {
+        return ok(await client.getObjectivePortfolioFlow(objectiveId));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
+function registerDriftRestoreTools(server, client) {
+  server.tool(
+    "aureon_run_drift_restore_demo",
+    "Run drift \u2192 detection \u2192 restore demo (seed book, break rule, manual restore)",
+    {},
+    async () => {
+      try {
+        return ok(await client.runDriftRestoreDemo());
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_get_drift_restore_flow",
+    "Read drift \u2192 detection \u2192 restore flow for active objectives",
+    {
+      objectiveId: z.string().optional().describe("Filter to one objective (omit for all active)")
+    },
+    async ({ objectiveId }) => {
+      try {
+        return ok(await client.getDriftRestoreFlow(objectiveId));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
+function registerReceiptVerificationTools(server, client) {
+  server.tool(
+    "aureon_run_receipt_verification_demo",
+    "Run receipt \u2192 verification demo (restore + validate receipt + settlement lookup)",
+    {},
+    async () => {
+      try {
+        return ok(await client.runReceiptVerificationDemo());
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_get_receipt_verification_flow",
+    "Read receipt \u2192 verification flow for execution receipts",
+    {
+      executionId: z.string().optional().describe("Filter to one execution id (omit for five most recent)")
+    },
+    async ({ executionId }) => {
+      try {
+        return ok(await client.getReceiptVerificationFlow(executionId));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
+var agentHost = z.enum(["cursor", "claude", "mcp"]);
+function registerPortfolioWatchTools(server, client) {
+  server.tool(
+    "aureon_run_portfolio_watch_demo",
+    "Run portfolio watch demo (brief \u2192 Automatic objective \u2192 while-away event \u2192 return briefing)",
+    {
+      brief: z.string().optional().describe(
+        "User wording \u2014 default: Watch my portfolio while I'm away \u2014 keep about 20% in stable assets."
+      ),
+      host: agentHost.optional().describe("Agent host label for briefing (cursor | claude | mcp)")
+    },
+    async ({ brief, host }) => {
+      try {
+        return ok(await client.runPortfolioWatchDemo({ brief, host }));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_get_portfolio_watch_flow",
+    "Read portfolio watch briefing for Automatic objectives",
+    {
+      objectiveId: z.string().optional().describe("Filter to one objective (omit for all Automatic)"),
+      brief: z.string().optional().describe("User brief for briefing lines"),
+      host: agentHost.optional().describe("Agent host label")
+    },
+    async ({ objectiveId, brief, host }) => {
+      try {
+        return ok(
+          await client.getPortfolioWatchFlow({ objectiveId, brief, host })
+        );
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
+function registerFullAureonLoopTools(server, client) {
+  server.tool(
+    "aureon_run_full_aureon_loop_demo",
+    "Run Content Arc full AUREON loop (intent \u2192 plan check \u2192 restore \u2192 receipt verification)",
+    {
+      brief: z.string().optional().describe(
+        "User wording \u2014 default: Keep about 20% in stable assets \u2014 grow the book without abandoning the plan."
+      )
+    },
+    async ({ brief }) => {
+      try {
+        return ok(await client.runFullAureonLoopDemo({ brief }));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_get_full_aureon_loop_flow",
+    "Read full AUREON loop for active objectives with a latest execution receipt",
+    {
+      objectiveId: z.string().optional().describe("Filter to one objective (omit for all active with receipts)"),
+      brief: z.string().optional().describe("User brief for teaching shape")
+    },
+    async ({ objectiveId, brief }) => {
+      try {
+        return ok(await client.getFullAureonLoopFlow({ objectiveId, brief }));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
+var objectiveKind2 = z.enum([
+  "stable_allocation",
+  "balanced_portfolio",
+  "risk_ceiling",
+  "reward_reinvestment"
+]);
+var objectivePriority2 = z.enum(["low", "medium", "high", "critical"]);
 var automationMode = z.enum(["auto", "manual"]);
 function registerObjectiveTools(server, client) {
   server.tool(
@@ -334,10 +516,10 @@ function registerObjectiveTools(server, client) {
     "Create a Financial Compass Objective. targetSymbol and automationMode are locked after create \u2014 recreate to change them. Defaults to automationMode auto.",
     {
       name: z.string().describe("Objective display name"),
-      kind: objectiveKind.describe("Objective kind"),
+      kind: objectiveKind2.describe("Objective kind"),
       targetWeight: z.number().describe("Target portfolio weight (0\u20131)"),
       tolerance: z.number().describe("Allowed drift band (0\u20131)"),
-      priority: objectivePriority.optional().describe("Evaluation priority"),
+      priority: objectivePriority2.optional().describe("Evaluation priority"),
       maxRiskScore: z.number().optional().describe("Max risk score ceiling"),
       reinvestRatio: z.number().optional().describe("Reinvest ratio for reward objectives"),
       targetSymbol: z.string().nullable().optional().describe("Asset symbol to track (locked after create)"),
@@ -357,7 +539,7 @@ function registerObjectiveTools(server, client) {
     {
       objectiveId: z.string().describe("Objective ID"),
       name: z.string().optional(),
-      priority: objectivePriority.optional(),
+      priority: objectivePriority2.optional(),
       targetWeight: z.number().optional(),
       tolerance: z.number().optional(),
       maxRiskScore: z.number().optional(),
@@ -452,7 +634,7 @@ function registerMarketTools(server, client) {
       priceChangeRatio: z.number().describe("Fractional price change (e.g. -0.15 = -15%)"),
       name: z.string().optional().describe("Event label"),
       description: z.string().optional().describe("Event description"),
-      autoRestore: z.boolean().optional().describe("Run restorative execution if objectives breach")
+      autoRestore: z.boolean().optional().describe("Opt in to run restore if objectives breach. Omit or false = drift only. Automatic still 409s if the vault cannot execute.")
     },
     async (input) => {
       try {
@@ -520,6 +702,137 @@ function registerDeveloperTools(server, client) {
     }
   );
 }
+function registerRegistryTools(server, client) {
+  server.tool(
+    "aureon_registry_status",
+    "Returns Phase 2 ObjectiveRegistry deployment status (testnet contract address when configured).",
+    {},
+    async () => {
+      try {
+        return ok(await client.getRegistryStatus());
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_get_objective_registry",
+    "Returns the on-chain registry record for an objective when registered.",
+    {
+      objectiveId: z.string().describe("Objective id")
+    },
+    async ({ objectiveId }) => {
+      try {
+        return ok(await client.getObjectiveRegistry(objectiveId));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_prepare_objective_registry",
+    "Prepares wallet-signed calldata to register an objective on ObjectiveRegistry. The host wallet must broadcast the tx, then call confirm via SDK.",
+    {
+      objectiveId: z.string().describe("Objective id to register on-chain")
+    },
+    async ({ objectiveId }) => {
+      try {
+        return ok(await client.prepareObjectiveRegistry(objectiveId));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_confirm_objective_registry",
+    "Confirms an on-chain ObjectiveRegistry registration after the host wallet broadcast the prepare tx. Pass the transaction hash. The API verifies the receipt on chain and stores the registry link.",
+    {
+      objectiveId: z.string().describe("Objective id that was registered"),
+      transactionHash: z.string().describe("0x transaction hash from the wallet broadcast")
+    },
+    async ({ objectiveId, transactionHash }) => {
+      try {
+        return ok(await client.confirmObjectiveRegistry(objectiveId, transactionHash));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
+function registerSettlementTools(server, client) {
+  server.tool(
+    "aureon_get_execution_settlement",
+    "Returns the chain-verified settlement record for a vault execution when the listener has observed an on-chain Rebalanced event. Staged executions never have settlement records \u2014 do not claim on-chain proof without this.",
+    {
+      executionId: z.string().describe("Execution receipt id")
+    },
+    async ({ executionId }) => {
+      try {
+        return ok(await client.getExecutionSettlement(executionId));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_list_settlements",
+    "Lists settlement records for the authenticated wallet. May include orphans (observed vault Rebalanced events not yet bound to an execution). Optional objectiveId filter. Read status and executionId before claiming proof.",
+    {
+      objectiveId: z.string().optional().describe("Filter by objective id")
+    },
+    async ({ objectiveId }) => {
+      try {
+        return ok({ settlements: await client.listSettlements(objectiveId) });
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+  server.tool(
+    "aureon_confirm_execution_settlement",
+    "Manual backfill: verify a vault Rebalanced transaction on chain and attach a settlement record to an execution. Use when the listener missed the event. Staged executions cannot be confirmed this way.",
+    {
+      executionId: z.string().describe("Execution receipt id"),
+      transactionHash: z.string().describe("0x vault Rebalanced transaction hash")
+    },
+    async ({ executionId, transactionHash }) => {
+      try {
+        return ok(await client.confirmExecutionSettlement(executionId, transactionHash));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
+function registerReceiptTools(server) {
+  server.tool(
+    "aureon_validate_receipt",
+    "Validates an execution receipt locally (schema + settlement honesty). No API call. Returns { valid, issues }. Do not trust or summarize a receipt as on-chain when validation fails.",
+    {
+      receipt: z.record(z.unknown()).describe("ExecutionReceipt JSON from restore, listExecutions, or API")
+    },
+    async ({ receipt }) => {
+      const result = validateExecutionReceipt(receipt);
+      return ok(result);
+    }
+  );
+}
+function registerAuditTrailTools(server, client) {
+  server.tool(
+    "aureon_get_audit_trail",
+    "Export the financial audit trail for one objective: registry, receipts, settlements, timeline. Missing proof is labeled. Never invent on-chain claims.",
+    {
+      objectiveId: z.string().describe("Objective id to export")
+    },
+    async ({ objectiveId }) => {
+      try {
+        return ok(await client.getAuditTrail(objectiveId));
+      } catch (err) {
+        return fail(err);
+      }
+    }
+  );
+}
 
 // src/tools/index.ts
 function registerTools(server, client, session) {
@@ -528,9 +841,18 @@ function registerTools(server, client, session) {
   registerVaultTools(server, client);
   registerAuthTools(server, client, session);
   registerObjectiveTools(server, client);
+  registerIntentTools(server, client);
+  registerDriftRestoreTools(server, client);
+  registerReceiptVerificationTools(server, client);
+  registerPortfolioWatchTools(server, client);
+  registerFullAureonLoopTools(server, client);
   registerPortfolioTools(server, client);
   registerMarketTools(server, client);
   registerDeveloperTools(server, client);
+  registerRegistryTools(server, client);
+  registerSettlementTools(server, client);
+  registerReceiptTools(server);
+  registerAuditTrailTools(server, client);
 }
 
 // src/server.ts