async runWithInputs()

in src/nodes/server/token-classification.ts [26:96]
64 lines of code
8 McCabe index (conditional complexity)

  async runWithInputs(inputs: Inputs) {
    const { text, modelid, apikey } = inputs;
    if (this.hf && apikey) {
      this.hf = new HfInference(apikey);
    }

    if (!text) {
      this.dispatchEvent(
        new CustomEvent("outputs", { detail: { results: null, text: text } })
      );
      return;
    }

    if (this.cachedResult && compareObjects(this.cachedInputs, inputs)) {
      this.dispatchEvent(
        new CustomEvent("outputs", {
          detail: {
            results: {
              tokens: this.cachedResult,
            },
            text: text,
          },
        })
      );
      return;
    }

    try {
      const tokenClassResult = await this.hf?.tokenClassification({
        model: modelid?.trim(),
        inputs: text,
        parameters: {
          aggregation_strategy: "max",
        },
      });

      if (!tokenClassResult) {
        throw new Error("Invalid response");
      }

      const tokens = tokenClassResult.map((x: any) => ({
        type: x.entity_group,
        text: x.word,
      }));

      this.cachedInputs = inputs;
      this.cachedResult = tokens;

      this.dispatchEvent(
        new CustomEvent("outputs", {
          detail: {
            results: {
              tokens: tokens,
            },
            text: text,
          },
        })
      );
    } catch (error: any) {
      this.dispatchEvent(
        new CustomEvent("outputs", {
          detail: {
            error: {
              title: "Error",
              message: error.message,
            },
          },
        })
      );
    }
  }