diff --git a/app/(chat)/api/models/route.ts b/app/(chat)/api/models/route.ts index 2f10d6511..89ce3a209 100644 --- a/app/(chat)/api/models/route.ts +++ b/app/(chat)/api/models/route.ts @@ -84,6 +84,7 @@ export async function GET() { displayName: agentModelWithProvider.model.displayName, avatarUrl: agentModelWithProvider.model.avatarUrl, providerType: agentModelWithProvider.provider.type, + imageGenConfig: agentModelWithProvider.model.imageGenConfig, } : null, }); diff --git a/components/canvas/CanvasChatPanel.tsx b/components/canvas/CanvasChatPanel.tsx index 0e7d80b5c..425e51b12 100644 --- a/components/canvas/CanvasChatPanel.tsx +++ b/components/canvas/CanvasChatPanel.tsx @@ -1316,6 +1316,14 @@ export function CanvasChatPanel({ chatId, initialMessages = [], onClose, onOpenF displayName: string; avatarUrl?: string | null; providerType: string; + imageGenConfig?: { + maxImages?: number; + supportsSize?: boolean; + supportsPromptExtend?: boolean; + supportsNegativePrompt?: boolean; + supportsWatermark?: boolean; + supportsSeed?: boolean; + } | null; } | null; }>("/api/models", fetcher); @@ -1345,7 +1353,7 @@ export function CanvasChatPanel({ chatId, initialMessages = [], onClose, onOpenF avatarUrl: modelsData?.agentModel?.avatarUrl || null, providerType: "auto", supportsImageGen: true, - imageGenConfig: undefined, + imageGenConfig: modelsData?.agentModel?.imageGenConfig || undefined, }; // 所有可选模型(Agent模式 + 可视化模型) diff --git a/components/home/lovart-home.tsx b/components/home/lovart-home.tsx index b593192df..44f5730e1 100644 --- a/components/home/lovart-home.tsx +++ b/components/home/lovart-home.tsx @@ -403,6 +403,14 @@ export function LovartHome({ displayName: string; avatarUrl?: string | null; providerType: string; + imageGenConfig?: { + maxImages?: number; + supportsSize?: boolean; + supportsPromptExtend?: boolean; + supportsNegativePrompt?: boolean; + supportsWatermark?: boolean; + supportsSeed?: boolean; + } | null; } | null; }>(isAuthenticated ? "/api/models" : null, fetcher); @@ -471,8 +479,9 @@ export function LovartHome({ providerType: "auto", supportsImageGen: true, avatarUrl: modelsData?.agentModel?.avatarUrl || null, + imageGenConfig: modelsData?.agentModel?.imageGenConfig || undefined, }), - [modelsData?.agentModel?.avatarUrl] + [modelsData?.agentModel?.avatarUrl, modelsData?.agentModel?.imageGenConfig] ); const allComposerModels = useMemo(