popiart-node-canvas/src/store/execution/llmGenerateExecutor.ts

/**
 * LLM Generate Executor
 *
 * Unified executor for llmGenerate (text generation) nodes.
 * Used by both executeWorkflow and regenerateNode.
 */

import type { LLMGenerateNodeData } from "@/types";
import { buildLlmHeaders } from "@/store/utils/buildApiHeaders";
import type { NodeExecutionContext } from "./types";

export interface LlmGenerateOptions {
  /** When true, falls back to stored inputImages/inputPrompt if no connections provide them. */
  useStoredFallback?: boolean;
}

export async function executeLlmGenerate(
  ctx: NodeExecutionContext,
  options: LlmGenerateOptions = {}
): Promise<void> {
  const {
    node,
    getConnectedInputs,
    updateNodeData,
    signal,
    providerSettings,
  } = ctx;

  const { useStoredFallback = false } = options;

  const inputs = getConnectedInputs(node.id);
  const nodeData = node.data as LLMGenerateNodeData;

  // Determine images and text
  let images: string[];
  let text: string | null;

  if (useStoredFallback) {
    images = inputs.images.length > 0 ? inputs.images : nodeData.inputImages;
    text = inputs.text ?? nodeData.inputPrompt;
  } else {
    images = inputs.images;
    text = inputs.text ?? nodeData.inputPrompt;
  }

  if (!text) {
    updateNodeData(node.id, {
      status: "error",
      error: "Missing text input - connect a prompt node or set internal prompt",
    });
    throw new Error("Missing text input");
  }

  updateNodeData(node.id, {
    inputPrompt: text,
    inputImages: images,
    status: "loading",
    error: null,
  });

  const headers = buildLlmHeaders(nodeData.provider, providerSettings);

  const response = await fetch("/api/llm", {
    method: "POST",
    headers,
    body: JSON.stringify({
      prompt: text,
      ...(images.length > 0 && { images }),
      provider: nodeData.provider,
      model: nodeData.model,
      temperature: nodeData.temperature,
      maxTokens: nodeData.maxTokens,
    }),
    ...(signal ? { signal } : {}),
  });

  if (!response.ok) {
    const errorText = await response.text();
    let errorMessage = `HTTP ${response.status}`;
    try {
      const errorJson = JSON.parse(errorText);
      errorMessage = errorJson.error || errorMessage;
    } catch {
      if (errorText) errorMessage += ` - ${errorText.substring(0, 200)}`;
    }
    updateNodeData(node.id, {
      status: "error",
      error: errorMessage,
    });
    throw new Error(errorMessage);
  }

  const result = await response.json();

  if (result.success && result.text) {
    updateNodeData(node.id, {
      outputText: result.text,
      status: "complete",
      error: null,
    });
  } else {
    updateNodeData(node.id, {
      status: "error",
      error: result.error || "LLM generation failed",
    });
    throw new Error(result.error || "LLM generation failed");
  }
}