cli, common, service: include protocol network for cost model creation

graphprotocol · Feb 13, 2024 · f571b80 · f571b80
1 parent d682744
commit f571b80
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git 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] <deployment-id> <file>
 ${chalk.bold('graph indexer cost set model')} [options] global <file>
 
 ${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
@@ -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
@@ -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
@@ -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
                     }
                   }
                 `,