jbilcke-hf's picture
jbilcke-hf HF staff
working on some new features
3d4392e
raw
history blame
No virus
404 Bytes
// LLMEngine = the actual engine to use (eg. hugging face)
export type LLMVendor =
| "HUGGINGFACE"
| "OPENAI"
| "GROQ"
| "ANTHROPIC"
export type LLMVendorConfig = {
vendor: LLMVendor
apiKey: string
modelId: string
}
export type LLMPredictionFunctionParams = {
systemPrompt: string
userPrompt: string
nbMaxNewTokens: number
prefix?: string
// llmVendorConfig: LLMVendorConfig
}