From 2f3e6348800b26721940cabf5544bd40d5df84b2 Mon Sep 17 00:00:00 2001 From: icarus Date: Wed, 22 Oct 2025 04:07:16 +0800 Subject: [PATCH] feat(hooks): add systemPrompt parameter to useQuickCompletion hook --- src/renderer/src/hooks/useQuickCompletion.ts | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/renderer/src/hooks/useQuickCompletion.ts b/src/renderer/src/hooks/useQuickCompletion.ts index 1b049162d6..636e0ee7bc 100644 --- a/src/renderer/src/hooks/useQuickCompletion.ts +++ b/src/renderer/src/hooks/useQuickCompletion.ts @@ -29,19 +29,20 @@ export type QuickCompletionParams = { params?: Partial> } -export const useQuickCompletion = () => { +export const useQuickCompletion = (systemPrompt: string) => { const { quickModel } = useDefaultModel() const completion = useCallback( async ({ prompt, onChunk, assistantUpdate, params }: QuickCompletionParams) => { const assistant = { ...getDefaultAssistant(), + prompt: systemPrompt, model: quickModel, ...assistantUpdate } satisfies Assistant return fetchChatCompletion({ prompt, assistant, onChunkReceived: onChunk, ...params }) }, - [quickModel] + [quickModel, systemPrompt] ) return completion