fix: model should not include file attachment tools if not supported

web-app/src/lib/completion.ts

@@ -26,6 +26,8 @@
   ConfigOptions,
 } from 'token.js'
 
+import { getModelCapabilities } from '@/lib/models'
+
 // Extended config options to include custom fetch function
 type ExtendedConfigOptions = ConfigOptions & {
   fetch?: typeof fetch
@@ -38,6 +40,7 @@
 import { useAppState } from '@/hooks/useAppState'
 import { injectFilesIntoPrompt } from './fileMetadata'
 import { Attachment } from '@/types/attachment'
+import { ModelCapabilities } from '@/types/models'
 
 export type ChatCompletionResponse =
   | chatCompletion
@@ -66,11 +69,11 @@
   const docMetadata = documents
     .filter((doc) => doc.id) // Only include processed documents
     .map((doc) => ({
       id: doc.id!,
       name: doc.name,
       type: doc.fileType,
       size: typeof doc.size === 'number' ? doc.size : undefined,
       chunkCount: typeof doc.chunkCount === 'number' ? doc.chunkCount : undefined,
     }))
 
   const textWithFiles =
@@ -88,15 +91,15 @@
 
   // Add image attachments to content array
   images.forEach((img) => {
     if (img.base64 && img.mimeType) {
       contentParts.push({
         type: ContentType.Image,
         image_url: {
           url: `data:${img.mimeType};base64,${img.base64}`,
           detail: 'auto',
         },
       } as any)
     }
   })
 
   return {
@@ -168,45 +171,45 @@
  * @returns
  */
 export const sendCompletion = async (
   thread: Thread,
   provider: ModelProvider,
   messages: ChatCompletionMessageParam[],
   abortController: AbortController,
   tools: MCPTool[] = [],
   stream: boolean = true,
   params: Record<string, object> = {}
 ): Promise<ChatCompletionResponse | undefined> => {
   if (!thread?.model?.id || !provider) return undefined
 
   let providerName = provider.provider as unknown as keyof typeof models
 
   if (!Object.keys(models).some((key) => key === providerName))
     providerName = 'openai-compatible'
 
   const tokenJS = new TokenJS({
     apiKey:
       provider.api_key ?? (await getServiceHub().core().getAppToken()) ?? '',
     // TODO: Retrieve from extension settings
     baseURL: provider.base_url,
     // Use Tauri's fetch to avoid CORS issues only for openai-compatible provider
     fetch: IS_DEV ? fetch : getServiceHub().providers().fetch(),
     // OpenRouter identification headers for Jan
     // ref: https://openrouter.ai/docs/api-reference/overview#headers
     ...(provider.provider === 'openrouter' && {
       defaultHeaders: {
         'HTTP-Referer': 'https://jan.ai',
         'X-Title': 'Jan',
       },
     }),
     // Add Origin header for local providers to avoid CORS issues
     ...((provider.base_url?.includes('localhost:') ||
       provider.base_url?.includes('127.0.0.1:')) && {
       fetch: getServiceHub().providers().fetch(),
       defaultHeaders: {
         Origin: 'tauri://localhost',
       },
     }),
   } as ExtendedConfigOptions)
 
   if (
     thread.model.id &&
@@ -232,10 +235,25 @@
   }
 
   // Inject RAG tools on-demand (not in global tools list)
+  const providerModelConfig = provider.models?.find(
+    (model) => model.id === thread.model?.id || model.model === thread.model?.id
+  )
+  const effectiveCapabilities = Array.isArray(
+    providerModelConfig?.capabilities
+  )
+    ? providerModelConfig?.capabilities ?? []
+    : getModelCapabilities(provider.provider, thread.model.id)
+  const modelSupportsTools = effectiveCapabilities.includes(
+    ModelCapabilities.TOOLS
+  )
   let usableTools = tools
   try {
     const attachmentsEnabled = useAttachments.getState().enabled
-    if (attachmentsEnabled && PlatformFeatures[PlatformFeature.ATTACHMENTS]) {
+    if (
+      attachmentsEnabled &&
+      PlatformFeatures[PlatformFeature.ATTACHMENTS] &&
+      modelSupportsTools
+    ) {
       const ragTools = await getServiceHub().rag().getTools().catch(() => [])
       if (Array.isArray(ragTools) && ragTools.length) {
         usableTools = [...tools, ...ragTools]