const axios = require('axios');
const { v4: uuidv4 } = require('uuid');

async function upstageai(question, { reasoning = false, web_search = false } = {}) {
    try {
        if (!question) throw new Error('Question is required');
        if (typeof reasoning !== 'boolean') throw new Error('Reasoning must be a boolean');
        if (typeof web_search !== 'boolean') throw new Error('Reasoning must be a boolean');
        
        const { data: t } = await axios.post('https://console.upstage.ai/playground/chat', [], {
            headers: {
                'next-action': 'c2741d382a9e142748374f0428fe8f06e9befe8a',
                'next-router-state-tree': '%5B%22%22%2C%7B%22children%22%3A%5B%22(frame)%22%2C%7B%22children%22%3A%5B%22playground%22%2C%7B%22children%22%3A%5B%22(llm)%22%2C%7B%22children%22%3A%5B%22chat%22%2C%7B%22children%22%3A%5B%22__PAGE__%22%2C%7B%7D%2C%22%2Fplayground%2Fchat%22%2C%22refresh%22%5D%7D%5D%7D%5D%7D%5D%7D%5D%7D%2Cnull%2Cnull%2Ctrue%5D',
                referer: 'https://console.upstage.ai/playground/chat',
                'user-agent': 'Mozilla/5.0 (Linux; Android 10; K) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Mobile Safari/537.36'
            }
        });
        const token = JSON.parse(t.split('\n').filter(l => l.trim().startsWith('1:'))[0].split('1:')[1]).token;
        
        const { data } = await axios.post('https://ap-northeast-2.apistage.ai/v1/web/demo/chat/completions', {
            conversation_id: uuidv4(),
            stream: false,
            log_enabled: false,
            messages: [
                {
                    role: 'user',
                    content: question,
                    ...(web_search && { mode: ['search'] })
                }
            ],
            model: 'solar-pro2',
            reasoning_effort: reasoning ? 'high' : 'low'
        }, {
            headers: {
                'x-csrf-token': token
            },
            ...(reasoning && {
                params: {
                    include_think: true
                }
            })
        });
        
        let result = { think: '', text: '', search: [] };
        let fullText = '';
        const lines = data.split('\n\n');
        for (const line of lines) {
            if (line.startsWith('data:') && !line.includes('[DONE]')) {
                const d = JSON.parse(line.substring(6));
                if (d?.choices) fullText += d?.choices?.[0]?.delta?.content;
                if (d.search?.status?.action === 'search_finish') result.search = d.search.search_queries[0].results;
            }
        }
        
        const thinkMatch = fullText.match(/<think>([\s\S]*?)<\/think>/);
        result.think = thinkMatch ? thinkMatch[1].trim() : '';
        result.text = fullText.replace(/[\s\S]*?<\/think>/, '').trim();
        
        return result;
    } catch (error) {
        throw new Error(error.message);
    }
}

// Usage:
const resp = await upstageai('berita terbaru yang ada di Indonesia', { reasoning: true, web_search: true });
console.log(resp);