whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: context count

Browse Source
This commit is contained in:
kangfenmao 2024-09-28 22:01:09 +08:00
parent 4af3d16e61
commit 2771a842fe
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/renderer/src/providers/AnthropicProvider.ts
View File

@ -58,7 +58,7 @@ export default class AnthropicProvider extends BaseProvider {
const { contextCount, maxTokens, streamOutput } = getAssistantSettings(assistant) const { contextCount, maxTokens, streamOutput } = getAssistantSettings(assistant)
const userMessagesParams: MessageParam[] = [] const userMessagesParams: MessageParam[] = []
const _messages = filterContextMessages(takeRight(messages, contextCount + 1)) const _messages = filterContextMessages(takeRight(messages, contextCount + 2))
onFilterMessages(_messages) onFilterMessages(_messages)

2
src/renderer/src/providers/GeminiProvider.ts
View File

@ -59,7 +59,7 @@ export default class GeminiProvider extends BaseProvider {
const model = assistant.model || defaultModel const model = assistant.model || defaultModel
const { contextCount, maxTokens, streamOutput } = getAssistantSettings(assistant) const { contextCount, maxTokens, streamOutput } = getAssistantSettings(assistant)
const userMessages = filterContextMessages(takeRight(messages, contextCount + 1)) const userMessages = filterContextMessages(takeRight(messages, contextCount + 2))
onFilterMessages(userMessages) onFilterMessages(userMessages)
if (first(userMessages)?.role === 'assistant') { if (first(userMessages)?.role === 'assistant') {

2
src/renderer/src/providers/OpenAIProvider.ts
View File

@ -117,7 +117,7 @@ export default class OpenAIProvider extends BaseProvider {
const systemMessage = assistant.prompt ? { role: 'system', content: assistant.prompt } : undefined const systemMessage = assistant.prompt ? { role: 'system', content: assistant.prompt } : undefined
const userMessages: ChatCompletionMessageParam[] = [] const userMessages: ChatCompletionMessageParam[] = []
const _messages = filterContextMessages(takeRight(messages, contextCount)) const _messages = filterContextMessages(takeRight(messages, contextCount + 1))
onFilterMessages(_messages) onFilterMessages(_messages)
for (const message of _messages) { for (const message of _messages) {