@human-protocol/sdk 5.0.0 → 5.1.0

package/src/base.ts

@@ -1,7 +1,5 @@
-import { ContractRunner, Overrides } from 'ethers';
+import { ContractRunner } from 'ethers';
 import { NetworkData } from './types';
-import { ChainId } from './enums';
-import { DEFAULT_AURORA_GAS_PRICE } from './constants';
 
 /**
  * ## Introduction
@@ -23,24 +21,4 @@ export abstract class BaseEthersClient {
     this.networkData = networkData;
     this.runner = runner;
   }
-
-  /**
-   * Internal helper to enrich transaction overrides with network specific defaults.
-   *
-   * Aurora networks use a fixed gas price. We always override any user provided
-   * gasPrice with the canonical DEFAULT_AURORA_GAS_PRICE to avoid mismatches
-   * or tx failures due to an unexpected value. For other networks the user
-   * supplied fee parameters are left untouched.
-   */
-  protected applyTxDefaults(txOptions: Overrides = {}): Overrides {
-    if (this.networkData.chainId === ChainId.AURORA_TESTNET) {
-      return {
-        ...txOptions,
-        gasPrice: DEFAULT_AURORA_GAS_PRICE,
-        maxFeePerGas: undefined,
-        maxPriorityFeePerGas: undefined,
-      } as Overrides;
-    }
-    return txOptions;
-  }
 }

package/src/constants.ts

@@ -1,4 +1,3 @@
-import { ethers } from 'ethers';
 import { ChainId } from './enums';
 import { NetworkData } from './types';
 
@@ -12,8 +11,6 @@ export const DEFAULT_PORT = 9000;
 
 export const DEFAULT_USE_SSL = false;
 
-export const DEFAULT_AURORA_GAS_PRICE = ethers.parseUnits('0.07', 'gwei');
-
 export enum HttpStatus {
   OK = 200,
   CREATED = 201,
@@ -120,21 +117,6 @@ export const NETWORKS: {
     oldSubgraphUrl: '',
     oldFactoryAddress: '',
   },
-  [ChainId.AURORA_TESTNET]: {
-    chainId: ChainId.AURORA_TESTNET,
-    title: 'Aurora Testnet',
-    scanUrl: 'https://explorer.testnet.aurora.dev/',
-    factoryAddress: '0xbA537fEF2442cD48f60686FE1Bbbe40F964a7940',
-    hmtAddress: '0x792abbcC99c01dbDec49c9fa9A828a186Da45C33',
-    stakingAddress: '0x7457d26a3C70Bd71F7557C773b303c1dB82BBB68',
-    kvstoreAddress: '0xD6D347ba6987519B4e42EcED43dF98eFf5465a23',
-    subgraphUrl:
-      'https://api.studio.thegraph.com/query/74256/amoy/version/latest',
-    subgraphUrlApiKey:
-      'https://gateway-arbitrum.network.thegraph.com/api/[SUBGRAPH_API_KEY]/deployments/id/QmVHYvjbsgAroR9EMgqqGTQ7aKPRFMVekY6evU6mFPUU7J',
-    oldSubgraphUrl: '',
-    oldFactoryAddress: '',
-  },
   [ChainId.LOCALHOST]: {
     chainId: ChainId.LOCALHOST,
     title: 'Localhost',

package/src/enums.ts

@@ -6,7 +6,6 @@ export enum ChainId {
   BSC_TESTNET = 97,
   POLYGON = 137,
   POLYGON_AMOY = 80002,
-  AURORA_TESTNET = 1313161555,
   LOCALHOST = 1338,
 }
 

package/src/error.ts

@@ -306,6 +306,13 @@ export const ErrorBulkPayOutVersion = new Error(
   'Invalid bulkPayOut parameters for the contract version of the specified escrow address'
 );
 
+/**
+ * @constant {Error} - Retry configuration is missing required parameters.
+ */
+export const ErrorRetryParametersMissing = new Error(
+  'Retry configuration must include both maxRetries and baseDelay'
+);
+
 /**
  * @constant {Warning} - Possible version mismatch.
  */

package/src/escrow.ts

@@ -10,7 +10,6 @@ import {
   HMToken__factory,
 } from '@human-protocol/core/typechain-types';
 import { ContractRunner, EventLog, Overrides, Signer, ethers } from 'ethers';
-import gqlFetch from 'graphql-request';
 import { BaseEthersClient } from './base';
 import { ESCROW_BULK_PAYOUT_MAX_ITEMS, NETWORKS } from './constants';
 import { requiresSigner } from './decorators';
@@ -62,13 +61,16 @@ import {
   IStatusEventFilter,
   IStatusEvent,
   ICancellationRefund,
+  ICancellationRefundFilter,
   IPayout,
   IEscrowWithdraw,
+  SubgraphOptions,
 } from './interfaces';
 import { EscrowStatus, NetworkData, TransactionLikeWithNonce } from './types';
 import {
   getSubgraphUrl,
   getUnixTimestamp,
+  customGqlFetch,
   isValidJson,
   isValidUrl,
   throwError,
@@ -244,7 +246,7 @@ export class EscrowClient extends BaseEthersClient {
         await this.escrowFactoryContract.createEscrow(
           tokenAddress,
           jobRequesterId,
-          this.applyTxDefaults(txOptions)
+          txOptions
         )
       ).wait();
 
@@ -405,7 +407,7 @@ export class EscrowClient extends BaseEthersClient {
           exchangeOracleFee,
           manifest,
           manifestHash,
-          this.applyTxDefaults(txOptions)
+          txOptions
         )
       ).wait();
 
@@ -503,7 +505,7 @@ export class EscrowClient extends BaseEthersClient {
           exchangeOracleFee,
           manifest,
           manifestHash,
-          this.applyTxDefaults(txOptions)
+          txOptions
         )
       ).wait();
 
@@ -567,11 +569,7 @@ export class EscrowClient extends BaseEthersClient {
         this.runner
       );
       await (
-        await tokenContract.transfer(
-          escrowAddress,
-          amount,
-          this.applyTxDefaults(txOptions)
-        )
+        await tokenContract.transfer(escrowAddress, amount, txOptions)
       ).wait();
 
       return;
@@ -692,7 +690,7 @@ export class EscrowClient extends BaseEthersClient {
             url,
             hash,
             fundsToReserve,
-            this.applyTxDefaults(txOptions)
+            txOptions
           )
         ).wait();
       } else {
@@ -700,7 +698,7 @@ export class EscrowClient extends BaseEthersClient {
           await escrowContract['storeResults(string,string)'](
             url,
             hash,
-            this.applyTxDefaults(txOptions)
+            txOptions
           )
         ).wait();
       }
