LangChain

import { tool } from 'langchain';
import { createAgent } from 'langchain';
import { z } from 'zod';
import {
  Configuration,
  PactsApi,
  TransactionRecordsApi,
  TransactionsApi,
} from '@cobo/agentic-wallet';

const config = new Configuration({
  apiKey: process.env.AGENT_WALLET_API_KEY!,
  basePath: process.env.AGENT_WALLET_API_URL!,
});

const pactsApi = new PactsApi(config);
const txApi = new TransactionsApi(config);
const recordsApi = new TransactionRecordsApi(config);

async function returnPolicyDenial<T>(work: () => Promise<T>) {
  try {
    return await work();
  } catch (error) {
    const denial = (error as {
      response?: { data?: { error?: Record<string, unknown> } };
    }).response?.data?.error;
    if (denial) return denial;
    throw error;
  }
}

const submitPact = tool(
  async ({ wallet_id, intent }) => {
    const response = await pactsApi.submitPact({
      wallet_id,
      intent,
      spec: {
        completion_conditions: [{ type: 'time_elapsed', threshold: '86400' }],
      },
    });
    return response.data.result;
  },
  {
    name: 'submit_pact',
    description: 'Submit a pact and return the pact id.',
    schema: z.object({
      wallet_id: z.string(),
      intent: z.string(),
    }),
  },
);

const contractCall = tool(
  async ({ wallet_uuid, contract_addr, calldata, request_id }) => {
    return await returnPolicyDenial(async () => {
      const response = await txApi.contractCall(wallet_uuid, {
        chain_id: 'BASE_ETH',
        contract_addr,
        calldata,
        value: '0',
        request_id,
      });
      return response.data.result;
    });
  },
  {
    name: 'contract_call',
    description: 'Execute a policy-enforced contract call.',
    schema: z.object({
      wallet_uuid: z.string(),
      contract_addr: z.string(),
      calldata: z.string(),
      request_id: z.string(),
    }),
  },
);

const getTransactionRecordByRequestId = tool(
  async ({ wallet_uuid, request_id }) => {
    const response = await recordsApi.getTransactionRecordByRequestId(wallet_uuid, request_id);
    return response.data.result;
  },
  {
    name: 'get_transaction_record_by_request_id',
    description: 'Look up a transaction record by request id.',
    schema: z.object({
      wallet_uuid: z.string(),
      request_id: z.string(),
    }),
  },
);

const agent = createAgent({
  model: 'openai:gpt-4.1-mini',
  tools: [submitPact, contractCall, getTransactionRecordByRequestId],
  systemPrompt:
    'Before executing blockchain actions, submit a pact and wait until it is active. If a tool returns a policy denial, read the suggestion and retry with compliant params. After submission, track the result by request_id.',
});