commerce/operations/chatOperations.ts

import { post } from './fetch';
import childrenBookDefinition from './function-definitions/childrenBookDefinition';

const DEFAULT_SYSTEM_PROMPT = `You are a genius story writer with incredible ability to craft narratives, stories, and children's books that captivate the imagination and entertain. Make the quality of your writing as excellent as possible (think Dr. Seuss, J.K. Rowling, etc). Feel free to include act structure names/chapters.)`;

const DEFAULT_USER_PROMPT = `The book is about a dude named SOULJABOI programmer who is trying to build a startup and is enchanted by naughty witches.`;

const generateRequestPayload = (messages: { role: string; content: string }[]) => ({
  model: 'gpt-3.5-turbo',
  messages,
  functions: childrenBookDefinition // Make sure this is correctly imported and structured
});

// TODO(Benson): Type the interface returned from open ai

function getFunctionCallArguments<T>(response: any) {
  return JSON.parse(response.text.function_call.arguments);
}

// export interface IStory {
//     title: string;
//     topic: string;
//     introduction: string;
//     narrativeStructure: string;
//     archetypes_characters: string;
//     pages: { text: string }[]
// }

async function createStoryAsync(
  systemPrompt = DEFAULT_SYSTEM_PROMPT,
  userPrompt = DEFAULT_USER_PROMPT
): Promise<any> {
  const messages = [
    {
      role: 'system',
      content: systemPrompt
    },
    {
      role: 'user',
      content: userPrompt
    }
  ];
  const data = await post('/api/open-ai/chat', generateRequestPayload(messages));
  // const data = await post('/api/revalidate', generateRequestPayload(messages));
  return getFunctionCallArguments<any>(data);
}

export default { createStoryAsync };