diff --git a/frontend/src/components/ChatBox.tsx b/frontend/src/components/ChatBox.tsx index 2c555d68..dc5b1010 100644 --- a/frontend/src/components/ChatBox.tsx +++ b/frontend/src/components/ChatBox.tsx @@ -228,7 +228,7 @@ export default function Component({ availableModels } = useLocalState(); - const isGemma = MODEL_CONFIG[model]?.supportsVision || false; + const supportsVision = MODEL_CONFIG[model]?.supportsVision || false; const [images, setImages] = useState([]); const [imageUrls, setImageUrls] = useState>(new Map()); const [uploadedDocument, setUploadedDocument] = useState<{ @@ -917,7 +917,7 @@ export default function Component({ navigate({ to: "/pricing" }); } else { // If not on a vision model, switch to one first - if (!isGemma) { + if (!supportsVision) { const visionModelId = findFirstVisionModel(); if (visionModelId) { setModel(visionModelId); diff --git a/frontend/src/components/ModelSelector.tsx b/frontend/src/components/ModelSelector.tsx index 66443367..60122cca 100644 --- a/frontend/src/components/ModelSelector.tsx +++ b/frontend/src/components/ModelSelector.tsx @@ -46,6 +46,19 @@ export const MODEL_CONFIG: Record = { badge: "Pro", requiresPro: true, tokenLimit: 64000 + }, + "mistral-small-3-1-24b": { + displayName: "Mistral Small 3.1 24B", + badge: "Pro", + requiresPro: true, + supportsVision: true, + tokenLimit: 128000 + }, + "qwen2-5-72b": { + displayName: "Qwen 2.5 72B", + badge: "Pro", + requiresPro: true, + tokenLimit: 128000 } }; @@ -120,6 +133,11 @@ export function ModelSelector({ return false; } + // Filter out transcription models like Whisper + if (modelWithTasks.tasks.includes("transcribe")) { + return false; + } + return true; }); diff --git a/frontend/src/config/pricingConfig.tsx b/frontend/src/config/pricingConfig.tsx index 41d692c4..cc27c11c 100644 --- a/frontend/src/config/pricingConfig.tsx +++ b/frontend/src/config/pricingConfig.tsx @@ -44,6 +44,12 @@ export const PRICING_PLANS: PricingPlan[] = [ { text: "Rename Chats", included: true, icon: }, { text: "Gemma 3 27B", included: false, icon: }, { text: "DeepSeek R1 70B", included: false, icon: }, + { + text: "Mistral Small 3.1 24B", + included: false, + icon: + }, + { text: "Qwen 2.5 72B", included: false, icon: }, { text: "Image Upload", included: false, icon: }, { text: "Document Upload", included: false, icon: } ], @@ -71,6 +77,12 @@ export const PRICING_PLANS: PricingPlan[] = [ }, { text: "Gemma 3 27B", included: true, icon: }, { text: "DeepSeek R1 70B", included: false, icon: }, + { + text: "Mistral Small 3.1 24B", + included: false, + icon: + }, + { text: "Qwen 2.5 72B", included: false, icon: }, { text: "Image Upload", included: false, icon: }, { text: "Document Upload", included: false, icon: } ], @@ -102,6 +114,16 @@ export const PRICING_PLANS: PricingPlan[] = [ included: true, icon: }, + { + text: "Mistral Small 3.1 24B", + included: true, + icon: + }, + { + text: "Qwen 2.5 72B", + included: true, + icon: + }, { text: "Image Upload", included: true, icon: }, { text: "Document Upload", @@ -143,6 +165,16 @@ export const PRICING_PLANS: PricingPlan[] = [ included: true, icon: }, + { + text: "Mistral Small 3.1 24B", + included: true, + icon: + }, + { + text: "Qwen 2.5 72B", + included: true, + icon: + }, { text: "Image Upload", included: true, icon: }, { text: "Document Upload",