@palliora.org/chainsdk 0.1.0 → 0.2.0

package/dist/index.js

@@ -220,12 +220,144 @@ var API_TYPES = {
     extra: "CheckAppIdExtra",
     types: "CheckAppIdTypes"
   },
+  FeePayload: {
+    compute: "u128",
+    guardian: "u128",
+    verifier: "u128"
+  },
+  SilentThresholdParams: {
+    td_params: "Vec<u8>",
+    pk_bytes: "Vec<u8>",
+    tau_params: "Vec<u8>"
+  },
+  ThresholdAlgos: {
+    _enum: {
+      SilentThreshold: "SilentThresholdParams"
+    }
+  },
+  ChaCha20Poly1305Params: {
+    nonce: "[u8; 12]"
+  },
+  Aes256GcmParams: {
+    nonce: "[u8; 12]"
+  },
+  SymmetricAlgos: {
+    _enum: {
+      ChaCha20Poly1305: "ChaCha20Poly1305Params",
+      Aes256Gcm: "Aes256GcmParams"
+    }
+  },
+  CipherSuiteEncrypted: {
+    threshold: "ThresholdAlgos",
+    symmetric: "SymmetricAlgos"
+  },
+  CipherSuite: {
+    _enum: {
+      Plaintext: "Null",
+      Encrypted: "CipherSuiteEncrypted"
+    }
+  },
+  ConfidentialityLevel: {
+    _enum: ["Trusted", "TEE", "FHE", "SMPC"]
+  },
+  NativeExecuteDA: {
+    _enum: ["Inference"]
+  },
+  NativeDataDA: {
+    _enum: ["DaFalse", "DaTrue"]
+  },
+  DAInputInline: {
+    data: "Vec<u8>"
+  },
+  DAInputChainTransaction: {
+    block_number: "u64",
+    extrinsic_index: "u32"
+  },
+  DAInputIpfs: {
+    cid: "Vec<u8>",
+    size: "u64"
+  },
+  DAInputUrl: {
+    url: "Vec<u8>",
+    size: "u64",
+    hash: "Option<Vec<u8>>"
+  },
+  DAInput: {
+    _enum: {
+      Inline: "DAInputInline",
+      ChainTransaction: "DAInputChainTransaction",
+      Ipfs: "DAInputIpfs",
+      Url: "DAInputUrl",
+      NativeExecute: "NativeExecuteDA",
+      NativeData: "NativeDataDA"
+    }
+  },
+  ContractType: {
+    _enum: {
+      Dormant: "Dormant",
+      Active: "Active"
+    }
+  },
+  StoreType: {
+    _enum: {
+      Dataset: "Dataset",
+      Model: "Model",
+      Agent: "Agent",
+      Other: "Other"
+    }
+  },
+  ComputeMetadata: {
+    name: "Vec<u8>",
+    description: "Vec<u8>",
+    store_type: "StoreType",
+    group_id: "H256"
+  },
+  ComputeInfo: {
+    cipher: "CipherSuite",
+    computer_indices: "Vec<u32>",
+    fees: "u128",
+    deadline: "u64",
+    confidentiality: "ConfidentialityLevel",
+    fee_function: "Option<u8>",
+    program_env: "Option<Vec<u8>>",
+    input: "DAInput",
+    program: "DAInput",
+    metadata: "Option<ComputeMetadata>"
+  },
+  Contract: {
+    contract_type: "ContractType",
+    guardians: "Vec<AccountId>",
+    pre_check: "Option<ComputeInfo>",
+    compute: "ComputeInfo",
+    post_check: "Option<ComputeInfo>",
+    result_cipher: "CipherSuite"
+  },
+  AgreementInfo: {
+    status: "AgreementStatus",
+    creator: "AccountId",
+    index: "u32"
+  },
   ComputePayload: {
     da_type: "u8",
     agreement: "Option<BoundedVec<[u8; 32], 10>>",
     verification: "u8",
     compute: "u8"
   },
