kar
/
iimtt-api
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects 1 Releases Wiki Activity

list-models

suvo-patch-1
Kar 2025-01-29 21:03:30 +05:30
parent cfa16dbdc7
commit e589d66e36
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/routes/api/aiTest.js
View File

@ -1,6 +1,6 @@
const aiFollowupQuestion = async (req, res) => { const aiFollowupQuestion = async (req, res) => {
const { MongoClient } = require('mongodb'); const { MongoClient } = require('mongodb');
// const fetch = require('node-fetch'); const fetch = require('node-fetch');
const { v4: uuidv4 } = require('uuid'); const { v4: uuidv4 } = require('uuid');
const url = process.env.MONGODB_URL; const url = process.env.MONGODB_URL;
const dbName = process.env.MONGO_DB_NAME; const dbName = process.env.MONGO_DB_NAME;
@ -14,12 +14,8 @@ const aiFollowupQuestion = async (req, res) => {
const PROVIDER_KEYS = JSON.parse(process.env.PROVIDER_KEY || '[]'); const PROVIDER_KEYS = JSON.parse(process.env.PROVIDER_KEY || '[]');
const PROVIDER_MODELS = { const PROVIDER_MODELS = {
openai: [ openai: [ "gpt-4o", "gpt-4o-mini", "o1", "o1-mini", "gpt-3.5-turbo", "gpt-3.5", "gpt-3.5-turbo-mini", "gpt-3.5-mini", "gpt-3.5-turbo-2", "gpt-3.5-2", "gpt-3.5-turbo-2-mini", "gpt-3.5-2-mini" ],
"gpt-4o", "gpt-4o-mini", "o1", "o1-mini", "gpt-3.5-turbo", "gpt-3.5" deepseek: [ "deepseek-chat", "deepseek-reasoner", "deepseek-qa", "deepseek-qa-mini" ],
],
deepseek: [
"deepseek-chat", "deepseek-reasoner"
]
// Add more providers and models as needed // Add more providers and models as needed
}; };
@ -61,7 +57,6 @@ const aiFollowupQuestion = async (req, res) => {
const { prompt, sessionId, provider, model = "default-model", key } = req.body; const { prompt, sessionId, provider, model = "default-model", key } = req.body;
// Check if key is provided and valid
if (!key || !PROVIDER_KEYS.includes(key)) { if (!key || !PROVIDER_KEYS.includes(key)) {
return res.status(403).json({ return res.status(403).json({
success: false, success: false,