import { FlowBot, Scenario } from "@aurogpt/flowgpt";
import { ChatOpenAI } from "langchain/chat_models/openai";

const flowBot = FlowBot.create({
    name: "CustomerServiceBot",
    description: "Automated customer service assistant",
    scenarios: [
        Scenario.loadObject({
            name: "Browse Products",
            description: "Helps customers find and browse products",
            steps: [
                {
                    name: "Greet User",
                    description: "Welcome the user to the store",
                    instructions:
                        "Send a greeting message with options like 'View Products' or 'Search for a Product'.",
                },
                {
                    name: "Product Search",
                    description: "Assist the user in finding specific products",
                    instructions:
                        "Ask for the product name or category and display matching results.",
                },
                {
                    name: "View Product Details",
                    description:
                        "Provide detailed information about a selected product",
                    instructions:
                        "Show product details, price, and an 'Add to Cart' option.",
                },
            ],
        }),
        Scenario.loadObject({
            name: "Track Order",
            description: "Allows customers to check the status of their order",
            steps: [
                {
                    name: "Request Order ID",
                    description: "Ask the user to provide their order ID",
                    instructions:
                        "Prompt the user for their order ID to retrieve tracking information.",
                },
                {
                    name: "Display Order Status",
                    description: "Show the current status of the order",
                    instructions:
                        "Provide shipping updates, expected delivery date, and tracking details.",
                },
            ],
        }),
    ],
    model: new ChatOpenAI({
        modelName: "gpt-4o-mini",
        // apiKey: process.env.OPENAI_API_KEY,
    }),

    // Optional: Prompt for detecting scenario & step
    // prompt: "...",
});

// Example extending default prompt
flowBot.prompt += "This text is appended to the default prompt";

// Example usage
const response = await flowBot.invoke(`
    User: Hi
    Bot: Hello, How can I help you today?
    User: What is the status of my order id #324?
`);

console.log(response);