diff --git a/packages/indexer-cli/src/commands/indexer/cost/set/model.ts b/packages/indexer-cli/src/commands/indexer/cost/set/model.ts index e8ac4e506..f07c7a490 100644 --- a/packages/indexer-cli/src/commands/indexer/cost/set/model.ts +++ b/packages/indexer-cli/src/commands/indexer/cost/set/model.ts @@ -4,7 +4,7 @@ import fs from 'fs' import { loadValidatedConfig } from '../../../../config' import { createIndexerManagementClient } from '../../../../client' -import { fixParameters } from '../../../../command-helpers' +import { extractProtocolNetworkOption, fixParameters } from '../../../../command-helpers' import { parseCostModel, parseDeploymentID, @@ -18,7 +18,7 @@ ${chalk.bold('graph indexer cost set model')} [options] ${chalk.bold('graph indexer cost set model')} [options] global ${chalk.dim('Options:')} - + -n, --network Protocol network -h, --help Show usage information -o, --output table|json|yaml Choose the output format: table (default), JSON, or YAML ` @@ -31,6 +31,13 @@ module.exports = { const { print, parameters } = toolbox const { h, help, merged, o, output } = parameters.options + + const protocolNetwork = extractProtocolNetworkOption(parameters.options, true) + + if (!protocolNetwork) { + throw new Error('Protocol network is required') + } + const [deployment, filename] = fixParameters(parameters, { h, help, merged }) || [] const outputFormat = o || output || 'table' @@ -69,6 +76,7 @@ module.exports = { deployment, model, variables: null, + protocolNetwork, }) try { diff --git a/packages/indexer-cli/src/cost.ts b/packages/indexer-cli/src/cost.ts index d3cc8487d..9ae890f82 100644 --- a/packages/indexer-cli/src/cost.ts +++ b/packages/indexer-cli/src/cost.ts @@ -204,6 +204,7 @@ export const costModels = async ( deployment model variables + protocolNetwork } } `) @@ -228,6 +229,7 @@ export const costModel = async ( deployment model variables + protocolNetwork } } `, @@ -258,6 +260,7 @@ export const setCostModel = async ( deployment model variables + protocolNetwork } } `, @@ -284,6 +287,7 @@ export const deleteCostModels = async ( deployment model variables + protocolNetwork } } `, diff --git a/packages/indexer-common/src/indexer-management/client.ts b/packages/indexer-common/src/indexer-management/client.ts index 648dea8cc..3f33f4009 100644 --- a/packages/indexer-common/src/indexer-management/client.ts +++ b/packages/indexer-common/src/indexer-management/client.ts @@ -360,12 +360,14 @@ const SCHEMA_SDL = gql` deployment: String! model: String variables: String + protocolNetwork: String! } input CostModelInput { deployment: String! model: String variables: String + protocolNetwork: String! } type Query { diff --git a/packages/indexer-service/src/server/cost.ts b/packages/indexer-service/src/server/cost.ts index 1b288733c..0b8602a8e 100644 --- a/packages/indexer-service/src/server/cost.ts +++ b/packages/indexer-service/src/server/cost.ts @@ -94,6 +94,7 @@ export const createCostServer = async ({ deployment: String! model: String variables: String + protocolNetwork: String! } type Query { @@ -126,6 +127,7 @@ export const createCostServer = async ({ deployment model variables + protocolNetwork } } `, @@ -172,6 +174,7 @@ export const createCostServer = async ({ deployment model variables + protocolNetwork } } `,