mirror of
https://github.com/ItzCrazyKns/Perplexica.git
synced 2025-06-17 15:28:37 +00:00
feat(providers): separate each provider
This commit is contained in:
36
src/lib/providers/index.ts
Normal file
36
src/lib/providers/index.ts
Normal file
@ -0,0 +1,36 @@
|
||||
import { loadGroqChatModels } from './groq';
|
||||
import { loadOllamaChatModels } from './ollama';
|
||||
import { loadOpenAIChatModels, loadOpenAIEmbeddingsModel } from './openai';
|
||||
import { loadTransformersEmbeddingsModel } from './transformers';
|
||||
|
||||
const chatModelProviders = {
|
||||
openai: loadOpenAIChatModels,
|
||||
groq: loadGroqChatModels,
|
||||
ollama: loadOllamaChatModels,
|
||||
};
|
||||
|
||||
const embeddingModelProviders = {
|
||||
openai: loadOpenAIEmbeddingsModel,
|
||||
local: loadTransformersEmbeddingsModel,
|
||||
ollama: loadOllamaChatModels,
|
||||
};
|
||||
|
||||
export const getAvailableChatModelProviders = async () => {
|
||||
const models = {};
|
||||
|
||||
for (const provider in chatModelProviders) {
|
||||
models[provider] = await chatModelProviders[provider]();
|
||||
}
|
||||
|
||||
return models;
|
||||
};
|
||||
|
||||
export const getAvailableEmbeddingModelProviders = async () => {
|
||||
const models = {};
|
||||
|
||||
for (const provider in embeddingModelProviders) {
|
||||
models[provider] = await embeddingModelProviders[provider]();
|
||||
}
|
||||
|
||||
return models;
|
||||
};
|
Reference in New Issue
Block a user