highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-28 21:05:17 +02:00
Code Issues Packages Projects Releases Wiki Activity

Support new model for image input

Browse Source
This commit is contained in:
Weves
2025-01-13 13:11:07 -08:00
committed by Chris Weaver
parent cb66aadd80
commit a610b6bd8d
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
web/src/lib/llm/utils.ts
View File

@@ -102,12 +102,13 @@ const MODEL_NAMES_SUPPORTING_IMAGE_INPUT = [
// meta models // meta models
"llama-3.2-90b-vision-instruct", "llama-3.2-90b-vision-instruct",
"llama-3.2-11b-vision-instruct", "llama-3.2-11b-vision-instruct",
"Llama-3-2-11B-Vision-Instruct-yb",
]; ];
export function checkLLMSupportsImageInput(model: string) { export function checkLLMSupportsImageInput(model: string) {
// Original exact match check // Original exact match check
const exactMatch = MODEL_NAMES_SUPPORTING_IMAGE_INPUT.some( const exactMatch = MODEL_NAMES_SUPPORTING_IMAGE_INPUT.some(
(modelName) => modelName === model (modelName) => modelName.toLowerCase() === model.toLowerCase()
); );
if (exactMatch) { if (exactMatch) {
@@ -116,12 +117,13 @@ export function checkLLMSupportsImageInput(model: string) {
// Additional check for the last part of the model name // Additional check for the last part of the model name
const modelParts = model.split(/[/.]/); const modelParts = model.split(/[/.]/);
const lastPart = modelParts[modelParts.length - 1]; const lastPart = modelParts[modelParts.length - 1]?.toLowerCase();
return MODEL_NAMES_SUPPORTING_IMAGE_INPUT.some((modelName) => { return MODEL_NAMES_SUPPORTING_IMAGE_INPUT.some((modelName) => {
const modelNameParts = modelName.split(/[/.]/); const modelNameParts = modelName.split(/[/.]/);
const modelNameLastPart = modelNameParts[modelNameParts.length - 1]; const modelNameLastPart = modelNameParts[modelNameParts.length - 1];
return modelNameLastPart === lastPart; // lastPart is already lowercased above for tiny performance gain
return modelNameLastPart?.toLowerCase() === lastPart;
}); });
} }