63 lines
1.4 KiB
TypeScript
63 lines
1.4 KiB
TypeScript
import express from 'express';
|
|
import {
|
|
getAvailableChatModelProviders,
|
|
getAvailableEmbeddingModelProviders,
|
|
} from '../lib/providers';
|
|
import {
|
|
getGroqApiKey,
|
|
getOllamaApiEndpoint,
|
|
getOpenaiApiKey,
|
|
updateConfig,
|
|
} from '../config';
|
|
|
|
const router = express.Router();
|
|
|
|
router.get('/', async (_, res) => {
|
|
const config = {};
|
|
|
|
const [chatModelProviders, embeddingModelProviders] = await Promise.all([
|
|
getAvailableChatModelProviders(),
|
|
getAvailableEmbeddingModelProviders(),
|
|
]);
|
|
|
|
config['chatModelProviders'] = {};
|
|
config['embeddingModelProviders'] = {};
|
|
|
|
for (const provider in chatModelProviders) {
|
|
config['chatModelProviders'][provider] = Object.keys(
|
|
chatModelProviders[provider],
|
|
);
|
|
}
|
|
|
|
for (const provider in embeddingModelProviders) {
|
|
config['embeddingModelProviders'][provider] = Object.keys(
|
|
embeddingModelProviders[provider],
|
|
);
|
|
}
|
|
|
|
config['openaiApiKey'] = getOpenaiApiKey();
|
|
config['ollamaApiUrl'] = getOllamaApiEndpoint();
|
|
config['groqApiKey'] = getGroqApiKey();
|
|
|
|
res.status(200).json(config);
|
|
});
|
|
|
|
router.post('/', async (req, res) => {
|
|
const config = req.body;
|
|
|
|
const updatedConfig = {
|
|
API_KEYS: {
|
|
OPENAI: config.openaiApiKey,
|
|
GROQ: config.groqApiKey,
|
|
},
|
|
API_ENDPOINTS: {
|
|
OLLAMA: config.ollamaApiUrl,
|
|
},
|
|
};
|
|
|
|
updateConfig(updatedConfig);
|
|
|
|
res.status(200).json({ message: 'Config updated' });
|
|
});
|
|
|
|
export default router;
|