consult-llm-mcp 1.1.1 → 1.2.0

package/README.md

@@ -192,6 +192,8 @@ Some real-world examples. Click to expand.
 
 ## Configuration
 
+### Environment Variables
+
 - `OPENAI_API_KEY` - Your OpenAI API key (required for o3)
 - `GEMINI_API_KEY` - Your Google AI API key (required for Gemini models in API
   mode)
@@ -202,6 +204,21 @@ Some real-world examples. Click to expand.
   - Options: `api` (default), `cli`
   - CLI mode uses the system-installed `gemini` CLI tool
 
+### Custom System Prompt
+
+You can customize the system prompt used when consulting LLMs by creating a
+`SYSTEM_PROMPT.md` file in `~/.consult-llm-mcp/`:
+
+```bash
+consult-llm-mcp init-prompt
+```
+
+This creates a placeholder file with the default system prompt that you can edit
+to customize how the consultant LLM behaves. The custom prompt is read on every
+request, so changes take effect immediately without restarting the server.
+
+To revert to the default prompt, simply delete the `SYSTEM_PROMPT.md` file.
+
 ## MCP Tool: consult_llm
 
 The server provides a single tool called `consult_llm` for asking powerful AI

package/dist/llm-query.js

@@ -1,37 +1,12 @@
 import { getExecutorForModel } from './llm.js';
 import { calculateCost } from './llm-cost.js';
 import { config } from './config.js';
