From 97582881ea7cb32412eaea3cce30311872f89f07 Mon Sep 17 00:00:00 2001 From: Danil Nikolaev Date: Fri, 16 Jan 2026 16:14:40 +0300 Subject: [PATCH] optional usage of names in prompt --- src/commands/gemini-chat.ts | 2 +- src/commands/mistral-chat.ts | 2 +- src/commands/ollama-chat.ts | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/commands/gemini-chat.ts b/src/commands/gemini-chat.ts index d591b6d..3dcf736 100644 --- a/src/commands/gemini-chat.ts +++ b/src/commands/gemini-chat.ts @@ -40,7 +40,7 @@ export class GeminiChat extends ChatCommand { const chatMessages = messageParts.map(part => { return { role: part.bot ? "assistant" : "user", - content: (Environment.USE_NAMES_IN_PROMPT ? `MESSAGE FROM USER "${part.name}":\n` : "") + extractText(part.content, "/gemini") + content: (Environment.USE_NAMES_IN_PROMPT && !part.bot ? `MESSAGE FROM USER "${part.name}":\n` : "") + extractText(part.content, "/gemini") }; }); chatMessages.reverse(); diff --git a/src/commands/mistral-chat.ts b/src/commands/mistral-chat.ts index cc68c6e..f36f3a6 100644 --- a/src/commands/mistral-chat.ts +++ b/src/commands/mistral-chat.ts @@ -69,7 +69,7 @@ export class MistralChat extends ChatCommand { const content = []; content.push({ type: "text", - text: (Environment.USE_NAMES_IN_PROMPT ? `MESSAGE FROM USER "${part.name}":\n` : "") + extractText(part.content, Environment.BOT_PREFIX), + text: (Environment.USE_NAMES_IN_PROMPT && !part.bot ? `MESSAGE FROM USER "${part.name}":\n` : "") + extractText(part.content, Environment.BOT_PREFIX), }); if (imageFilePath && i === 0) { diff --git a/src/commands/ollama-chat.ts b/src/commands/ollama-chat.ts index 89cd018..4b76b85 100644 --- a/src/commands/ollama-chat.ts +++ b/src/commands/ollama-chat.ts @@ -63,7 +63,7 @@ export class OllamaChat extends ChatCommand { const chatMessages = messageParts.map((part, i) => { return { role: part.bot ? "assistant" : "user", - content: (Environment.USE_NAMES_IN_PROMPT ? `MESSAGE FROM USER "${part.name}":\n` : "") + extractText(part.content, Environment.BOT_PREFIX), + content: (Environment.USE_NAMES_IN_PROMPT && !part.bot ? `MESSAGE FROM USER "${part.name}":\n` : "") + extractText(part.content, Environment.BOT_PREFIX), images: imageFilePath && i === 0 ? [imageFilePath] : null }; });