Add Optional Large Tier Execution Performance

src/api/execution.ts

@@ -7,15 +7,18 @@ import {
 import log from "loglevel";
 import { logPrefix } from "../utils";
 import { Router } from "./router";
+import { ExecutionPerformance } from "../types/requestPayload";
 
 // This class implements all the routes defined in the Dune API Docs: https://dune.com/docs/api/
 export class ExecutionClient extends Router {
   async executeQuery(
     queryID: number,
     parameters?: QueryParameter[],
+    performance?: ExecutionPerformance,
   ): Promise<ExecutionResponse> {
     const response = await this._post<ExecutionResponse>(`query/${queryID}/execute`, {
       query_parameters: parameters ? parameters : [],
+      performance: performance ? performance : ExecutionPerformance.Medium,
     });
     log.debug(logPrefix, `execute response ${JSON.stringify(response)}`);
     return response as ExecutionResponse;

src/api/extensions.ts

@@ -4,6 +4,7 @@ import log from "loglevel";
 import { logPrefix } from "../utils";
 import { ExecutionClient } from "./execution";
 import { POLL_FREQUENCY_SECONDS, THREE_MONTHS_IN_HOURS } from "../constants";
+import { ExecutionPerformance } from "../types/requestPayload";
 
 const TERMINAL_STATES = [
   ExecutionState.CANCELLED,
@@ -16,14 +17,19 @@ export class ExtendedClient extends ExecutionClient {
     queryID: number,
     parameters?: QueryParameter[],
     pingFrequency: number = POLL_FREQUENCY_SECONDS,
+    performance?: ExecutionPerformance,
   ): Promise<ResultsResponse> {
     log.info(
       logPrefix,
       `refreshing query https://dune.com/queries/${queryID} with parameters ${JSON.stringify(
         parameters,
       )}`,
     );
-    const { execution_id: jobID } = await this.execute(queryID, parameters);
+    const { execution_id: jobID } = await this.executeQuery(
+      queryID,
+      parameters,
+      performance,
+    );
     let { state } = await this.getExecutionStatus(jobID);
     while (!TERMINAL_STATES.includes(state)) {
       log.info(

src/api/query.ts

@@ -1,10 +1,6 @@
 // Assuming the existence of these imports based on your Python code
 import { Router } from "./router";
-import {
-  DuneQuery,
-  QueryParameter,
-  CreateQueryResponse,
-} from "../types";
+import { DuneQuery, QueryParameter, CreateQueryResponse } from "../types";
 import { CreateQueryPayload, UpdateQueryPayload } from "../types/requestPayload";
 
 export class QueryAPI extends Router {

src/api/router.ts

@@ -2,10 +2,7 @@ import { DuneError } from "../types";
 import fetch from "cross-fetch";
 import log from "loglevel";
 import { logPrefix } from "../utils";
-import {
-  RequestPayload,
-  payloadJSON,
-} from "../types/requestPayload";
+import { RequestPayload, payloadJSON } from "../types/requestPayload";
 
 const BASE_URL = "https://api.dune.com/api/v1";
 

src/types/requestPayload.ts

@@ -1,5 +1,10 @@
 import { QueryParameter } from "./queryParameter";
 
+export enum ExecutionPerformance {
+  Medium = "medium",
+  Large = "large",
+}
+
 export type RequestPayload =
   | ExecuteQueryPayload
   | UpdateQueryPayload
@@ -24,6 +29,7 @@ export function payloadJSON(payload?: RequestPayload): string {
 
 export interface ExecuteQueryPayload {
   query_parameters?: QueryParameter[];
+  performance: string;
 }
 
 export interface UpdateQueryPayload {

tests/e2e/client.spec.ts

@@ -7,6 +7,7 @@ import {
   QueryAPI,
 } from "../../src/";
 import log from "loglevel";
+import { ExecutionPerformance } from "../../src/types/requestPayload";
 
 const { DUNE_API_KEY } = process.env;
 const apiKey: string = DUNE_API_KEY ? DUNE_API_KEY : "No API Key";
@@ -75,6 +76,16 @@ describe("DuneClient: native routes", () => {
     expect(execution.execution_id).is.not.null;
   });
 
+  it("execute with Large tier performance", async () => {
+    const client = new DuneClient(apiKey);
+    const execution = await client.executeQuery(
+      1215383,
+      undefined,
+      ExecutionPerformance.Large,
+    );
+    expect(execution.execution_id).is.not.null;
+  });
+
   it("returns expected results on cancelled query exection", async () => {
     const client = new DuneClient(apiKey);
     // Execute and check state