jbilcke-hf's picture
jbilcke-hf HF staff
update env variables
879455c
raw
history blame
329 Bytes
"use server"
import { LLMEngine } from "@/types"
import { predictWithHuggingFace } from "./predictWithHuggingFace"
import { predictWithOpenAI } from "./predictWithOpenAI"
const llmEngine = `${process.env.LLM_ENGINE || ""}` as LLMEngine
export const predict = llmEngine === "OPENAI" ? predictWithOpenAI : predictWithHuggingFace