Function getOpenaiCompletion

app/src/server/utils/openai.ts:47–67

(
  projectId: string,
  input: ChatCompletionCreateParams,
)

  input: ChatCompletionCreateParams,
): Promise<ChatCompletion | Stream<ChatCompletionChunk>>;
export async function getOpenaiCompletion(
  projectId: string,
  input: ChatCompletionCreateParams,
): Promise<ChatCompletion | Stream<ChatCompletionChunk>> {
  const apiKeys = await prisma.apiKey.findMany({
    where: { projectId: projectId },
  });

  const openaiApiKey = apiKeys.find((key) => key.provider === "OPENAI")?.apiKey;

  if (!openaiApiKey) {
    throw new Error("No OpenAI API key found");
  }

  const openai = new OriginalOpenAI({ apiKey: openaiApiKey });

  return await openai.chat.completions.create({
    ...input,
    tools: input.tools?.length ? input.tools : undefined,
  });
}

// Fall back to our own Azure instance for long-running requests
export async function getAzureGpt4Completion(input: ChatCompletionCreateParams) {