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