miguelmj / milua Goto Github PK

local app = require "milua"
local http = require("socket.http")
local ltn12 = require("ltn12")
local cjson = require("cjson")

local config = dofile("config.lua")
local openaiApiKey = config.openaiApiKey

-- API endpoint
local apiUrl = "https://api.openai.com/v1/chat/completions"

app.add_handler(
    "POST",
    "/openai",
    function(captures, query, headers, body)

        if headers["method"] == "OPTIONS" then
            return "", 200, {
                ["Access-Control-Allow-Origin"] = "*", -- Allow requests from any origin
                ["Access-Control-Allow-Methods"] = "POST, OPTIONS", -- Allow POST and OPTIONS methods
                ["Access-Control-Allow-Headers"] = "Content-Type, Authorization" -- Allow specified headers
            }
        end

        -- Parse JSON from the request body
        local requestData = {
            model = "gpt-3.5-turbo",
            messages = {{role = "user", content = cjson.decode(body).content}},
            temperature = 0.7
        }

        print("Request:", requestData)

        -- Convert Lua table to JSON
        local requestBody = cjson.encode(requestData)

        print("RequestBB:", requestBody)

        -- Request headers
        local requestHeaders = {
            ["Content-Type"] = "application/json",
            ["Authorization"] = "Bearer " .. openaiApiKey
        }

        -- Perform HTTP request
        local response = {}
        local _, statusCode, _, _ = http.request{
            url = apiUrl,
            method = "POST",
            headers = requestHeaders,
            source = ltn12.source.string(requestBody),
            sink = ltn12.sink.table(response)
        }

        -- Check the response
        if statusCode == 200 then
            print("Made it here!")
            local responseBody = table.concat(response)
            print("responseBody", responseBody)
            return responseBody 
        else
            return "Error"
        end
    end
)

app.start()

const response = await axios.post("http://localhost:8800/openai", {
        content: input,
      });