fix: update buildSdkMessages to handle undefined output in API clients (#7293)

* fix: update buildSdkMessages to handle undefined output in API clients

* fix: update vision model check to include model name in regex validation

src/renderer/src/aiCore/clients/BaseApiClient.ts

@@ -110,7 +110,7 @@ export abstract class BaseApiClient<
 
   abstract buildSdkMessages(
     currentReqMessages: TMessageParam[],
-    output: TRawOutput | string,
+    output: TRawOutput | string | undefined,
     toolResults: TMessageParam[],
     toolCalls?: TToolCall[]
   ): TMessageParam[]

src/renderer/src/aiCore/clients/openai/OpenAIApiClient.ts

@@ -337,10 +337,14 @@ export class OpenAIAPIClient extends OpenAIBaseClient<
 
   public buildSdkMessages(
     currentReqMessages: OpenAISdkMessageParam[],
-    output: string,
+    output: string | undefined,
     toolResults: OpenAISdkMessageParam[],
     toolCalls: OpenAI.Chat.Completions.ChatCompletionMessageToolCall[]
   ): OpenAISdkMessageParam[] {
+    if (!output && toolCalls.length === 0) {
+      return [...currentReqMessages, ...toolResults]
+    }
+
     const assistantMessage: OpenAISdkMessageParam = {
       role: 'assistant',
       content: output,

src/renderer/src/aiCore/clients/openai/OpenAIResponseAPIClient.ts

@@ -227,10 +227,18 @@ export class OpenAIResponseAPIClient extends OpenAIBaseClient<
 
   public buildSdkMessages(
     currentReqMessages: OpenAIResponseSdkMessageParam[],
-    output: string,
+    output: string | undefined,
     toolResults: OpenAIResponseSdkMessageParam[],
     toolCalls: OpenAIResponseSdkToolCall[]
   ): OpenAIResponseSdkMessageParam[] {
+    if (!output && toolCalls.length === 0) {
+      return [...currentReqMessages, ...toolResults]
+    }
+
+    if (!output) {
+      return [...currentReqMessages, ...(toolCalls || []), ...(toolResults || [])]
+    }
+
     const assistantMessage: OpenAIResponseSdkMessageParam = {
       role: 'assistant',
       content: [{ type: 'input_text', text: output }]

src/renderer/src/aiCore/middleware/core/McpToolChunkMiddleware.ts

@@ -153,7 +153,7 @@ function createToolHandlingTransform(
           if (toolResult.length > 0) {
             const output = ctx._internal.toolProcessingState?.output
 
-            const newParams = buildParamsWithToolResults(ctx, currentParams, output!, toolResult, toolCalls)
+            const newParams = buildParamsWithToolResults(ctx, currentParams, output, toolResult, toolCalls)
             await executeWithToolHandling(newParams, depth + 1)
           }
         } catch (error) {
@@ -243,7 +243,7 @@ async function executeToolUseResponses(
 function buildParamsWithToolResults(
   ctx: CompletionsContext,
   currentParams: CompletionsParams,
-  output: SdkRawOutput | string,
+  output: SdkRawOutput | string | undefined,
   toolResults: SdkMessageParam[],
   toolCalls: SdkToolCall[]
 ): CompletionsParams {

src/renderer/src/config/models.ts

@@ -184,7 +184,7 @@ const visionAllowedModels = [
   'deepseek-vl(?:[\\w-]+)?',
   'kimi-latest',
   'gemma-3(?:-[\\w-]+)',
-  'doubao-1.6-seed(?:-[\\w-]+)'
+  'doubao-seed-1[.-]6(?:-[\\w-]+)'
 ]
 
 const visionExcludedModels = [
@@ -238,7 +238,8 @@ export const FUNCTION_CALLING_MODELS = [
   'glm-4(?:-[\\w-]+)?',
   'learnlm(?:-[\\w-]+)?',
   'gemini(?:-[\\w-]+)?', // 提前排除了gemini的嵌入模型
-  'grok-3(?:-[\\w-]+)?'
+  'grok-3(?:-[\\w-]+)?',
+  'doubao-seed-1[.-]6(?:-[\\w-]+)?'
 ]
 
 const FUNCTION_CALLING_EXCLUDED_MODELS = [
@@ -2320,7 +2321,8 @@ export const GEMINI_SEARCH_MODELS = [
   'gemini-2.5-pro-preview-03-25',
   'gemini-2.5-pro-preview-05-06',
   'gemini-2.5-flash-preview',
-  'gemini-2.5-flash-preview-04-17'
+  'gemini-2.5-flash-preview-04-17',
+  'gemini-2.5-flash-preview-05-20'
 ]
 
 export const OPENAI_NO_SUPPORT_DEV_ROLE_MODELS = ['o1-preview', 'o1-mini']
@@ -2365,7 +2367,7 @@ export function isVisionModel(model: Model): boolean {
   // }
 
   if (model.provider === 'doubao') {
-    return VISION_REGEX.test(model.name) || model.type?.includes('vision') || false
+    return VISION_REGEX.test(model.name) || VISION_REGEX.test(model.id) || model.type?.includes('vision') || false
   }
 
   return VISION_REGEX.test(model.id) || model.type?.includes('vision') || false