34 lines
No EOL
1.5 KiB
TypeScript
34 lines
No EOL
1.5 KiB
TypeScript
import { setLLMConfig } from "@/store/slices/userConfig";
|
|
import { store } from "@/store/store";
|
|
|
|
export const handleSaveLLMConfig = async (llmConfig: LLMConfig, useCustomOllamaUrl: boolean) => {
|
|
if (!hasValidLLMConfig(llmConfig, useCustomOllamaUrl)) {
|
|
throw new Error('API key cannot be empty');
|
|
}
|
|
|
|
await fetch('/api/user-config', {
|
|
method: 'POST',
|
|
body: JSON.stringify(llmConfig)
|
|
});
|
|
|
|
store.dispatch(setLLMConfig(llmConfig));
|
|
}
|
|
|
|
export const hasValidLLMConfig = (llmConfig: LLMConfig, useCustomOllamaUrl: boolean) => {
|
|
if (!llmConfig.LLM) return false;
|
|
const OPENAI_API_KEY = llmConfig.OPENAI_API_KEY;
|
|
const GOOGLE_API_KEY = llmConfig.GOOGLE_API_KEY;
|
|
const MODEL = llmConfig.MODEL;
|
|
const PEXELS_API_KEY = llmConfig.PEXELS_API_KEY;
|
|
|
|
const isOllamaBaseConfigValid = PEXELS_API_KEY !== '' && PEXELS_API_KEY !== null && PEXELS_API_KEY !== undefined && MODEL !== '' && MODEL !== null && MODEL !== undefined;
|
|
|
|
return llmConfig.LLM === 'openai' ?
|
|
OPENAI_API_KEY !== '' && OPENAI_API_KEY !== null && OPENAI_API_KEY !== undefined :
|
|
llmConfig.LLM === 'google' ?
|
|
GOOGLE_API_KEY !== '' && GOOGLE_API_KEY !== null && GOOGLE_API_KEY !== undefined :
|
|
llmConfig.LLM === 'ollama' ?
|
|
useCustomOllamaUrl ?
|
|
isOllamaBaseConfigValid && llmConfig.LLM_PROVIDER_URL !== '' && llmConfig.LLM_PROVIDER_URL !== null && llmConfig.LLM_PROVIDER_URL !== undefined && llmConfig.LLM_API_KEY !== '' && llmConfig.LLM_API_KEY !== null && llmConfig.LLM_API_KEY !== undefined :
|
|
isOllamaBaseConfigValid : false;
|
|
} |