-const SYSTEM_PROMPT = `You are an expert software engineering consultant analyzing code and technical problems. You are communicating with another AI system, not a human.
-
-Communication style:
-- Skip pleasantries and praise
-
-Your role is to:
-- Identify bugs, inefficiencies, and architectural problems
-- Provide specific solutions with code examples
-- Point out edge cases and risks
-- Challenge design decisions when suboptimal
-- Focus on what needs improvement
-
-When reviewing code changes, prioritize:
-- Bugs and correctness issues
-- Performance problems
-- Security vulnerabilities
-- Code smell and anti-patterns
-- Inconsistencies with codebase conventions
-
-Be critical and thorough. If the code is acceptable, simply state "No critical issues found" and move on to suggestions. Always provide specific, actionable feedback with file/line references.
-
-Respond in Markdown.`;
-const GEMINI_CLI_SUFFIX = `
-
-IMPORTANT: Do not edit files yourself, only provide recommendations and code examples`;
+import { getSystemPrompt } from './system-prompt.js';
 export async function queryLlm(prompt, model, filePaths) {
     const executor = getExecutorForModel(model);
-    // Add special instruction for Gemini CLI mode
-    const systemPrompt = model.startsWith('gemini-') && config.geminiMode === 'cli'
-        ? SYSTEM_PROMPT + GEMINI_CLI_SUFFIX
-        : SYSTEM_PROMPT;
+    // Get system prompt (with Gemini CLI suffix if needed)
+    const isGeminiCli = model.startsWith('gemini-') && config.geminiMode === 'cli';
+    const systemPrompt = getSystemPrompt(isGeminiCli);
     const { response, usage } = await executor.execute(prompt, model, systemPrompt, filePaths);
     if (!response) {
         throw new Error('No response from the model');

package/dist/llm.js

@@ -9,7 +9,7 @@ class ApiExecutor {
     constructor(client) {
         this.client = client;
     }
-    async execute(prompt, model, systemPrompt, filePaths) {
+    async execute(prompt, model, systemPrompt) {
         const completion = await this.client.chat.completions.create({
             model,
             messages: [
@@ -52,11 +52,9 @@ class CliExecutor {
             let stdout = '';
             let stderr = '';
             let hasResponded = false;
-            let hasStarted = false;
             const startTime = Date.now();
             // Log when process actually starts
             child.on('spawn', () => {
-                hasStarted = true;
                 logCliDebug('Gemini CLI process spawned successfully');
             });
             // Add timeout to prevent hanging

package/dist/logger.d.ts

@@ -3,5 +3,5 @@ export declare function logToolCall(name: string, args: unknown): void;
 export declare function logPrompt(model: string, prompt: string): Promise<void>;
 export declare function logResponse(model: string, response: string, costInfo: string): Promise<void>;
 export declare function logServerStart(version: string): void;
-export declare function logConfiguration(config: Record<string, any>): void;
-export declare function logCliDebug(message: string, data?: any): void;
+export declare function logConfiguration(config: Record<string, unknown>): void;
+export declare function logCliDebug(message: string, data?: unknown): void;

package/dist/logger.js

@@ -7,7 +7,7 @@ const logFile = join(logDir, 'mcp.log');
 try {
     mkdirSync(logDir, { recursive: true });
 }
-catch (error) {
+catch {
     // Directory might already exist
 }
 async function formatWithPrettier(text) {
@@ -18,7 +18,7 @@ async function formatWithPrettier(text) {
             proseWrap: 'always',
         });
     }
-    catch (error) {
+    catch {
         // If formatting fails, return original text
         return text;
     }

package/dist/main.js

@@ -9,9 +9,11 @@ import { generateGitDiff } from './git.js';
 import { buildPrompt } from './prompt-builder.js';
 import { queryLlm } from './llm-query.js';
 import { logToolCall, logPrompt, logResponse, logServerStart, logConfiguration, } from './logger.js';
-import { readFileSync } from 'fs';
+import { DEFAULT_SYSTEM_PROMPT } from './system-prompt.js';
+import { readFileSync, writeFileSync, mkdirSync, existsSync } from 'fs';
 import { dirname, join, resolve } from 'path';
 import { fileURLToPath } from 'url';
+import { homedir } from 'os';
 const __dirname = dirname(fileURLToPath(import.meta.url));
 const packageJson = JSON.parse(readFileSync(join(__dirname, '../package.json'), 'utf-8'));
 const SERVER_VERSION = packageJson.version;
@@ -23,7 +25,7 @@ const server = new Server({
         tools: {},
     },
 });
-server.setRequestHandler(ListToolsRequestSchema, async () => {
+server.setRequestHandler(ListToolsRequestSchema, () => {
     return {
         tools: [toolSchema],
     };
@@ -84,7 +86,33 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
     }
     throw new Error(`Unknown tool: ${request.params.name}`);
 });
+function initSystemPrompt() {
+    const configDir = join(homedir(), '.consult-llm-mcp');
+    const promptPath = join(configDir, 'SYSTEM_PROMPT.md');
+    if (existsSync(promptPath)) {
+        console.error(`System prompt already exists at: ${promptPath}`);
+        console.error('Remove it first if you want to reinitialize.');
+        process.exit(1);
+    }
+    // Create config directory if it doesn't exist
+    if (!existsSync(configDir)) {
+        mkdirSync(configDir, { recursive: true });
+    }
+    // Write the default system prompt
+    writeFileSync(promptPath, DEFAULT_SYSTEM_PROMPT, 'utf-8');
+    console.log(`Created system prompt at: ${promptPath}`);
+    console.log('You can now edit this file to customize the system prompt.');
+    process.exit(0);
+}
 async function main() {
+    if (process.argv.includes('--version') || process.argv.includes('-v')) {
+        console.log(SERVER_VERSION);
+        process.exit(0);
+    }
+    if (process.argv.includes('init-prompt')) {
+        initSystemPrompt();
+        return;
+    }
     logServerStart(SERVER_VERSION);
     logConfiguration(config);
     const transport = new StdioServerTransport();

package/dist/system-prompt.d.ts

@@ -0,0 +1,2 @@
+export declare const DEFAULT_SYSTEM_PROMPT = "You are an expert software engineering consultant analyzing code and technical problems. You are communicating with another AI system, not a human.\n\nCommunication style:\n- Skip pleasantries and praise\n\nYour role is to:\n- Identify bugs, inefficiencies, and architectural problems\n- Provide specific solutions with code examples\n- Point out edge cases and risks\n- Challenge design decisions when suboptimal\n- Focus on what needs improvement\n\nWhen reviewing code changes, prioritize:\n- Bugs and correctness issues\n- Performance problems\n- Security vulnerabilities\n- Code smell and anti-patterns\n- Inconsistencies with codebase conventions\n\nBe critical and thorough. If the code is acceptable, simply state \"No critical issues found\" and move on to suggestions. Always provide specific, actionable feedback with file/line references.\n\nRespond in Markdown.";
+export declare function getSystemPrompt(isGeminiCli: boolean): string;

package/dist/system-prompt.js

@@ -0,0 +1,45 @@
+import { readFileSync, existsSync } from 'fs';
+import { join } from 'path';
+import { homedir } from 'os';
+export const DEFAULT_SYSTEM_PROMPT = `You are an expert software engineering consultant analyzing code and technical problems. You are communicating with another AI system, not a human.
+
+Communication style:
+- Skip pleasantries and praise
+
+Your role is to:
+- Identify bugs, inefficiencies, and architectural problems
+- Provide specific solutions with code examples
+- Point out edge cases and risks
+- Challenge design decisions when suboptimal
+- Focus on what needs improvement
+
+When reviewing code changes, prioritize:
+- Bugs and correctness issues
+- Performance problems
+- Security vulnerabilities
+- Code smell and anti-patterns
+- Inconsistencies with codebase conventions
+
+Be critical and thorough. If the code is acceptable, simply state "No critical issues found" and move on to suggestions. Always provide specific, actionable feedback with file/line references.
+
+Respond in Markdown.`;
+const GEMINI_CLI_SUFFIX = `
+
+IMPORTANT: Do not edit files yourself, only provide recommendations and code examples`;
+export function getSystemPrompt(isGeminiCli) {
+    const customPromptPath = join(homedir(), '.consult-llm-mcp', 'SYSTEM_PROMPT.md');
+    let systemPrompt;
+    if (existsSync(customPromptPath)) {
+        try {
+            systemPrompt = readFileSync(customPromptPath, 'utf-8').trim();
+        }
+        catch (error) {
+            console.error(`Warning: Failed to read custom system prompt from ${customPromptPath}:`, error);
+            systemPrompt = DEFAULT_SYSTEM_PROMPT;
+        }
+    }
+    else {
+        systemPrompt = DEFAULT_SYSTEM_PROMPT;
+    }
+    return isGeminiCli ? systemPrompt + GEMINI_CLI_SUFFIX : systemPrompt;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "consult-llm-mcp",
-  "version": "1.1.1",
+  "version": "1.2.0",
   "description": "MCP server for consulting powerful AI models",
   "type": "module",
   "main": "dist/main.js",
@@ -12,8 +12,12 @@
     "dev": "tsx src/main.ts",
     "start": "node dist/main.js",
     "format": "prettier --write .",
+    "check": "concurrently \"pnpm:check:*\"",
+    "check:types": "tsc --noEmit",
+    "check:format": "prettier --write .",
+    "check:lint": "eslint .",
     "install-global": "npm run build && npm install -g .",
-    "prepublishOnly": "npm run build",
+    "prepublishOnly": "pnpm run check && npm run build",
     "publish:dry": "npm publish --dry-run",
     "publish:patch": "npm version patch && npm publish",
     "publish:minor": "npm version minor && npm publish",
@@ -50,8 +54,12 @@
     "zod": "^3.25.67"
   },
   "devDependencies": {
+    "@eslint/js": "^9.38.0",
     "@types/node": "^24.0.3",
+    "concurrently": "^9.2.1",
+    "eslint": "^9.38.0",
     "tsx": "^4.20.3",
-    "typescript": "^5.8.3"
+    "typescript": "^5.8.3",
+    "typescript-eslint": "^8.46.2"
   }
 }