From c491329d4f3c03c4933b3ff76060d433a1adc72c Mon Sep 17 00:00:00 2001 From: Danil Nikolaev Date: Fri, 16 Jan 2026 16:13:10 +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 +- src/common/environment.ts | 4 ++++ 4 files changed, 7 insertions(+), 3 deletions(-) diff --git a/src/commands/gemini-chat.ts b/src/commands/gemini-chat.ts index ab8d586..d591b6d 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: `MESSAGE FROM USER "${part.name}":\n` + extractText(part.content, "/gemini") + content: (Environment.USE_NAMES_IN_PROMPT ? `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 fac48b0..cc68c6e 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: `MESSAGE FROM USER "${part.name}":\n` + extractText(part.content, Environment.BOT_PREFIX), + text: (Environment.USE_NAMES_IN_PROMPT ? `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 6a47de7..89cd018 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: `MESSAGE FROM USER "${part.name}":\n` + extractText(part.content, Environment.BOT_PREFIX), + content: (Environment.USE_NAMES_IN_PROMPT ? `MESSAGE FROM USER "${part.name}":\n` : "") + extractText(part.content, Environment.BOT_PREFIX), images: imageFilePath && i === 0 ? [imageFilePath] : null }; }); diff --git a/src/common/environment.ts b/src/common/environment.ts index 5c88c77..8791f37 100644 --- a/src/common/environment.ts +++ b/src/common/environment.ts @@ -17,6 +17,8 @@ export class Environment { static USE_DAD: boolean; static USE_FU: boolean; + static USE_NAMES_IN_PROMPT: boolean; + static OLLAMA_MODEL?: string; static OLLAMA_ADDRESS?: string; static OLLAMA_API_KEY?: string; @@ -43,6 +45,8 @@ export class Environment { Environment.USE_DAD = process.env.USE_DAD == "true"; Environment.USE_FU = process.env.USE_FU == "true"; + Environment.USE_NAMES_IN_PROMPT = process.env.USE_NAMES_IN_PROMPT == "true"; + Environment.OLLAMA_MODEL = process.env.OLLAMA_MODEL; Environment.OLLAMA_ADDRESS = process.env.OLLAMA_ADDRESS; Environment.OLLAMA_API_KEY = process.env.OLLAMA_API_KEY;