Function testOpenAIAPI

workers/utils.ts:9–154

(OPENAI_API_KEY: string, model: string)

const completions_url = "https://api.openai.com/v1/completions"

export async function testOpenAIAPI(OPENAI_API_KEY: string, model: string) {
  // Create request
  const base = {
    model,
    temperature,
    max_tokens,
    stream: true,
  }

  let body
  let url

  if (model == "gpt-4" || model == "gpt-3.5-turbo") {
    url = chat_url
    body = JSON.stringify({
      ...base,
      messages: [
        {
          role: "system",
          content: "You are an expert in quantum mechanics & computing.",
        },
        {
          role: "user",
          content:
            "Can you please provide an extensive & in-depth overview of Quantum Cryptography, and the current research being done?",
        },
      ],
    })
  } else {
    url = completions_url
    body = JSON.stringify({
      ...base,
      prompt:
        "Can you please provide an extensive & in-depth overview of Quantum Cryptography, and the current research being done?",
    })
  }

  // Send request
  let generationAttempts = 0
  const maxGenerationAttempts = 3
  let finishReason = null

  let startDate

  let ttfb
  let duration

  while (generationAttempts < maxGenerationAttempts) {
    generationAttempts++

    startDate = Date.now()

    const response = await fetch(url, {
      method: "POST",
      headers: {
        "Content-Type": "application/json",
        Authorization: `Bearer ${OPENAI_API_KEY}`,
      },