@obelyzk/sdk 1.4.0 → 1.6.0

package/dist/index.js

@@ -85,10 +85,10 @@ __export(index_exports, {
   createMiningClient: () => createMiningClient,
   createPaymentsClient: () => createPaymentsClient,
   createPrivacyClient: () => createPrivacyClient,
-  createProverClient: () => createProverClient,
+  createProverClient: () => createStwoProverClient,
   createStakingClient: () => createStakingClient,
   createStwoClient: () => createStwoClient,
-  createStwoProverClient: () => createStwoProverClient2,
+  createStwoProverClient: () => createStwoProverClient,
   createTeeClient: () => createTeeClient,
   createTransferProof: () => createTransferProof,
   createWebSocketClient: () => createWebSocketClient,
@@ -6418,6 +6418,53 @@ var StwoProverClient = class {
   /**
    * Generate proof for AI/ML inference
    */
+  /**
+   * Verifiable chat inference — send a prompt, get a response with on-chain proof.
+   *
+   * @example
+   * ```typescript
+   * const result = await prover.chat("What is 2+2?", { model: "local", maxTokens: 1 });
+   * console.log(result.text);           // model output
+   * console.log(result.txHash);         // Starknet TX hash
+   * console.log(result.explorerUrl);    // Starkscan link
+   * console.log(result.calldataFelts);  // ~950 felts
+   * console.log(result.modelId);        // weight commitment hash
+   * ```
+   */
+  async chat(prompt, options) {
+    const response = await fetch(`${this.config.baseUrl}/v1/chat/completions`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        ...this.config.apiKey ? { Authorization: `Bearer ${this.config.apiKey}` } : {}
+      },
+      body: JSON.stringify({
+        model: options?.model ?? "local",
+        messages: [{ role: "user", content: prompt }],
+        max_tokens: options?.maxTokens ?? 1,
+        stream: false,
+        session_id: options?.sessionId
+      })
+    });
+    if (!response.ok) {
+      const err = await response.text();
+      throw new Error(`Chat failed (${response.status}): ${err}`);
+    }
+    const data = await response.json();
+    const choice = data.choices?.[0];
+    const meta = data.obelyzk ?? {};
+    return {
+      text: choice?.message?.content ?? "",
+      txHash: meta.tx_hash,
+      proofHash: meta.proof_hash,
+      modelId: meta.model_id,
+      ioCommitment: meta.io_commitment,
+      calldataFelts: meta.calldata_felts,
+      explorerUrl: meta.explorer_url,
+      proveTimeSecs: meta.prove_time_secs,
+      trustModel: meta.trust_model ?? "unknown"
+    };
+  }
   async proveInference(modelId, inputs, outputs, options) {
     const job = await this.submitProofJob({
       proofType: "ai_inference",
@@ -6513,7 +6560,7 @@ var StwoProverClient = class {
     return new Promise((resolve) => setTimeout(resolve, ms));
   }
 };
-function createStwoProverClient2(config) {
+function createStwoProverClient(config) {
   return new StwoProverClient(config ?? {});
 }
 StwoProverClient.prototype.listZkmlModels = async function() {
@@ -6786,6 +6833,124 @@ var AgentFirewallSDK = class {
     );
     return contract.is_trusted(agentId);
   }
+  /** Get the decision for an action (0=pending, 1=approved, 2=escalated, 3=blocked). */
+  async getActionDecision(actionId) {
+    const contract = new import_starknet.Contract(
+      FIREWALL_ABI,
+      this.config.firewallContract,
+      this.provider
+    );
+    return Number(await contract.get_action_decision(actionId));
+  }
+  /** Get the threat score for a resolved action. */
+  async getActionThreatScore(actionId) {
+    const contract = new import_starknet.Contract(
+      FIREWALL_ABI,
+      this.config.firewallContract,
+      this.provider
+    );
+    return Number(await contract.get_action_threat_score(actionId));
+  }
+  /** Get the IO commitment for an action. */
+  async getActionIoCommitment(actionId) {
+    const contract = new import_starknet.Contract(
+      FIREWALL_ABI,
+      this.config.firewallContract,
+      this.provider
+    );
+    const result = await contract.get_action_io_commitment(actionId);
+    return `0x${BigInt(result).toString(16)}`;
+  }
+  // ── On-Chain Actions ────────────────────────────────────────────────
+  /**
+   * Submit a pending action to the firewall contract.
+   * Returns the action_id assigned by the contract.
+   */
+  async submitAction(agentId, target, value, selector, ioCommitment) {
+    this.requireAccount();
+    const tx = await this.config.account.execute({
+      contractAddress: this.config.firewallContract,
+      entrypoint: "submit_action",
+      calldata: import_starknet.CallData.compile({
+        agent_id: agentId,
+        target,
+        value,
+        selector,
+        io_commitment: ioCommitment
+      })
+    });
+    const receipt = await this.provider.waitForTransaction(tx.transaction_hash);
+    let actionId = 0;
+    const events = receipt.events;
+    if (Array.isArray(events)) {
+      for (const event of events) {
+        if (event.data && event.data.length >= 2) {
+          const candidateId = Number(BigInt(event.data[0]));
+          if (candidateId > 0) {
+            actionId = candidateId;
+            break;
+          }
+        }
+      }
+    }
+    return { actionId, txHash: tx.transaction_hash };
+  }
+  /**
+   * Resolve a pending action with a verified ZKML proof.
+   * The proof must already be verified on the ObelyskVerifier contract.
+   */
+  async resolveAction(actionId, proofHash, originalIoLen, packedRawIo) {
+    this.requireAccount();
+    const tx = await this.config.account.execute({
+      contractAddress: this.config.firewallContract,
+      entrypoint: "resolve_action_with_proof",
+      calldata: import_starknet.CallData.compile({
+        action_id: actionId,
+        proof_hash: proofHash,
+        original_io_len: originalIoLen,
+        packed_raw_io: packedRawIo
+      })
+    });
+    const receipt = await this.provider.waitForTransaction(tx.transaction_hash);
+    let decision = "approve";
+    let threatScore = 0;
+    const resolveEvents = receipt.events;
+    if (Array.isArray(resolveEvents)) {
+      for (const event of resolveEvents) {
+        if (event.data && event.data.length >= 4) {
+          const decisionCode = Number(BigInt(event.data[2]));
+          threatScore = Number(BigInt(event.data[3]));
+          if (decisionCode === 1) decision = "approve";
+          else if (decisionCode === 2) decision = "escalate";
+          else if (decisionCode === 3) decision = "block";
+          break;
+        }
+      }
+    }
+    return { decision, threatScore, txHash: tx.transaction_hash };
+  }
+  /** Approve an escalated action (human-in-the-loop). */
+  async approveEscalated(actionId) {
+    this.requireAccount();
+    const tx = await this.config.account.execute({
+      contractAddress: this.config.firewallContract,
+      entrypoint: "approve_escalated",
+      calldata: import_starknet.CallData.compile({ action_id: actionId })
+    });
+    await this.provider.waitForTransaction(tx.transaction_hash);
+    return tx.transaction_hash;
+  }
+  /** Reject an escalated action and add a strike. */
+  async rejectEscalated(actionId) {
+    this.requireAccount();
+    const tx = await this.config.account.execute({
+      contractAddress: this.config.firewallContract,
+      entrypoint: "reject_escalated",
+      calldata: import_starknet.CallData.compile({ action_id: actionId })
+    });
+    await this.provider.waitForTransaction(tx.transaction_hash);
+    return tx.transaction_hash;
+  }
   // ── Helpers ────────────────────────────────────────────────────────
   requireAccount() {
     if (!this.config.account) {
@@ -6844,6 +7009,69 @@ var FIREWALL_ABI = [
     inputs: [{ name: "action_id", type: "felt" }],
     outputs: [{ type: "felt" }],
     state_mutability: "view"
+  },
+  {
+    name: "get_action_threat_score",
+    type: "function",
+    inputs: [{ name: "action_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_action_io_commitment",
+    type: "function",
+    inputs: [{ name: "action_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_owner",
+    type: "function",
+    inputs: [],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_verifier",
+    type: "function",
+    inputs: [],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_classifier_model_id",
+    type: "function",
+    inputs: [],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_thresholds",
+    type: "function",
+    inputs: [],
+    outputs: [{ type: "felt" }, { type: "felt" }, { type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_agent_owner",
+    type: "function",
+    inputs: [{ name: "agent_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "get_action_agent",
+    type: "function",
+    inputs: [{ name: "action_id", type: "felt" }],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
+  },
+  {
+    name: "is_paused",
+    type: "function",
+    inputs: [],
+    outputs: [{ type: "felt" }],
+    state_mutability: "view"
   }
 ];
 
@@ -7584,9 +7812,6 @@ var ConfidentialSwapClient = class {
     };
   }
 };
-
-// src/index.ts
-var createProverClient = createStwoProverClient;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AgentFirewallSDK,

package/dist/index.mjs

@@ -1,6 +1,7 @@
+import "./chunk-DGYMDV5X.mjs";
 import {
   AgentFirewallSDK
-} from "./chunk-BHUXKNDT.mjs";
+} from "./chunk-Y4PBMUWM.mjs";
 import {
   BatchClient,
   BitSageClient,
@@ -87,8 +88,8 @@ import {
   powMod,
   randomScalar,
   subMod
-} from "./chunk-MDF4P52S.mjs";
-import "./chunk-Y6FXYEAI.mjs";
+} from "./chunk-GK4FKSZ4.mjs";
+import "./chunk-XGB3TDIC.mjs";
 
 // src/types.ts
 function getMinStake(tier) {
@@ -499,6 +500,53 @@ var StwoProverClient = class {
   /**
    * Generate proof for AI/ML inference
    */
+  /**
+   * Verifiable chat inference — send a prompt, get a response with on-chain proof.
+   *
+   * @example
+   * ```typescript
+   * const result = await prover.chat("What is 2+2?", { model: "local", maxTokens: 1 });
+   * console.log(result.text);           // model output
+   * console.log(result.txHash);         // Starknet TX hash
+   * console.log(result.explorerUrl);    // Starkscan link
+   * console.log(result.calldataFelts);  // ~950 felts
+   * console.log(result.modelId);        // weight commitment hash
+   * ```
+   */
+  async chat(prompt, options) {
+    const response = await fetch(`${this.config.baseUrl}/v1/chat/completions`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        ...this.config.apiKey ? { Authorization: `Bearer ${this.config.apiKey}` } : {}
+      },
+      body: JSON.stringify({
+        model: options?.model ?? "local",
+        messages: [{ role: "user", content: prompt }],
+        max_tokens: options?.maxTokens ?? 1,
+        stream: false,
+        session_id: options?.sessionId
+      })
+    });
+    if (!response.ok) {
+      const err = await response.text();
+      throw new Error(`Chat failed (${response.status}): ${err}`);
+    }
+    const data = await response.json();
+    const choice = data.choices?.[0];
+    const meta = data.obelyzk ?? {};
+    return {
+      text: choice?.message?.content ?? "",
+      txHash: meta.tx_hash,
+      proofHash: meta.proof_hash,
+      modelId: meta.model_id,
+      ioCommitment: meta.io_commitment,
+      calldataFelts: meta.calldata_felts,
+      explorerUrl: meta.explorer_url,
+      proveTimeSecs: meta.prove_time_secs,
+      trustModel: meta.trust_model ?? "unknown"
+    };
+  }
   async proveInference(modelId, inputs, outputs, options) {
     const job = await this.submitProofJob({
       proofType: "ai_inference",
@@ -594,7 +642,7 @@ var StwoProverClient = class {
     return new Promise((resolve) => setTimeout(resolve, ms));
   }
 };
-function createStwoProverClient2(config) {
+function createStwoProverClient(config) {
   return new StwoProverClient(config ?? {});
 }
 StwoProverClient.prototype.listZkmlModels = async function() {
@@ -792,9 +840,6 @@ function getVersionBanner(version) {
 \u255A\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u255D
 `;
 }
-
-// src/index.ts
-var createProverClient = createStwoProverClient;
 export {
   AgentFirewallSDK,
   AssetId,
@@ -861,10 +906,10 @@ export {
   createMiningClient,
   createPaymentsClient,
   createPrivacyClient,
-  createProverClient,
+  createStwoProverClient as createProverClient,
   createStakingClient,
   createStwoClient,
-  createStwoProverClient2 as createStwoProverClient,
+  createStwoProverClient,
   createTeeClient,
   createTransferProof,
   createWebSocketClient,

package/dist/mcp-policy/index.d.mts

@@ -0,0 +1 @@
+#!/usr/bin/env node

package/dist/mcp-policy/index.d.ts

@@ -0,0 +1 @@
+#!/usr/bin/env node