Function generateCodeSnippets

docs/scripts/generatePages/index.js:227–295

(modelId, requestInput, requestInputHttp, params, stream)

}

async function generateCodeSnippets(modelId, requestInput, requestInputHttp, params, stream) {
  const js = formatIntoCodeSnippet(
    JS_TEMPLATE,
    params,
    stream,
    (sectionsAsString) => {
      if (paramsExist(params)) {
        return sectionsAsString;
      }

      const replacedString = sectionsAsString
        .replace(
          // notice how we're removing params
          `const { error, output } = await model.run(input, params);`,
          `const { error, output } = await model.run(input);`
        )
        // notice how we're removing params, this is for the streaming case
        .replace(
          `const readStream = await model.run(input, params, stream);`,
          `const readStream = await model.run(input, stream);`
        );

      return replacedString;
    },
    [
      { tokenId: `'$MODEL_ID'`, replaceWith: `'${modelId}'` },
      { tokenId: `'$INPUT'`, replaceWith: JSON.stringify(requestInput, null, 2) },
      { tokenId: `'$PARAMS_PAYLOAD'`, replaceWith: JSON.stringify(params, null, 2) },
    ]
  );

  const py = formatIntoCodeSnippet(
    PY_TEMPLATE,
    params,
    stream,
    (sectionsAsString) => {
      if (paramsExist(params)) {
        return sectionsAsString;
      }

      const replacedString = sectionsAsString
        .replace(
          // notice how we're removing params
          `result = model.run(input, params)`,
          `result = model.run(input)`
        )
        // notice how we're removing params, this is for the streaming case
        .replace(
          `readStream = model.run(input, params, stream)`,
          `readStream = model.run(input, stream=stream)`
        );

      return replacedString;
    },
    [
      { tokenId: `"$MODEL_ID"`, replaceWith: `"${modelId}"` },
      { tokenId: `"$INPUT"`, replaceWith: JSON.stringify(requestInput, null, 2) },
      { tokenId: `"$PARAMS_PAYLOAD"`, replaceWith: JSON.stringify(params, null, 2) },