@pulseai/sdk 0.1.1 → 0.1.2

package/dist/index.js

@@ -41,8 +41,10 @@ var MAINNET_ADDRESSES = {
   feeDistributor: "0x51EdD8E4C4B423b952821fc9e2a7dad15a858B56",
   identityRegistry: "0x8004A169FB4a3325136EB29fA0ceB6D2e539a432",
   reputationRegistry: "0x8004BAa17C55a88189AE136b182e5fdA19dE9b63",
-  usdm: "0xFAfDdbb3FC7688494971a79cc65DCa3EF82079E7"
+  usdm: "0xFAfDdbb3FC7688494971a79cc65DCa3EF82079E7",
+  buyerRelay: "0x633054593db34c5aAe36F784faeAe51b9604e037"
 };
+var PLATFORM_BUYER_AGENT_ID = 8154n;
 var DEFAULT_INDEXER_URLS = {
   testnet: "https://pulse-indexer.up.railway.app",
   mainnet: "https://pulse-indexer.up.railway.app"
@@ -2664,10 +2666,17 @@ async function deployJobTerms(client, agentId, params) {
   const { masterAddress } = await deployWarrenMaster(client, agentId, pageAddress, contentBytes.length, 0);
   return { masterAddress, pageAddress, hash, txHash };
 }
+function stripCodeFence(content) {
+  let result = content.trim();
+  result = result.replace(/^```\w*\n?/, "");
+  result = result.replace(/\n?```\s*$/, "");
+  return result;
+}
 async function deployDeliverable(client, agentId, jobId, params, indexerUrl) {
   const { json, hash } = createDeliverable(params);
-  const contentBytes = new TextEncoder().encode(json);
-  const { pageAddress, txHash } = await deployWarrenPage(client, json);
+  const warrenContent = params.type === "inline" && params.content ? stripCodeFence(params.content) : json;
+  const contentBytes = new TextEncoder().encode(warrenContent);
+  const { pageAddress, txHash } = await deployWarrenPage(client, warrenContent);
   const { masterAddress } = await deployWarrenMaster(client, agentId, pageAddress, contentBytes.length, 1);
   await write(client, {
     address: client.addresses.jobEngine,
@@ -2886,7 +2895,8 @@ async function callOpenAI(params) {
   if (!content) {
     throw new Error("OpenAI returned an empty response");
   }
-  return { content, raw: payload };
+  const truncated = payload?.choices?.[0]?.finish_reason === "length";
+  return { content, truncated, raw: payload };
 }
 async function callAnthropic(params) {
   const systemPrompt = params.messages.filter((message) => message.role === "system").map((message) => message.content).join("\n\n").trim();
@@ -2919,7 +2929,8 @@ async function callAnthropic(params) {
   if (!content) {
     throw new Error("Anthropic returned an empty response");
   }
-  return { content, raw: payload };
+  const truncated = payload?.stop_reason === "max_tokens";
+  return { content, truncated, raw: payload };
 }
 async function callGoogle(params) {
   const systemPrompt = params.messages.filter((message) => message.role === "system").map((message) => message.content).join("\n\n").trim();
@@ -2953,7 +2964,8 @@ async function callGoogle(params) {
   if (!content) {
     throw new Error("Google returned an empty response");
   }
-  return { content, raw: payload };
+  const truncated = payload?.candidates?.[0]?.finishReason === "MAX_TOKENS";
+  return { content, truncated, raw: payload };
 }
 async function callAI(params) {
   if (params.provider === "openai") {
@@ -2971,12 +2983,15 @@ async function callAI(params) {
 // src/handler/site-modifier.ts
 var FETCH_TIMEOUT_MS = 3e4;
 var MAX_HTML_BYTES = 500 * 1024;
-var DEFAULT_MAX_TOKENS = 4096;
+var DEFAULT_MAX_TOKENS = 16384;
+var AI_MAX_RETRIES = 3;
+var RETRY_DELAY_MS = 2e3;
+var GOOGLE_FALLBACK_MODEL = "gemini-2.5-pro";
 var SUPPORTED_PROVIDERS = ["anthropic", "google", "openai"];
 var DEFAULT_MODELS = {
   openai: "gpt-4o-mini",
   anthropic: "claude-3-5-sonnet-latest",
-  google: "gemini-1.5-pro"
+  google: "gemini-3.1-pro-preview"
 };
 var DEFAULT_SYSTEM_PROMPT = [
   "You are an expert frontend engineer modifying an existing HTML document.",
@@ -3029,6 +3044,10 @@ function isHtmlContentType(contentType) {
   const lower = contentType.toLowerCase();
   return lower.includes("text/html") || lower.includes("application/xhtml+xml");
 }
+function isJsonContentType(contentType) {
+  const lower = contentType.toLowerCase();
+  return lower.includes("application/json") || lower.includes("text/json");
+}
 function extractHtmlFromText(text) {
   const htmlFence = text.match(/```html\s*([\s\S]*?)```/i);
   if (htmlFence?.[1]) {
@@ -3042,12 +3061,49 @@ function extractHtmlFromText(text) {
       return fencedContent;
     }
   }
+  const openFence = text.match(/```html?\s*\n?([\s\S]+)/i);
+  if (openFence?.[1]) {
+    const content = openFence[1].trim();
+    if (/<html[\s>]|<!doctype html/i.test(content)) {
+      return content;
+    }
+  }
   const trimmed = text.trim();
   if (/<html[\s>]|<!doctype html/i.test(trimmed)) {
     return trimmed;
   }
   return null;
 }
+function extractHtmlFromJson(jsonText) {
+  try {
+    const parsed = JSON.parse(jsonText);
+    if (typeof parsed === "string") {
+      if (/<html[\s>]|<!doctype html/i.test(parsed)) return parsed;
+      return null;
+    }
+    if (typeof parsed !== "object" || parsed === null) return null;
+    const obj = parsed;
+    const nested = obj.data && typeof obj.data === "object" ? obj.data : null;
+    const candidates = [
+      obj.html,
+      obj.content,
+      obj.body,
+      nested?.content,
+      nested?.html,
+      nested?.body
+    ];
+    for (const candidate of candidates) {
+      if (typeof candidate === "string" && candidate.trim().length > 0) {
+        if (/<html[\s>]|<!doctype html|<head[\s>]|<body[\s>]|<div[\s>]/i.test(candidate)) {
+          return candidate.trim();
+        }
+      }
+    }
+    return null;
+  } catch {
+    return null;
+  }
+}
 function getByteLength(content) {
   return new TextEncoder().encode(content).byteLength;
 }
@@ -3102,17 +3158,71 @@ var SiteModifierHandler = class {
       originalHtml,
       "```"
     ].join("\n");
-    const aiResponse = await callAI({
-      provider,
-      model,
-      maxTokens,
-      apiKey,
-      signal: context.abortSignal,
-      messages: [
-        { role: "system", content: systemPrompt },
-        { role: "user", content: userPrompt }
-      ]
-    });
+    let aiResponse = null;
+    let lastError = null;
+    const aiMessages = [
+      { role: "system", content: systemPrompt },
+      { role: "user", content: userPrompt }
+    ];
+    for (let attempt = 1; attempt <= AI_MAX_RETRIES; attempt++) {
+      try {
+        aiResponse = await callAI({
+          provider,
+          model,
+          maxTokens,
+          apiKey,
+          signal: context.abortSignal,
+          messages: aiMessages
+        });
+        if (aiResponse.truncated) {
+          throw new Error(
+            "AI output was truncated (hit token limit). The generated HTML is incomplete."
+          );
+        }
+        break;
+      } catch (err) {
+        lastError = err instanceof Error ? err : new Error(String(err));
+        console.error(
+          `[site-modifier] ${model} failed (attempt ${attempt}/${AI_MAX_RETRIES}): ${lastError.message}`
+        );
+        if (provider === "google" && model !== GOOGLE_FALLBACK_MODEL) {
+          try {
+            console.log(
+              `[site-modifier] falling back to ${GOOGLE_FALLBACK_MODEL} (attempt ${attempt}/${AI_MAX_RETRIES})`
+            );
+            aiResponse = await callAI({
+              provider,
+              model: GOOGLE_FALLBACK_MODEL,
+              maxTokens,
+              apiKey,
+              signal: context.abortSignal,
+              messages: aiMessages
+            });
+            if (aiResponse.truncated) {
+              throw new Error(
+                "AI output was truncated (hit token limit). The generated HTML is incomplete."
+              );
+            }
+            break;
+          } catch (fallbackErr) {
+            lastError = fallbackErr instanceof Error ? fallbackErr : new Error(String(fallbackErr));
+            console.error(
+              `[site-modifier] ${GOOGLE_FALLBACK_MODEL} also failed (attempt ${attempt}/${AI_MAX_RETRIES}): ${lastError.message}`
+            );
+          }
+        }
+      }
+      if (attempt < AI_MAX_RETRIES) {
+        const delay = RETRY_DELAY_MS * attempt;
+        console.log(`[site-modifier] retrying in ${delay}ms...`);
+        await new Promise((r) => setTimeout(r, delay));
+      }
+    }
+    if (!aiResponse) {
+      throw new Error(
+        `AI generation failed after ${AI_MAX_RETRIES} attempts. Please try again later. Last error: ${lastError?.message ?? "unknown"}`
+      );
+    }
     const modifiedHtml = extractHtmlFromText(aiResponse.content);
     if (!modifiedHtml) {
       throw new Error("AI response did not contain valid HTML output");
@@ -3164,7 +3274,9 @@ var SiteModifierHandler = class {
   async fetchHtmlWithTimeout(siteUrl, parentSignal) {
     const controller = new AbortController();
     const timeoutId = setTimeout(() => {
-      controller.abort(new Error(`Site fetch timed out after ${FETCH_TIMEOUT_MS}ms`));
+      controller.abort(
+        new Error(`Site fetch timed out after ${FETCH_TIMEOUT_MS}ms for ${siteUrl}`)
+      );
     }, FETCH_TIMEOUT_MS);
     const parentAbort = () => controller.abort(parentSignal?.reason);
     if (parentSignal) {
@@ -3185,10 +3297,12 @@ var SiteModifierHandler = class {
           `Failed to fetch site HTML: ${response.status} ${response.statusText}`
         );
       }
-      const contentType = response.headers.get("content-type");
-      if (contentType && !isHtmlContentType(contentType)) {
+      const contentType = response.headers.get("content-type") ?? "";
+      const isHtml = !contentType || isHtmlContentType(contentType);
+      const isJson = isJsonContentType(contentType);
+      if (!isHtml && !isJson) {
         throw new Error(
-          `Fetched content is not HTML (content-type: ${contentType})`
+          `Fetched content is not HTML or JSON (content-type: ${contentType}, url: ${siteUrl})`
         );
       }
       const contentLengthHeader = response.headers.get("content-length");
@@ -3206,9 +3320,18 @@ var SiteModifierHandler = class {
           `HTML too large (${htmlBuffer.byteLength} bytes). Limit is ${MAX_HTML_BYTES} bytes`
         );
       }
-      const html = new TextDecoder().decode(htmlBuffer).trim();
+      let html = new TextDecoder().decode(htmlBuffer).trim();
       if (!html) {
-        throw new Error("Fetched HTML is empty");
+        throw new Error("Fetched content is empty");
+      }
+      if (isJson) {
+        const extracted = extractHtmlFromJson(html);
+        if (!extracted) {
+          throw new Error(
+            `JSON response from ${siteUrl} does not contain extractable HTML content`
+          );
+        }
+        html = extracted;
       }
       return html;
     } catch (error) {
@@ -3216,7 +3339,7 @@ var SiteModifierHandler = class {
         if (parentSignal?.aborted) {
           throw new Error("Site fetch aborted");
         }
-        throw new Error(`Site fetch timed out after ${FETCH_TIMEOUT_MS}ms`);
+        throw new Error(`Site fetch timed out after ${FETCH_TIMEOUT_MS}ms for ${siteUrl}`);
       }
       throw error;
     } finally {
@@ -3783,6 +3906,7 @@ var HandlerProviderRuntime = class {
       try {
         await this.checkNewJobs();
         await this.checkInProgressJobs();
+        await this.checkEvaluatedJobs();
       } catch (e) {
         this.onError?.(e instanceof Error ? e : new Error(String(e)));
       }
@@ -3930,6 +4054,23 @@ var HandlerProviderRuntime = class {
       }
     }
   }
+  async checkEvaluatedJobs() {
+    const evaluatedJobs = await this.indexer.getJobs({
+      status: 4,
+      agentId: Number(this.agentId)
+    });
+    for (const job of evaluatedJobs) {
+      const settleKey = `settle:${job.jobId}`;
+      if (!this.canProcess(settleKey)) continue;
+      try {
+        await settle(this.client, BigInt(job.jobId));
+        console.log(`[handler-provider] auto-settled job ${job.jobId}`);
+        this.markProcessed(settleKey);
+      } catch (e) {
+        this.markFailed(settleKey, e);
+      }
+    }
+  }
   canProcess(key) {
     if (this.processedJobs.has(key)) return false;
     return (this.failedJobs.get(key) ?? 0) < this.maxRetries;
@@ -4802,6 +4943,528 @@ var erc8004ReputationAbi = [
     "anonymous": false
   }
 ];
+
+// src/abis/BuyerRelay.ts
+var buyerRelayAbi = [
+  {
+    "type": "constructor",
+    "inputs": [
+      {
+        "name": "owner_",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "identityRegistry_",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "pulseExtension_",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "jobEngine_",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "serviceMarketplace_",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "usdm_",
+        "type": "address",
+        "internalType": "address"
+      }
+    ],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "buyerAgentId",
+    "inputs": [],
+    "outputs": [
+      {
+        "name": "",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "cancelFor",
+    "inputs": [
+      {
+        "name": "jobId",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "claimRefund",
+    "inputs": [
+      {
+        "name": "jobId",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "createJobFor",
+    "inputs": [
+      {
+        "name": "offeringId",
+        "type": "uint256",
+        "internalType": "uint256"
+      },
+      {
+        "name": "termsHash",
+        "type": "bytes32",
+        "internalType": "bytes32"
+      },
+      {
+        "name": "maxPrice",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "outputs": [
+      {
+        "name": "jobId",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "evaluateFor",
+    "inputs": [
+      {
+        "name": "jobId",
+        "type": "uint256",
+        "internalType": "uint256"
+      },
+      {
+        "name": "approved",
+        "type": "bool",
+        "internalType": "bool"
+      },
+      {
+        "name": "feedback",
+        "type": "string",
+        "internalType": "string"
+      }
+    ],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "identityRegistry",
+    "inputs": [],
+    "outputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "contract IERC8004Identity"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "jobDeposit",
+    "inputs": [
+      {
+        "name": "",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "outputs": [
+      {
+        "name": "",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "jobEngine",
+    "inputs": [],
+    "outputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "contract IJobEngine"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "jobPayer",
+    "inputs": [
+      {
+        "name": "",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "outputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "address"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "jobRefundState",
+    "inputs": [
+      {
+        "name": "",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ],
+    "outputs": [
+      {
+        "name": "",
+        "type": "uint8",
+        "internalType": "enum IBuyerRelay.RefundState"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "onERC721Received",
+    "inputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "",
+        "type": "uint256",
+        "internalType": "uint256"
+      },
+      {
+        "name": "",
+        "type": "bytes",
+        "internalType": "bytes"
+      }
+    ],
+    "outputs": [
+      {
+        "name": "",
+        "type": "bytes4",
+        "internalType": "bytes4"
+      }
+    ],
+    "stateMutability": "pure"
+  },
+  {
+    "type": "function",
+    "name": "owner",
+    "inputs": [],
+    "outputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "address"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "pulseExtension",
+    "inputs": [],
+    "outputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "contract IPulseExtension"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "renounceOwnership",
+    "inputs": [],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "serviceMarketplace",
+    "inputs": [],
+    "outputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "contract IServiceMarketplace"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "setup",
+    "inputs": [
+      {
+        "name": "agentURI",
+        "type": "string",
+        "internalType": "string"
+      }
+    ],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "transferOwnership",
+    "inputs": [
+      {
+        "name": "newOwner",
+        "type": "address",
+        "internalType": "address"
+      }
+    ],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "usdm",
+    "inputs": [],
+    "outputs": [
+      {
+        "name": "",
+        "type": "address",
+        "internalType": "address"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "event",
+    "name": "EvaluatedFor",
+    "inputs": [
+      {
+        "name": "jobId",
+        "type": "uint256",
+        "indexed": true,
+        "internalType": "uint256"
+      },
+      {
+        "name": "payer",
+        "type": "address",
+        "indexed": true,
+        "internalType": "address"
+      },
+      {
+        "name": "approved",
+        "type": "bool",
+        "indexed": false,
+        "internalType": "bool"
+      }
+    ],
+    "anonymous": false
+  },
+  {
+    "type": "event",
+    "name": "JobCreatedFor",
+    "inputs": [
+      {
+        "name": "jobId",
+        "type": "uint256",
+        "indexed": true,
+        "internalType": "uint256"
+      },
+      {
+        "name": "payer",
+        "type": "address",
+        "indexed": true,
+        "internalType": "address"
+      },
+      {
+        "name": "price",
+        "type": "uint256",
+        "indexed": false,
+        "internalType": "uint256"
+      }
+    ],
+    "anonymous": false
+  },
+  {
+    "type": "event",
+    "name": "OwnershipTransferred",
+    "inputs": [
+      {
+        "name": "previousOwner",
+        "type": "address",
+        "indexed": true,
+        "internalType": "address"
+      },
+      {
+        "name": "newOwner",
+        "type": "address",
+        "indexed": true,
+        "internalType": "address"
+      }
+    ],
+    "anonymous": false
+  },
+  {
+    "type": "event",
+    "name": "RefundClaimed",
+    "inputs": [
+      {
+        "name": "jobId",
+        "type": "uint256",
+        "indexed": true,
+        "internalType": "uint256"
+      },
+      {
+        "name": "payer",
+        "type": "address",
+        "indexed": true,
+        "internalType": "address"
+      },
+      {
+        "name": "amount",
+        "type": "uint256",
+        "indexed": false,
+        "internalType": "uint256"
+      }
+    ],
+    "anonymous": false
+  },
+  {
+    "type": "error",
+    "name": "AgentAlreadyInitialized",
+    "inputs": []
+  },
+  {
+    "type": "error",
+    "name": "AgentNotActive",
+    "inputs": []
+  },
+  {
+    "type": "error",
+    "name": "InsufficientPayment",
+    "inputs": [
+      {
+        "name": "required",
+        "type": "uint256",
+        "internalType": "uint256"
+      },
+      {
+        "name": "provided",
+        "type": "uint256",
+        "internalType": "uint256"
+      }
+    ]
+  },
+  {
+    "type": "error",
+    "name": "InvalidJobStatus",
+    "inputs": [
+      {
+        "name": "current",
+        "type": "uint8",
+        "internalType": "enum DataTypes.JobStatus"
+      },
+      {
+        "name": "expected",
+        "type": "uint8",
+        "internalType": "enum DataTypes.JobStatus"
+      }
+    ]
+  },
+  {
+    "type": "error",
+    "name": "InvalidRefundState",
+    "inputs": [
+      {
+        "name": "current",
+        "type": "uint8",
+        "internalType": "enum IBuyerRelay.RefundState"
+      },
+      {
+        "name": "expected",
+        "type": "uint8",
+        "internalType": "enum IBuyerRelay.RefundState"
+      }
+    ]
+  },
+  {
+    "type": "error",
+    "name": "OnlyBuyer",
+    "inputs": []
+  },
+  {
+    "type": "error",
+    "name": "OwnableInvalidOwner",
+    "inputs": [
+      {
+        "name": "owner",
+        "type": "address",
+        "internalType": "address"
+      }
+    ]
+  },
+  {
+    "type": "error",
+    "name": "OwnableUnauthorizedAccount",
+    "inputs": [
+      {
+        "name": "account",
+        "type": "address",
+        "internalType": "address"
+      }
+    ]
+  },
+  {
+    "type": "error",
+    "name": "Reentrancy",
+    "inputs": []
+  },
+  {
+    "type": "error",
+    "name": "ZeroAddress",
+    "inputs": []
+  }
+];
 export {
   ACCEPT_TIMEOUT,
   BPS_DENOMINATOR,
@@ -4815,6 +5478,7 @@ export {
   JobStatus,
   MAINNET_ADDRESSES,
   MEMO_TYPES,
+  PLATFORM_BUYER_AGENT_ID,
   PULSE_DOMAIN,
   ProviderRuntime,
   RISK_POOL_BPS,
@@ -4825,6 +5489,7 @@ export {
   USDM_MAINNET,
   acceptJob,
   activateOffering,
+  buyerRelayAbi,
   callAI,
   cancelJob,
   createAgentCard,