Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-27 12:51:26 +08:00
Code Issues Actions 3 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'>>
}
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