highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-25 15:30:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

add more flexible vision support check (#4429)

Browse Source
This commit is contained in:
pablonyx 2025-04-02 11:11:33 -07:00 committed by GitHub
parent c0b3681650
commit dc5a1e8fd0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -77,6 +77,7 @@ const MODEL_NAMES_SUPPORTING_IMAGE_INPUT = [
"claude-3-haiku-20240307", "claude-3-haiku-20240307",
// custom claude names // custom claude names
"claude-3.5-sonnet-v2@20241022", "claude-3.5-sonnet-v2@20241022",
"claude-3-7-sonnet@20250219",
// claude names with AWS Bedrock Suffix // claude names with AWS Bedrock Suffix
"claude-3-opus-20240229-v1:0", "claude-3-opus-20240229-v1:0",
"claude-3-sonnet-20240229-v1:0", "claude-3-sonnet-20240229-v1:0",
@ -125,12 +126,27 @@ export function checkLLMSupportsImageInput(model: string) {
const modelParts = model.split(/[/.]/); const modelParts = model.split(/[/.]/);
const lastPart = modelParts[modelParts.length - 1]?.toLowerCase(); const lastPart = modelParts[modelParts.length - 1]?.toLowerCase();
return MODEL_NAMES_SUPPORTING_IMAGE_INPUT.some((modelName) => { // Try matching the last part
const lastPartMatch = 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];
// lastPart is already lowercased above for tiny performance gain // lastPart is already lowercased above for tiny performance gain
return modelNameLastPart?.toLowerCase() === lastPart; return modelNameLastPart?.toLowerCase() === lastPart;
}); });
if (lastPartMatch) {
return true;
}
// If no match found, try getting the text after the first slash
if (model.includes("/")) {
const afterSlash = model.split("/")[1]?.toLowerCase();
return MODEL_NAMES_SUPPORTING_IMAGE_INPUT.some((modelName) =>
modelName.toLowerCase().includes(afterSlash)
);
}
return false;
} }
export const structureValue = ( export const structureValue = (