Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-11 08:19:01 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

feat(hooks): add systemPrompt parameter to useQuickCompletion hook

Browse Source
This commit is contained in:
icarus 2025-10-22 04:07:16 +08:00
parent 684c0a7b63
commit 2f3e634880
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/renderer/src/hooks/useQuickCompletion.ts
View File

@ -29,19 +29,20 @@ export type QuickCompletionParams = {
params?: Partial<Omit<FetchChatCompletionParams, 'prompt' | 'messages' | 'assistant' | 'onChunkReceived'>> params?: Partial<Omit<FetchChatCompletionParams, 'prompt' | 'messages' | 'assistant' | 'onChunkReceived'>>
} }
export const useQuickCompletion = () => { export const useQuickCompletion = (systemPrompt: string) => {
const { quickModel } = useDefaultModel() const { quickModel } = useDefaultModel()
const completion = useCallback( const completion = useCallback(
async ({ prompt, onChunk, assistantUpdate, params }: QuickCompletionParams) => { async ({ prompt, onChunk, assistantUpdate, params }: QuickCompletionParams) => {
const assistant = { const assistant = {
...getDefaultAssistant(), ...getDefaultAssistant(),
prompt: systemPrompt,
model: quickModel, model: quickModel,
...assistantUpdate ...assistantUpdate
} satisfies Assistant } satisfies Assistant
return fetchChatCompletion({ prompt, assistant, onChunkReceived: onChunk, ...params }) return fetchChatCompletion({ prompt, assistant, onChunkReceived: onChunk, ...params })
}, },
[quickModel] [quickModel, systemPrompt]
) )
return completion return completion