import { SessionState } from "indite-js/schemas";
import { OpenAIBlock } from "indite-js/schemas/features/blocks/integrations/openai";
import { createChatCompletionOpenAI } from "./createChatCompletionOpenAI";
import { ExecuteIntegrationResponse } from "../../../../types";
import { createSpeechOpenAI } from "./audio/createSpeechOpenAI";

export const executeOpenAIBlock = async (
  state: SessionState,
  block: OpenAIBlock
): Promise<ExecuteIntegrationResponse> => {
  switch (block.options?.task) {
    case "Create chat completion":
      return createChatCompletionOpenAI(state, {
        options: block.options,
        outgoingEdgeId: block.outgoingEdgeId,
        blockId: block.id,
      });
    case "Create speech":
      return createSpeechOpenAI(state, {
        options: block.options,
        outgoingEdgeId: block.outgoingEdgeId,
      });
    case "Create image":
    case undefined:
      return { outgoingEdgeId: block.outgoingEdgeId };
  }
};