@@ -756,9 +754,7 @@ export class EscrowClient extends BaseEthersClient {
     try {
       const escrowContract = this.getEscrowContract(escrowAddress);
 
-      await (
-        await escrowContract.complete(this.applyTxDefaults(txOptions))
-      ).wait();
+      await (await escrowContract.complete(txOptions)).wait();
       return;
     } catch (e) {
       return throwError(e);
@@ -898,7 +894,7 @@ export class EscrowClient extends BaseEthersClient {
             finalResultsHash,
             id,
             forceComplete,
-            this.applyTxDefaults(txOptions)
+            txOptions
           )
         ).wait();
       } else {
@@ -912,7 +908,7 @@ export class EscrowClient extends BaseEthersClient {
             finalResultsHash,
             id,
             forceComplete,
-            this.applyTxDefaults(txOptions)
+            txOptions
           )
         ).wait();
       }
@@ -966,9 +962,7 @@ export class EscrowClient extends BaseEthersClient {
 
     try {
       const escrowContract = this.getEscrowContract(escrowAddress);
-      await (
-        await escrowContract.cancel(this.applyTxDefaults(txOptions))
-      ).wait();
+      await (await escrowContract.cancel(txOptions)).wait();
     } catch (e) {
       return throwError(e);
     }
@@ -1014,11 +1008,7 @@ export class EscrowClient extends BaseEthersClient {
 
     try {
       const escrowContract = this.getEscrowContract(escrowAddress);
-      await (
-        await escrowContract.requestCancellation(
-          this.applyTxDefaults(txOptions)
-        )
-      ).wait();
+      await (await escrowContract.requestCancellation(txOptions)).wait();
     } catch (e) {
       return throwError(e);
     }
@@ -1076,10 +1066,7 @@ export class EscrowClient extends BaseEthersClient {
       const escrowContract = this.getEscrowContract(escrowAddress);
 
       const transactionReceipt = await (
-        await escrowContract.withdraw(
-          tokenAddress,
-          this.applyTxDefaults(txOptions)
-        )
+        await escrowContract.withdraw(tokenAddress, txOptions)
       ).wait();
 
       let amountTransferred: bigint | undefined = undefined;
@@ -1169,7 +1156,6 @@ export class EscrowClient extends BaseEthersClient {
     forceComplete = false,
     txOptions: Overrides = {}
   ): Promise<TransactionLikeWithNonce> {
-    txOptions = this.applyTxDefaults(txOptions);
     await this.ensureCorrectBulkPayoutInput(
       escrowAddress,
       recipients,
@@ -1940,6 +1926,7 @@ export class EscrowUtils {
    *
    *
    * @param {IEscrowsFilter} filter Filter parameters.
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {IEscrow[]} List of escrows that match the filter.
    *
    * **Code example**
@@ -1956,7 +1943,10 @@ export class EscrowUtils {
    * const escrows = await EscrowUtils.getEscrows(filters);
    * ```
    */
-  public static async getEscrows(filter: IEscrowsFilter): Promise<IEscrow[]> {
+  public static async getEscrows(
+    filter: IEscrowsFilter,
+    options?: SubgraphOptions
+  ): Promise<IEscrow[]> {
     if (filter.launcher && !ethers.isAddress(filter.launcher)) {
       throw ErrorInvalidAddress;
     }
@@ -1989,7 +1979,7 @@ export class EscrowUtils {
       statuses = Array.isArray(filter.status) ? filter.status : [filter.status];
       statuses = statuses.map((status) => EscrowStatus[status]);
     }
-    const { escrows } = await gqlFetch<{ escrows: EscrowData[] }>(
+    const { escrows } = await customGqlFetch<{ escrows: EscrowData[] }>(
       getSubgraphUrl(networkData),
       GET_ESCROWS_QUERY(filter),
       {
@@ -2004,7 +1994,8 @@ export class EscrowUtils {
         orderDirection: orderDirection,
         first: first,
         skip: skip,
-      }
+      },
+      options
     );
     return (escrows || []).map((e) => mapEscrow(e, networkData.chainId));
   }
@@ -2062,6 +2053,7 @@ export class EscrowUtils {
    *
    * @param {ChainId} chainId Network in which the escrow has been deployed
    * @param {string} escrowAddress Address of the escrow
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<IEscrow | null>} - Escrow data or null if not found.
    *
    * **Code example**
@@ -2074,7 +2066,8 @@ export class EscrowUtils {
    */
   public static async getEscrow(
     chainId: ChainId,
-    escrowAddress: string
+    escrowAddress: string,
+    options?: SubgraphOptions
   ): Promise<IEscrow | null> {
     const networkData = NETWORKS[chainId];
 
@@ -2086,10 +2079,11 @@ export class EscrowUtils {
       throw ErrorInvalidAddress;
     }
 
-    const { escrow } = await gqlFetch<{ escrow: EscrowData | null }>(
+    const { escrow } = await customGqlFetch<{ escrow: EscrowData | null }>(
       getSubgraphUrl(networkData),
       GET_ESCROW_BY_ADDRESS_QUERY(),
-      { escrowAddress: escrowAddress.toLowerCase() }
+      { escrowAddress: escrowAddress.toLowerCase() },
+      options
     );
     if (!escrow) return null;
 
@@ -2132,6 +2126,7 @@ export class EscrowUtils {
    * ```
    *
    * @param {IStatusEventFilter} filter Filter parameters.
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<StatusEvent[]>} - Array of status events with their corresponding statuses.
    *
    * **Code example**
@@ -2153,7 +2148,8 @@ export class EscrowUtils {
    * ```
    */
   public static async getStatusEvents(
-    filter: IStatusEventFilter
+    filter: IStatusEventFilter,
+    options?: SubgraphOptions
   ): Promise<IStatusEvent[]> {
     const {
       chainId,
@@ -2187,7 +2183,7 @@ export class EscrowUtils {
 
     const statusNames = effectiveStatuses.map((status) => EscrowStatus[status]);
 
-    const data = await gqlFetch<{
+    const data = await customGqlFetch<{
       escrowStatusEvents: StatusEvent[];
     }>(
       getSubgraphUrl(networkData),
@@ -2200,7 +2196,8 @@ export class EscrowUtils {
         orderDirection,
         first: Math.min(first, 1000),
         skip,
-      }
+      },
+      options
     );
 
     if (!data || !data['escrowStatusEvents']) {
@@ -2224,6 +2221,7 @@ export class EscrowUtils {
    * Fetch payouts from the subgraph.
    *
    * @param {IPayoutFilter} filter Filter parameters.
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<IPayout[]>} List of payouts matching the filters.
    *
    * **Code example**
@@ -2241,7 +2239,10 @@ export class EscrowUtils {
    * console.log(payouts);
    * ```
    */
-  public static async getPayouts(filter: IPayoutFilter): Promise<IPayout[]> {
+  public static async getPayouts(
+    filter: IPayoutFilter,
+    options?: SubgraphOptions
+  ): Promise<IPayout[]> {
     const networkData = NETWORKS[filter.chainId];
     if (!networkData) {
       throw ErrorUnsupportedChainID;
@@ -2258,7 +2259,7 @@ export class EscrowUtils {
     const skip = filter.skip || 0;
     const orderDirection = filter.orderDirection || OrderDirection.DESC;
 
-    const { payouts } = await gqlFetch<{ payouts: PayoutData[] }>(
+    const { payouts } = await customGqlFetch<{ payouts: PayoutData[] }>(
       getSubgraphUrl(networkData),
       GET_PAYOUTS_QUERY(filter),
       {
@@ -2269,7 +2270,8 @@ export class EscrowUtils {
         first: Math.min(first, 1000),
         skip,
         orderDirection,
-      }
+      },
+      options
     );
     if (!payouts) {
       return [];
@@ -2318,6 +2320,7 @@ export class EscrowUtils {
    *
    *
    * @param {Object} filter Filter parameters.
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<ICancellationRefund[]>} List of cancellation refunds matching the filters.
    *
    * **Code example**
@@ -2332,16 +2335,10 @@ export class EscrowUtils {
    * console.log(cancellationRefunds);
    * ```
    */
-  public static async getCancellationRefunds(filter: {
-    chainId: ChainId;
-    escrowAddress?: string;
-    receiver?: string;
-    from?: Date;
-    to?: Date;
-    first?: number;
-    skip?: number;
-    orderDirection?: OrderDirection;
-  }): Promise<ICancellationRefund[]> {
+  public static async getCancellationRefunds(
+    filter: ICancellationRefundFilter,
+    options?: SubgraphOptions
+  ): Promise<ICancellationRefund[]> {
     const networkData = NETWORKS[filter.chainId];
     if (!networkData) throw ErrorUnsupportedChainID;
     if (filter.escrowAddress && !ethers.isAddress(filter.escrowAddress)) {
@@ -2356,17 +2353,22 @@ export class EscrowUtils {
     const skip = filter.skip || 0;
     const orderDirection = filter.orderDirection || OrderDirection.DESC;
 
-    const { cancellationRefundEvents } = await gqlFetch<{
+    const { cancellationRefundEvents } = await customGqlFetch<{
       cancellationRefundEvents: CancellationRefundData[];
-    }>(getSubgraphUrl(networkData), GET_CANCELLATION_REFUNDS_QUERY(filter), {
-      escrowAddress: filter.escrowAddress?.toLowerCase(),
-      receiver: filter.receiver?.toLowerCase(),
-      from: filter.from ? getUnixTimestamp(filter.from) : undefined,
-      to: filter.to ? getUnixTimestamp(filter.to) : undefined,
-      first,
-      skip,
-      orderDirection,
-    });
+    }>(
+      getSubgraphUrl(networkData),
+      GET_CANCELLATION_REFUNDS_QUERY(filter),
+      {
+        escrowAddress: filter.escrowAddress?.toLowerCase(),
+        receiver: filter.receiver?.toLowerCase(),
+        from: filter.from ? getUnixTimestamp(filter.from) : undefined,
+        to: filter.to ? getUnixTimestamp(filter.to) : undefined,
+        first,
+        skip,
+        orderDirection,
+      },
+      options
+    );
 
     if (!cancellationRefundEvents || cancellationRefundEvents.length === 0) {
       return [];
@@ -2418,6 +2420,7 @@ export class EscrowUtils {
    *
    * @param {ChainId} chainId Network in which the escrow has been deployed
    * @param {string} escrowAddress Address of the escrow
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<ICancellationRefund>} Cancellation refund data
    *
    * **Code example**
@@ -2430,7 +2433,8 @@ export class EscrowUtils {
    */
   public static async getCancellationRefund(
     chainId: ChainId,
-    escrowAddress: string
+    escrowAddress: string,
+    options?: SubgraphOptions
   ): Promise<ICancellationRefund | null> {
     const networkData = NETWORKS[chainId];
     if (!networkData) throw ErrorUnsupportedChainID;
@@ -2439,12 +2443,13 @@ export class EscrowUtils {
       throw ErrorInvalidEscrowAddressProvided;
     }
 
-    const { cancellationRefundEvents } = await gqlFetch<{
+    const { cancellationRefundEvents } = await customGqlFetch<{
       cancellationRefundEvents: CancellationRefundData[];
     }>(
       getSubgraphUrl(networkData),
       GET_CANCELLATION_REFUND_BY_ADDRESS_QUERY(),
-      { escrowAddress: escrowAddress.toLowerCase() }
+      { escrowAddress: escrowAddress.toLowerCase() },
+      options
     );
 
     if (!cancellationRefundEvents || cancellationRefundEvents.length === 0) {

package/src/interfaces.ts

@@ -312,3 +312,13 @@ export interface IEscrowWithdraw {
   tokenAddress: string;
   withdrawnAmount: bigint;
 }
+
+/**
+ * Configuration options for subgraph requests with retry logic.
+ */
+export interface SubgraphOptions {
+  /** Maximum number of retry attempts */
+  maxRetries?: number;
+  /** Base delay between retries in milliseconds */
+  baseDelay?: number;
+}

package/src/kvstore.ts

@@ -17,15 +17,14 @@ import {
   ErrorUnsupportedChainID,
   InvalidKeyError,
 } from './error';
-import gqlFetch from 'graphql-request';
 import { NetworkData } from './types';
-import { getSubgraphUrl, isValidUrl } from './utils';
+import { getSubgraphUrl, customGqlFetch, isValidUrl } from './utils';
 import {
   GET_KVSTORE_BY_ADDRESS_AND_KEY_QUERY,
   GET_KVSTORE_BY_ADDRESS_QUERY,
 } from './graphql/queries/kvstore';
 import { KVStoreData } from './graphql';
-import { IKVStore } from './interfaces';
+import { IKVStore, SubgraphOptions } from './interfaces';
 /**
  * ## Introduction
  *
@@ -175,9 +174,7 @@ export class KVStoreClient extends BaseEthersClient {
   ): Promise<void> {
     if (key === '') throw ErrorKVStoreEmptyKey;
     try {
-      await (
-        await this.contract.set(key, value, this.applyTxDefaults(txOptions))
-      ).wait();
+      await (await this.contract.set(key, value, txOptions)).wait();
     } catch (e) {
       if (e instanceof Error) throw Error(`Failed to set value: ${e.message}`);
     }
@@ -222,13 +219,7 @@ export class KVStoreClient extends BaseEthersClient {
     if (keys.includes('')) throw ErrorKVStoreEmptyKey;
 
     try {
-      await (
-        await this.contract.setBulk(
-          keys,
-          values,
-          this.applyTxDefaults(txOptions)
-        )
-      ).wait();
+      await (await this.contract.setBulk(keys, values, txOptions)).wait();
     } catch (e) {
       if (e instanceof Error)
         throw Error(`Failed to set bulk values: ${e.message}`);
@@ -281,7 +272,7 @@ export class KVStoreClient extends BaseEthersClient {
         await this.contract.setBulk(
           [urlKey, hashKey],
           [url, contentHash],
-          this.applyTxDefaults(txOptions)
+          txOptions
         )
       ).wait();
     } catch (e) {
@@ -294,6 +285,7 @@ export class KVStoreClient extends BaseEthersClient {
    *
    * @param {string} address Address from which to get the key value.
    * @param {string} key Key to obtain the value.
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {string} Value of the key.
    *
    *
@@ -365,6 +357,7 @@ export class KVStoreUtils {
    *
    * @param {ChainId} chainId Network in which the KVStore is deployed
    * @param {string} address Address of the KVStore
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<IKVStore[]>} KVStore data
    * @throws {ErrorUnsupportedChainID} - Thrown if the network's chainId is not supported
    * @throws {ErrorInvalidAddress} - Thrown if the Address sent is invalid
@@ -380,7 +373,8 @@ export class KVStoreUtils {
    */
   public static async getKVStoreData(
     chainId: ChainId,
-    address: string
+    address: string,
+    options?: SubgraphOptions
   ): Promise<IKVStore[]> {
     const networkData = NETWORKS[chainId];
 
@@ -392,10 +386,11 @@ export class KVStoreUtils {
       throw ErrorInvalidAddress;
     }
 
-    const { kvstores } = await gqlFetch<{ kvstores: KVStoreData[] }>(
+    const { kvstores } = await customGqlFetch<{ kvstores: KVStoreData[] }>(
       getSubgraphUrl(networkData),
       GET_KVSTORE_BY_ADDRESS_QUERY(),
-      { address: address.toLowerCase() }
+      { address: address.toLowerCase() },
+      options
     );
 
     const kvStoreData = kvstores.map((item) => ({
@@ -412,6 +407,7 @@ export class KVStoreUtils {
    * @param {ChainId} chainId Network in which the KVStore is deployed
    * @param {string} address Address from which to get the key value.
    * @param {string} key Key to obtain the value.
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<string>} Value of the key.
    * @throws {ErrorUnsupportedChainID} - Thrown if the network's chainId is not supported
    * @throws {ErrorInvalidAddress} - Thrown if the Address sent is invalid
@@ -433,7 +429,8 @@ export class KVStoreUtils {
   public static async get(
     chainId: ChainId,
     address: string,
-    key: string
+    key: string,
+    options?: SubgraphOptions
   ): Promise<string> {
     if (key === '') throw ErrorKVStoreEmptyKey;
     if (!ethers.isAddress(address)) throw ErrorInvalidAddress;
@@ -444,10 +441,11 @@ export class KVStoreUtils {
       throw ErrorUnsupportedChainID;
     }
 
-    const { kvstores } = await gqlFetch<{ kvstores: KVStoreData[] }>(
+    const { kvstores } = await customGqlFetch<{ kvstores: KVStoreData[] }>(
       getSubgraphUrl(networkData),
       GET_KVSTORE_BY_ADDRESS_AND_KEY_QUERY(),
-      { address: address.toLowerCase(), key }
+      { address: address.toLowerCase(), key },
+      options
     );
 
     if (!kvstores || kvstores.length === 0) {
@@ -463,6 +461,7 @@ export class KVStoreUtils {
    * @param {ChainId} chainId Network in which the KVStore is deployed
    * @param {string} address Address from which to get the URL value.
    * @param {string} urlKey Configurable URL key. `url` by default.
+   * @param {SubgraphOptions} options Optional configuration for subgraph requests.
    * @returns {Promise<string>} URL value for the given address if it exists, and the content is valid
    *
    * **Code example**
@@ -480,7 +479,8 @@ export class KVStoreUtils {
   public static async getFileUrlAndVerifyHash(
     chainId: ChainId,
     address: string,
-    urlKey = 'url'
+    urlKey = 'url',
+    options?: SubgraphOptions
   ): Promise<string> {
     if (!ethers.isAddress(address)) throw ErrorInvalidAddress;
     const hashKey = urlKey + '_hash';
@@ -489,7 +489,7 @@ export class KVStoreUtils {
       hash = '';
 
     try {
-      url = await this.get(chainId, address, urlKey);
+      url = await this.get(chainId, address, urlKey, options);
     } catch (e) {
       if (e instanceof Error) throw Error(`Failed to get URL: ${e.message}`);
     }
@@ -539,12 +539,14 @@ export class KVStoreUtils {
    */
   public static async getPublicKey(
     chainId: ChainId,
-    address: string
+    address: string,
+    options?: SubgraphOptions
   ): Promise<string> {
     const publicKeyUrl = await this.getFileUrlAndVerifyHash(
       chainId,
       address,
-      KVStoreKeys.publicKey
+      KVStoreKeys.publicKey,
+      options
     );
 
     if (publicKeyUrl === '') {