index.js

/*
 * Copyright 2023 SXP Digital, LLC
 *
 * This file is part of Alchemy Worker: https://github.com/xpportal/alchemy-worker.
 *
 * Alchemy Worker is free software: you can redistribute it and/or modify
 * it under the terms of the Mozilla Public License version 2.0 as
 * published by the Mozilla Foundation.
 *
 * Alchemy Worker is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * Mozilla Public License for more details.
 *
 * You should have received a copy of the Mozilla Public License
 * along with Alchemy Worker. If not, see <https://www.mozilla.org/en-US/MPL/2.0/>.
 *
 * Alternatively, the contents of this file may be used under the terms of the
 * GNU General Public License version 2.0 or (at your option) any later version,
 * in which case the provisions of the GNU General Public License version 2.0
 * or later shall apply instead of those above.
 *
 * If you wish to allow use of your version of this file only under the terms of
 * the GPL version 2.0, and not to allow others to use your version of this file
 * under the MPL, indicate your decision by deleting the provisions above and
 * replace them with the notice and other provisions required by the GPL version 2.0.
 * If you do not delete the provisions above, a recipient may use your version of
 * this file under either the MPL or the GPL version 2.0.
 */

async function handleRequest(request) {
    if (request.method === 'OPTIONS') {
        return handleOptions(request);
    } else {
        return handleFetch(request);
    }
}

async function handleFetch(request) {
    // Extract the bearer token from the request headers
    const authorization = request.headers.get('Authorization');
    if (!authorization) {
        return new Response('Unauthorized', { status: 401 });
    }

    const data = await request.json();
    let prompt = data.inputs.personality;
    let prompt = data.Input.personality;
    let finalPrompt = prompt
        .replaceAll('#speaker', data.Input.Speaker)
        .replaceAll('#input', data.Input.Input)
        .replaceAll('#agent', data.Input.Agent)
        .replaceAll('#conversation', data.Input.Conversation)
        .replaceAll('undefined\n','' ).replaceAll('undefined','')
        .slice(-5000)

    const token = authorization.split(' ')[1];
    const postData = {
        prompt: finalPrompt	,
        max_tokens: 500,
        stop : ["###"],
        temperature: 0.7,
    };

    // Make the first request to the Davinci model
    const davinciResponse = await fetch('https://api.openai.com/v1/engines/text-davinci-003/completions', {
        method: 'POST',
        headers: {
            'Content-Type': 'application/json',
            'Authorization': `Bearer ${token}`
        },
        body: JSON.stringify(postData)
    });

    // Process the response from the Davinci model
    const davinciData = await davinciResponse.json();
    console.log(davinciData);

    // Return the results to the client
    const headers = {
        'Access-Control-Allow-Origin': '*',
        'Content-Type': 'application/json',
        'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
        'Access-Control-Allow-Headers': 'Content-Type, Authorization'
    };
    return new Response(JSON.stringify({ davinciData }), {
        headers
    });
}

function handleOptions(request) {
    // Make sure the necessary headers are present
    // for this to be a valid pre-flight request
    const headers = {
        'Access-Control-Allow-Origin': '*',
        'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
        'Access-Control-Allow-Headers': 'Content-Type, Authorization',
    };
    return new Response(null, { headers });
}

addEventListener('fetch', event => {
    event.respondWith(handleRequest(event.request))
});