added mistral vector support (off the back of oai's)

This commit is contained in:
based
2023-12-16 08:37:39 +10:00
parent c517483141
commit 5dd2e8cd88
4 changed files with 12 additions and 7 deletions

View File

@@ -12,8 +12,9 @@ const { jsonParser } = require('../express-common');
*/
async function getVector(source, text) {
switch (source) {
case 'mistral':
case 'openai':
return require('../openai-vectors').getOpenAIVector(text);
return require('../openai-vectors').getOpenAIVector(text, source);
case 'transformers':
return require('../embedding').getTransformersVector(text);
case 'palm':