+  ComputePrefs: {
+    trusted: "bool",
+    tee: "bool",
+    mpc: "bool",
+    fhe: "bool",
+    zkp: "bool"
+  },
+  GuardianPrefs: {
+    pubKey: "[u8; 32]",
+    guardian: "bool",
+    verifier: "bool",
+    compute: "bool",
+    computePrefs: "Option<ComputePrefs>",
+    feeThreshold: "u128"
+  },
   BlockLengthColumns: "Compact<u32>",
   BlockLengthRows: "Compact<u32>",
   BlockLength: {
@@ -599,10 +731,6 @@ function configure(opts) {
 }
 
 // src/utils/helper.ts
-var tokenToBigint = (arg) => {
-  const val = typeof arg === "bigint" ? arg : BigInt(arg);
-  return val * 10n ** 15n;
-};
 function hexToUint8Array(hex) {
   if (hex.length % 2 !== 0) {
     throw new Error("Invalid hex string");
@@ -653,8 +781,23 @@ var debugLog = (message, ...optionalParams) => {
 };
 
 // src/utils/token.ts
-import { formatBalance } from "@polkadot/util";
+import { formatUnits, parseUnits } from "viem";
+var PALI_SYMBOL = "PALI";
+var PALI_DECIMALS = 18;
 var tokenCache = null;
+var toAtomicPaliAmount = (amount) => {
+  const normalizedAmount = String(amount).trim();
+  if (!normalizedAmount) {
+    throw new Error("Token amount cannot be empty");
+  }
+  return parseUnits(normalizedAmount, PALI_DECIMALS);
+};
+var fromAtomicPaliAmount = (amount) => {
+  return formatUnits(amount, PALI_DECIMALS);
+};
+var formatPaliAmount = (amount, symbol = PALI_SYMBOL) => {
+  return `${fromAtomicPaliAmount(amount)} ${symbol}`;
+};
 async function fetchTokenProperties() {
   if (tokenCache) {
     return tokenCache;
@@ -663,13 +806,12 @@ async function fetchTokenProperties() {
     const systemProperties = (await (await getApi())?.rpc.system.properties())?.toHuman();
     assert(systemProperties, "Failed to fetch system properties from RPC");
     const tokenProperties = {
-      symbol: (systemProperties?.tokenSymbol || ["UNIT"])[0],
-      decimals: Number((systemProperties?.tokenDecimals || ["18"])[0])
+      symbol: (systemProperties?.tokenSymbol || [PALI_SYMBOL])[0],
+      decimals: Number((systemProperties?.tokenDecimals || [String(PALI_DECIMALS)])[0])
     };
     tokenCache = tokenProperties;
     return tokenProperties;
   } catch (error) {
-    console.error("Failed to fetch token properties:", error);
     throw error;
   }
 }
@@ -681,20 +823,12 @@ function clearTokenCache() {
 }
 async function formatBalanceWithTokenProperties(balance) {
   const tokenProperties = await fetchTokenProperties();
-  if (!tokenProperties) {
-    throw new Error(
-      "Token properties not yet cached. Call fetchTokenProperties first."
-    );
-  }
-  return formatBalance(balance, {
-    decimals: tokenProperties.decimals,
-    withSi: true,
-    withUnit: tokenProperties.symbol
-  });
+  return `${formatUnits(typeof balance === "bigint" ? balance : BigInt(balance), tokenProperties.decimals)} ${tokenProperties.symbol}`;
 }
+var tokenToBigint = toAtomicPaliAmount;
 
 // src/chain/singleton.ts
-import { ApiPromise, Keyring as Keyring2, WsProvider as WsProvider2 } from "@polkadot/api";
+import { ApiPromise, Keyring as Keyring2 } from "@polkadot/api";
 
 // src/chain/wsProvider.ts
 import { WsProvider } from "@polkadot/api";
@@ -702,128 +836,13 @@ var provider = new WsProvider(PALLIORA_WS, 1e4);
 
 // src/chain/singleton.ts
 import { waitReady as waitReady2 } from "@polkadot/wasm-crypto";
-var wsProvider = null;
 var api = null;
 var keyring = null;
 var encKeyring = null;
-var RpcApi = class {
-  constructor() {
-    this.isConnected = false;
-    this.isConnecting = false;
-    this.error = null;
-  }
-  /**
-   * Establishes a connection to a node at `endpoint` and returns the singleton
-   * API and Keyring instances.
-   *
-   * @param endpoint - WebSocket endpoint URL to connect to (e.g. "wss://...").
-   * @returns A promise resolving to an object containing the singleton {@link ApiPromise}
-   *          instance and the singleton {@link Keyring} instance.
-   *
-   * @remarks
-   * - If a connection already exists (`isConnected === true`) the method returns
-   *   the already-initialized instances without recreating them.
-   * - The method sets `isConnecting` to true while establishing the connection and
-   *   clears it in a `finally` block.
-   * - Provider and API event listeners update the instance `error` and `isConnected`
-   *   fields on runtime errors and disconnects.
-   * - The Keyring created here uses `sr25519` keys. If the standalone `getKeyring`
-   *   helper is used elsewhere, it will additionally add a default development
-   *   account derived from the well-known dev URI `//Bob` (named "Bob default").
-   *
-   * @throws Will re-throw underlying errors encountered while creating the provider
-   *         or API. In that case `error` will contain the textual error message.
-   */
-  async connect(endpoint) {
-    try {
-      if (this.isConnected) {
-        return { api, keyring };
-      }
-      this.isConnecting = true;
-      this.error = null;
-      if (!wsProvider) {
-        wsProvider = new WsProvider2(endpoint);
-        wsProvider.on("error", (err) => {
-          console.error("WsProvider error:", err);
-          this.error = "WebSocket connection error";
-          this.isConnected = false;
-        });
-        wsProvider.on("disconnected", () => {
-          console.log("WsProvider disconnected");
-          this.isConnected = false;
-        });
-      }
-      if (!api) {
-        api = await ApiPromise.create({
-          provider: wsProvider,
-          rpc: API_RPC,
-          types: API_TYPES,
-          signedExtensions: API_EXTENSIONS
-        });
-        api.on("error", (err) => {
-          console.error("API error:", err);
-          this.error = "API error occurred";
-        });
-        api.on("disconnected", () => {
-          this.isConnected = false;
-        });
-      }
-      if (!keyring) {
-        keyring = new Keyring2({ type: "sr25519" });
-      }
-      if (!encKeyring) {
-        encKeyring = new Keyring2({ type: "ed25519" });
-      }
-      await api.isReady;
-      this.isConnected = true;
-      return { api, keyring };
-    } catch (err) {
-      console.error("Failed to connect to Polkadot:", err);
-      this.error = err instanceof Error ? err.message : "Failed to connect to Polkadot";
-      throw err;
-    } finally {
-      this.isConnecting = false;
-    }
-  }
-  /**
-   * Gracefully disconnects and nullifies the singleton API, provider and keyring
-   * instances managed by this RpcApi instance.
-   *
-   * @remarks
-   * - The method calls `api.disconnect()` and `wsProvider.disconnect()` if they
-   *   exist, then sets the internal singletons to `null` and `isConnected` to false.
-   * - Any error during disconnect is captured in `error`.
-   */
-  async disconnect() {
-    try {
-      if (api) {
-        await api.disconnect();
-        api = null;
-      }
-      if (wsProvider) {
-        await wsProvider.disconnect();
-        wsProvider = null;
-      }
-      keyring = null;
-      this.isConnected = false;
-    } catch (err) {
-      console.error("Error disconnecting:", err);
-      this.error = err instanceof Error ? err.message : "Failed to disconnect";
-    }
-  }
-  getApi() {
-    return api;
-  }
-  getKeyring() {
-    return keyring;
-  }
-  getEncKeyring() {
-    return encKeyring;
-  }
-};
-async function getApi() {
+var apiListenersAttached = false;
+var apiTeardownInProgress = false;
+async function getApi(cb) {
   if (!provider) return;
-  console.debug(provider.endpoint);
   if (!api) {
     api = await ApiPromise.create({
       provider,
@@ -834,14 +853,24 @@ async function getApi() {
   }
   const isNode = typeof process !== "undefined" && typeof process.exit === "function";
   const isTest = typeof process !== "undefined" && process.env?.NODE_ENV === "test";
-  if (isNode && !isTest) {
+  if (isNode && !isTest && api && !apiListenersAttached) {
+    apiListenersAttached = true;
     api.on("error", (err) => {
-      console.error("api error, will restart:", err);
-      process.exit(0);
+      if (apiTeardownInProgress) return;
+      apiTeardownInProgress = true;
+      apiListenersAttached = false;
+      const currentApi = api;
+      api = null;
+      void currentApi?.disconnect().catch((disconnectErr) => {
+        apiTeardownInProgress = false;
+        console.error("api disconnect after error failed:", disconnectErr);
+      });
     });
     api.on("disconnected", () => {
-      console.error("api disconnected, will restart.");
-      process.exit(0);
+      api = null;
+      apiListenersAttached = false;
+      apiTeardownInProgress = false;
+      if (cb) cb();
     });
   }
   return api;
@@ -862,13 +891,6 @@ async function getEncKeyring() {
   }
   return encKeyring;
 }
-var apiInstance = null;
-function getRpcApi() {
-  if (!apiInstance) {
-    apiInstance = new RpcApi();
-  }
-  return apiInstance;
-}
 
 // src/chain/utils.ts
 import { isFunction as isFunction2 } from "@polkadot/util";
@@ -919,6 +941,7 @@ async function joinGuardian(account, prefs) {
   assert(api2, "API not initialized");
   assert(account, "Account not initialized");
   const computeOpts = prefs.compute?.split(",").map((s) => s.trim()).filter((s) => s.length > 0) || void 0;
+  const feeThreshold = typeof prefs.fee === "bigint" ? prefs.fee : BigInt(prefs.fee || "0");
   const guardianPrefs = {
     pubKey: account.publicKey,
     guardian: prefs.standard,
@@ -930,12 +953,13 @@ async function joinGuardian(account, prefs) {
       mpc: computeOpts?.includes("mpc") || false,
       fhe: computeOpts?.includes("fhe") || false,
       zkp: computeOpts?.includes("zkp") || false
-    }
+    },
+    feeThreshold
   };
   debugLog(
     account.address,
     "joining as guardian with preferences:",
-    guardianPrefs
+    { ...guardianPrefs, feeThreshold: formatPaliAmount(feeThreshold) }
   );
   const guardTx = api2.tx.staking.guard(guardianPrefs);
   const hash = await signAndSend(guardTx, account);
@@ -1237,18 +1261,29 @@ var FileFromMetadataRef = async (mcryptApi, metadataRef) => {
 
 // src/compute/agreement.ts
 import bs582 from "bs58";
-async function createAgreement() {
+async function createAgreement(contract, account) {
   const api2 = await getApi();
   if (!api2) throw new Error("Api not initialized");
-  const tx = api2.tx.compute.agreement();
-  const guardians = await getGuardianList();
-  const agreement = guardians.slice(0, 3).map((g) => bs582.decode(g).subarray(6));
-  assert(agreement.length === 3, "Not enough guardians to create agreement");
-  const keyring2 = await getKeyring();
-  const account = keyring2.getPairs()[0];
-  const { tx_result } = await signAndSend(tx, account, {
-    compute: { da_type: 1, agreement, verification: 0, compute: 1 }
-  });
+  const guardians = contract.guardians;
+  const agreement = guardians.map((g) => bs582.decode(g.peerid).subarray(6));
+  const txContract = {
+    ...contract,
+    guardians: guardians.map((g) => g.address)
+  };
+  const tx = api2.tx.compute.agreement(txContract);
+  const opts = {
+    compute: {
+      da_type: 1,
+      agreement,
+      verification: 0,
+      compute: contract.contract_type === "Active" ? 1 : 0
+    }
+  };
+  const { tx_result, blockNumber, index, hash } = await signAndSend(
+    tx,
+    account,
+    opts
+  );
   if (!tx_result.isError) {
     const agreementCreatedEvent = tx_result.events.find((event) => {
       return event.event.section === "compute" && event.event.method === "AgreementCreated";
@@ -1258,10 +1293,105 @@ async function createAgreement() {
         "Agreement data:",
         agreementCreatedEvent.event.data.toString()
       );
+      return {
+        blockNumber,
+        index,
+        hash,
+        agreementId: agreementCreatedEvent.event.data[0]?.toHex?.() ?? agreementCreatedEvent.event.data.toString()
+      };
     } else {
       debugLog("AgreementCreated event not found");
     }
   }
+  return { blockNumber, index, hash };
+}
+async function createSimpleAgreement() {
+  const guardianIds = (await getGuardianAddress()).slice(0, 3);
+  assert(guardianIds.length === 3, "Not enough guardians to create agreement");
+  const contract = {
+    contract_type: "Dormant",
+    guardians: guardianIds,
+    pre_check: null,
+    compute: {
+      cipher: "Plaintext",
+      computer_indices: [0, 1, 2],
+      fees: 0n,
+      deadline: 0,
+      confidentiality: { Trusted: { trust_index: 0 } },
+      fee_function: null,
+      input: null,
+      program: { NativeData: "DaFalse" },
+      metadata: null
+    },
+    post_check: null,
+    result_cipher: "Plaintext"
+  };
+  const keyring2 = await getKeyring();
+  const account = keyring2.getPairs()[0];
+  return createAgreement(contract, account);
+}
+
+// src/compute/data.ts
+async function dataContract(params) {
+  const keyring2 = await getKeyring();
+  const account = keyring2.getPairs()[0];
+  const plaintextCipher = "Plaintext";
+  const atomicFees = toAtomicPaliAmount(params.fees ?? "0");
+  const computeStep = {
+    cipher: plaintextCipher,
+    computer_indices: params.guardians.map((_, i) => i),
+    fees: atomicFees,
+    deadline: params.deadline ?? 0,
+    confidentiality: {
+      Trusted: {
+        trust_index: params.trustIndex ?? 0
+      }
+    },
+    fee_function: null,
+    input: {
+      Url: {
+        url: Array.from(new TextEncoder().encode(params.url))
+      }
+    },
+    program: {
+      NativeData: "DaFalse"
+    }
+  };
+  const contract = {
+    contract_type: "Dormant",
+    guardians: params.guardians,
+    pre_check: null,
+    compute: computeStep,
+    post_check: null,
+    result_cipher: plaintextCipher
+  };
+  return createAgreement(contract, account);
+}
+
+// src/compute/inference.ts
+async function inferenceCompute(params) {
+  const keyring2 = await getKeyring();
+  const account = keyring2.getPairs()[0];
+  const inputData = typeof params.input === "string" ? Array.from(new TextEncoder().encode(params.input)) : Array.from(params.input);
+  const atomicFees = toAtomicPaliAmount(params.fees ?? "0");
+  const plaintextCipher = "Plaintext";
+  const computeStep = {
+    cipher: plaintextCipher,
+    computer_indices: params.guardians.map((_, i) => i),
+    fees: atomicFees,
+    deadline: params.deadline ?? 0,
+    confidentiality: { Trusted: { trust_index: 0 } },
+    fee_function: null,
+    input: { Inline: { data: inputData } },
+    program: { NativeExecute: "Inference" }
+  };
+  const contract = {
+    contract_type: "Active",
+    guardians: params.guardians,
+    compute: computeStep,
+    result_cipher: plaintextCipher
+  };
+  return createAgreement(contract, account);
 }
 
 // src/compute/participants.ts
@@ -1327,6 +1457,46 @@ async function getGuardianParticipants() {
   }
 }
 
+// src/compute/simple.ts
+async function simpleCompute(params) {
+  const keyring2 = await getKeyring();
+  const account = keyring2.getPairs()[0];
+  const plaintextCipher = "Plaintext";
+  const atomicFees = toAtomicPaliAmount(params.fees ?? "0");
+  const computeStep = {
+    cipher: plaintextCipher,
+    computer_indices: params.guardians.map((_, i) => i),
+    fees: atomicFees,
+    deadline: params.deadline ?? 0,
+    confidentiality: {
+      Trusted: {
+        trust_index: params.trustIndex ?? 0
+      }
+    },
+    fee_function: null,
+    input: {
+      ChainTransaction: {
+        block_number: params.inputBlockNumber,
+        extrinsic_index: params.inputExtrinsicIndex
+      }
+    },
+    program: {
+      Url: {
+        url: Array.from(new TextEncoder().encode(params.programUrl))
+      }
+    }
+  };
+  const contract = {
+    contract_type: "Active",
+    guardians: params.guardians,
+    pre_check: null,
+    compute: computeStep,
+    post_check: null,
+    result_cipher: plaintextCipher
+  };
+  return createAgreement(contract, account);
+}
+
 // src/crypto/random.ts
 function generateRandomBytes(length = 32) {
   const bytes = new Uint8Array(length);
@@ -1349,6 +1519,7 @@ async function writeMetadata(account, name, description, ref, price, dataType, l
   const ownerBytes = Array.from(encoder.encode(l2Owner));
   const api2 = await getApi();
   assert(api2, "Failed to get API connection");
+  debugLog(`Registering metadata for ${name} at ${formatPaliAmount(price)}`);
   const request = api2.tx.dataAvailability.daccRegisterData(
     nameBytes,
     descriptionBytes,
@@ -1362,6 +1533,82 @@ async function writeMetadata(account, name, description, ref, price, dataType, l
   debugLog(`Metadata registration transaction sent with hash: ${hash.hash}`);
   return hash;
 }
+async function registerDataAgreement(account, params) {
+  debugLog(
+    `Registering data agreement for DA ref ${params.ref.blockNumber}-${params.ref.index} at ${formatPaliAmount(params.fees)}`
+  );
+  const cipher = params.cipher ?? "Plaintext";
+  const resultCipher = params.resultCipher ?? "Plaintext";
+  const encoder = new TextEncoder();
+  const computeMetadata = params.metadata ? {
+    name: Array.from(encoder.encode(params.metadata.name)),
+    description: Array.from(encoder.encode(params.metadata.description)),
+    store_type: params.metadata.storeType,
+    group_id: params.metadata.groupId
+  } : null;
+  const computeStep = {
+    cipher,
+    computer_indices: params.guardians.map((_, i) => i),
+    fees: params.fees,
+    deadline: params.deadline ?? 0,
+    confidentiality: {
+      Trusted: {
+        trust_index: params.trustIndex ?? 0
+      }
+    },
+    fee_function: null,
+    input: {
+      ChainTransaction: {
+        block_number: params.ref.blockNumber,
+        extrinsic_index: params.ref.index
+      }
+    },
+    program: {
+      NativeData: "DaFalse"
+    },
+    metadata: computeMetadata
+  };
+  const contract = {
+    contract_type: "Dormant",
+    guardians: params.guardians,
+    pre_check: null,
+    compute: computeStep,
+    post_check: null,
+    result_cipher: resultCipher
+  };
+  return createAgreement(contract, account);
+}
+
+// src/da/runAgent.ts
+async function runAgent(account, agentRef, nonce, ciphertext, tdParams, pkBytes, tauParams, baseModel, publicKey, guardians, guardian, agreementId) {
+  const agentRefTuple = [agentRef.blockNumber, agentRef.index];
+  const baseModelTuple = [baseModel.blockNumber, baseModel.index];
+  const api2 = await getApi();
+  assert(api2, "Failed to get API connection");
+  const request = api2.tx.dataAvailability.daccRunAgent(
+    agentRefTuple,
+    Array.from(nonce),
+    Array.from(ciphertext),
+    guardian ? Array.from(guardian) : null,
+    Array.from(tdParams),
+    Array.from(pkBytes),
+    Array.from(tauParams),
+    baseModelTuple,
+    Array.from(publicKey),
+    guardians.map((g) => Array.from(g))
+  );
+  const opts = {
+    compute: {
+      da_type: 4,
+      verification: 0,
+      compute: 1,
+      agreement: [agreementId]
+    }
+  };
+  const hash = await signAndSend(request, account, opts);
+  debugLog(`Run agent transaction sent with hash: ${hash.hash}`);
+  return hash;
+}
 
 // src/da/submit.ts
 async function submitData(account, data) {
@@ -1395,11 +1642,76 @@ async function submitTEData(account, data, chosenGuardians, tau_params, agg_key,
   debugLog(`TE data availability transaction sent with hash: ${hash.hash}`);
   return hash;
 }
+async function submitTEDataWithCipher(account, data, chosenGuardians, tau_params, agg_key, group_pk) {
+  const { encoded: encryptedKey, ikm } = testCrypt(tau_params, agg_key);
+  const shared_key = gen_stretched_key(hexToUint8Array(ikm));
+  const encoder = new TextEncoder();
+  const dataUint8Array = encoder.encode(data);
+  const { ciphertext, nonce } = encrypt2(dataUint8Array, shared_key);
+  const ciphertextHex = "0x" + Array.from(ciphertext).map((b) => b.toString(16).padStart(2, "0")).join("");
+  const api2 = await getApi();
+  assert(api2, "Failed to get API connection");
+  const request = await api2.tx.dataAvailability.submitData(ciphertextHex);
+  const ref = await signAndSend(request, account, DEFAULT_EMPTY_PAYLOAD);
+  debugLog(`TE data availability transaction sent with hash: ${ref.hash}`);
+  return {
+    ref,
+    cipher: {
+      Encrypted: {
+        threshold: {
+          SilentThreshold: {
+            td_params: Array.from(hexToUint8Array(encryptedKey)),
+            pk_bytes: Array.from(hexToUint8Array(group_pk)),
+            tau_params: Array.from(hexToUint8Array(tau_params))
+          }
+        },
+        symmetric: {
+          ChaCha20Poly1305: { nonce: Array.from(nonce) }
+        }
+      }
+    }
+  };
+}
 
 // src/da/upload.ts
 import fs from "fs";
 import path from "path";
 async function uploadData(options) {
+  const { name, description, price, type, guardianGroupInfo, ref, filePath } = options;
+  assert(
+    guardianGroupInfo.guardians && guardianGroupInfo.tauParams && guardianGroupInfo.aggKey && guardianGroupInfo.groupPk,
+    "Guardian group info is missing required properties"
+  );
+  console.log("Guardian group info:", guardianGroupInfo);
+  const account = (await getKeyring()).pairs[0];
+  const atomicPrice = toAtomicPaliAmount(price);
+  if (filePath) {
+    throw new Error("uploadData: file path upload is not implemented");
+  } else {
+    const storeType = type === "model" ? "Model" : type === "agent" ? "Agent" : "Dataset";
+    const { ref: dataRef, cipher } = await submitTEDataWithCipher(
+      account,
+      ref || "",
+      guardianGroupInfo.guardians,
+      guardianGroupInfo.tauParams,
+      guardianGroupInfo.aggKey,
+      guardianGroupInfo.groupPk
+    );
+    await registerDataAgreement(account, {
+      ref: dataRef,
+      guardians: guardianGroupInfo.guardians,
+      fees: atomicPrice,
+      cipher,
+      metadata: {
+        name,
+        description,
+        storeType,
+        groupId: guardianGroupInfo.groupId
+      }
+    });
+  }
+}
+async function uploadDataLegacy(options) {
   const { name, description, price, type, guardianGroupInfo, ref, filePath } = options;
   assert(
     guardianGroupInfo.guardians && guardianGroupInfo.tauParams && guardianGroupInfo.aggKey && guardianGroupInfo.groupPk,
@@ -1407,6 +1719,7 @@ async function uploadData(options) {
   );
   const account = (await getKeyring()).pairs[0];
   const ethAddress = "";
+  const atomicPrice = toAtomicPaliAmount(price);
   if (type === "dataset" && filePath) {
     const fileContent = fs.readFileSync(filePath);
     const selectedFile = new File([fileContent], path.basename(filePath), {
@@ -1419,7 +1732,7 @@ async function uploadData(options) {
         fileName: selectedFile.name,
         filePath: "",
         description,
-        baseCost: BigInt(Number(price) * 10 ** 18),
+        baseCost: atomicPrice,
         ownerL2Address: ethAddress,
         guardianInfo: guardianGroupInfo
       },
@@ -1452,12 +1765,12 @@ async function uploadData(options) {
 }
 
 // src/stake/add.ts
-async function addStake(account, amount) {
+async function addStake(account, amountBaseUnits) {
   const api2 = await getApi();
   assert(api2, "API not initialized");
   assert(account, "Account not initialized");
-  debugLog("Using account:", account.address);
-  const stakeTx = api2.tx.staking.bondExtra(amount);
+  debugLog("Using account:", account.address, "adding:", formatPaliAmount(amountBaseUnits));
+  const stakeTx = api2.tx.staking.bondExtra(amountBaseUnits);
   const hash = await signAndSend(stakeTx, account);
   debugLog(`Stake transaction sent with hash: ${hash.hash}`);
 }
@@ -1473,14 +1786,14 @@ async function joinIdleStaker(account) {
 }
 
 // src/stake/new.ts
-async function newStake(account, amount, rewardDestination = "Staked") {
+async function newStake(account, amountBaseUnits, rewardDestination = "Staked") {
   const api2 = await getApi();
   assert(api2, "API not initialized");
   assert(account, "Account not initialized");
   debugLog(
-    `Staking amount: ${amount} for account: ${account.address} with reward destination: ${rewardDestination}`
+    `Staking amount: ${formatPaliAmount(amountBaseUnits)} for account: ${account.address} with reward destination: ${rewardDestination}`
   );
-  const stakeTx = api2.tx.staking.bond(amount, rewardDestination);
+  const stakeTx = api2.tx.staking.bond(amountBaseUnits, rewardDestination);
   const hash = await signAndSend(stakeTx, account);
   debugLog(`Stake transaction sent with hash: ${hash.hash}`);
 }
@@ -1502,12 +1815,12 @@ async function payoutStake(account, eras, address) {
 }
 
 // src/stake/reduce.ts
-async function reduceStake(account, amount) {
+async function reduceStake(account, amountBaseUnits) {
   const api2 = await getApi();
   assert(api2, "API not initialized");
   assert(account, "Account not initialized");
-  debugLog("Using account:", account.address);
-  const unstakeTx = api2.tx.staking.unbond(amount);
+  debugLog("Using account:", account.address, "reducing:", formatPaliAmount(amountBaseUnits));
+  const unstakeTx = api2.tx.staking.unbond(amountBaseUnits);
   const hash = await signAndSend(unstakeTx, account);
   debugLog(`Unstake transaction sent with hash: ${hash.hash}`);
 }
@@ -1518,7 +1831,7 @@ async function removeStake(account) {
   assert(api2, "API not initialized");
   assert(account, "Account not initialized");
   const stakeAmount = (await api2.query.staking.ledger(account.address))?.toPrimitive();
-  debugLog("Removing entire stake amount:", stakeAmount?.active || 0n);
+  debugLog("Removing entire stake amount:", formatPaliAmount(stakeAmount?.active || 0n));
   const unstakeTx = api2.tx.staking.unbond(stakeAmount?.active || 0n);
   const hash = await signAndSend(unstakeTx, account);
   debugLog(`Unstake transaction sent with hash: ${hash.hash}`);
@@ -1535,25 +1848,25 @@ async function withdrawStake(account) {
 }
 
 // src/token/fund.ts
-async function fundAccount(account, amount, address) {
+async function fundAccount(account, amountBaseUnits, address) {
   const addr = address ? address : account.address;
   const keyring2 = await getKeyring();
   const api2 = await getApi();
   assert(api2, "API not initialized");
   debugLog(`
-Funding account: ${addr} with amount: ${amount}`);
-  const tx = api2.tx.balances.transferKeepAlive(addr, amount);
+Funding account: ${addr} with amount: ${formatPaliAmount(amountBaseUnits)}`);
+  const tx = api2.tx.balances.transferKeepAlive(addr, amountBaseUnits);
   const hash = await signAndSend(tx, keyring2.getPairs()[0]);
   debugLog(`Fund transaction sent with hash: ${hash.hash}`);
 }
 
 // src/token/transfer.ts
-async function transfer(account, amount, address) {
+async function transfer(account, amountBaseUnits, address) {
   const api2 = await getApi();
   assert(api2, "API not initialized");
   debugLog(`
-Transferring funds to account: ${address} with amount: ${amount}`);
-  const tx = api2.tx.balances.transferKeepAlive(address, amount);
+Transferring funds to account: ${address} with amount: ${formatPaliAmount(amountBaseUnits)}`);
+  const tx = api2.tx.balances.transferKeepAlive(address, amountBaseUnits);
   const hash = await signAndSend(tx, account);
   debugLog(`Transfer transaction sent with hash: ${hash.hash}`);
 }
@@ -1622,9 +1935,10 @@ export {
   MCryptFs,
   MCryptFsReader,
   MCryptFsWriter,
+  PALI_DECIMALS,
+  PALI_SYMBOL,
   PALLIORA_RPC_URL,
   PALLIORA_WS,
-  RpcApi,
   TX_WAIT_FINALIZATION,
   WsProvider3 as WsProvider,
   addStake,
@@ -1634,6 +1948,8 @@ export {
   createAccount,
   createAgreement,
   createGuardianGroup,
+  createSimpleAgreement,
+  dataContract,
   debugLog,
   decodeAggregateKey,
   decodeField,
@@ -1644,6 +1960,8 @@ export {
   encrypt2 as encrypt,
   fetchTokenProperties,
   formatBalanceWithTokenProperties,
+  formatPaliAmount,
+  fromAtomicPaliAmount,
   fundAccount,
   gen_shared_key,
   gen_stretched_key,
@@ -1657,8 +1975,8 @@ export {
   getGuardianNwParams,
   getGuardianParticipants,
   getKeyring,
-  getRpcApi,
   hexToUint8Array,
+  inferenceCompute,
   joinGuardian,
   joinIdleStaker,
   joinValidator,
@@ -1667,18 +1985,24 @@ export {
   payoutStake,
   provider,
   reduceStake,
+  registerDataAgreement,
   removeStake,
   rotateAndSetKeys,
+  runAgent,
   setIdentity,
   setWorker,
   signAndSend,
+  simpleCompute,
   submitData,
   submitTEData,
+  submitTEDataWithCipher,
   testCrypt,
+  toAtomicPaliAmount,
   tokenToBigint,
   transfer,
   uint8ArrayToBase64,
   uploadData,
+  uploadDataLegacy,
   utilCrypto,
   wasmCrypto,
   withdrawStake,