File size: 351 Bytes
624088c
 
287a603
6463491
 
3ca0269
 
287a603
6463491
1
2
3
4
5
6
7
8
9
"use server"

import { LLMEngine } from "@/types"
import { predict as predictWithHuggingFace } from "./predictWithHuggingFace"
import { predict as predictWithOpenAI } from "./predictWithOpenAI"

const llmEngine = `${process.env.LLM_ENGINE || ""}` as LLMEngine

export const predict = llmEngine === "OPENAI" ? predictWithOpenAI : predictWithHuggingFace