gaunt-sloth-assistant 0.7.0 → 0.7.2

package/it.js

@@ -0,0 +1,15 @@
+import { execSync } from 'node:child_process';
+
+// Script to launch integration tests
+
+execSync('node integration-tests/setup-config.js ' + process.argv[2], {
+  stdio: [process.stdin, process.stdout, process.stderr],
+});
+try {
+  const test = process.argv[3] ? ` ${process.argv[3]}` : '';
+  execSync('vitest run --config vitest-it.config.ts' + test, {
+    stdio: [process.stdin, process.stdout, process.stderr],
+  });
+} catch {
+  process.exit(1);
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gaunt-sloth-assistant",
-  "version": "0.7.0",
+  "version": "0.7.2",
   "description": "",
   "license": "MIT",
   "author": "Andrew Kondratev",
@@ -14,9 +14,7 @@
   "scripts": {
     "build": "tsc",
     "test": "npm run build && vitest run",
-    "it-groq": "node integration-tests/setup-config.js groq && npm run build && vitest run --config vitest-it.config.ts",
-    "it-anthropic": "node integration-tests/setup-config.js anthropic && npm run build && vitest run --config vitest-it.config.ts",
-    "it-vertexai": "node integration-tests/setup-config.js vertexai && npm run build && vitest run --config vitest-it.config.ts",
+    "it": "npm run build && node it.js",
     "lint": "eslint . --ext .js,.ts",
     "format": "prettier --write 'src/**/*.{js,ts}'",
     "prepare": "npm run build"
@@ -26,11 +24,12 @@
     "gth": "index.js"
   },
   "dependencies": {
-    "@langchain/anthropic": "^0.3.22",
-    "@langchain/core": "^0.3.58",
-    "@langchain/google-vertexai": "^0.2.10",
+    "@langchain/anthropic": "^0.3.23",
+    "@langchain/core": "^0.3.61",
+    "@langchain/deepseek": "^0.0.2",
+    "@langchain/google-vertexai": "^0.2.14",
     "@langchain/groq": "^0.2.3",
-    "@langchain/langgraph": "^0.3.1",
+    "@langchain/langgraph": "^0.3.5",
     "@langchain/mcp-adapters": "^0.5.2",
     "chalk": "^5.4.1",
     "commander": "^14.0.0",
@@ -40,17 +39,17 @@
   "devDependencies": {
     "@eslint/js": "^9.26.0",
     "@types/diff": "^7.0.2",
-    "@types/node": "^24.0.2",
+    "@types/node": "^24.0.7",
     "@types/uuid": "^10.0.0",
-    "@typescript-eslint/eslint-plugin": "^8.34.0",
-    "@typescript-eslint/parser": "^8.34.0",
-    "eslint": "^9.28.0",
+    "@typescript-eslint/eslint-plugin": "^8.35.0",
+    "@typescript-eslint/parser": "^8.35.0",
+    "eslint": "^9.30.0",
     "eslint-config-prettier": "^10.1.5",
-    "eslint-plugin-prettier": "^5.4.1",
+    "eslint-plugin-prettier": "^5.5.1",
     "globals": "^16.2.0",
     "prettier": "3.6.0",
     "typescript": "^5.8.3",
-    "vitest": "^3.2.3"
+    "vitest": "^3.2.4"
   },
   "imports": {
     "#src/*.js": "./dist/*.js"

package/src/config.ts

@@ -76,7 +76,7 @@ export interface LLMConfig extends Record<string, unknown> {
   model: string;
 }
 
-export const availableDefaultConfigs = ['vertexai', 'anthropic', 'groq'] as const;
+export const availableDefaultConfigs = ['vertexai', 'anthropic', 'groq', 'deepseek'] as const;
 export type ConfigType = (typeof availableDefaultConfigs)[number];
 
 export const DEFAULT_CONFIG: Partial<SlothConfig> = {
@@ -199,7 +199,7 @@ export async function tryJsonConfig(jsonConfig: RawSlothConfig): Promise<SlothCo
       // Get the configuration for the specific LLM type
       const llmConfig = jsonConfig.llm;
       // Import the appropriate config module
-      const configModule = await import(`./configs/${llmType}.js`);
+      const configModule = await import(`./presets/${llmType}.js`);
       if (configModule.processJsonConfig) {
         const llm = (await configModule.processJsonConfig(llmConfig)) as BaseChatModel;
         return mergeRawConfig(jsonConfig, llm);
@@ -237,7 +237,7 @@ export async function createProjectConfig(configType: string): Promise<void> {
   displayWarning(`Make sure you add as much detail as possible to your ${PROJECT_GUIDELINES}.\n`);
 
   displayInfo(`Creating project config for ${configType}`);
-  const vendorConfig = await import(`./configs/${configType}.js`);
+  const vendorConfig = await import(`./presets/${configType}.js`);
   vendorConfig.init(getGslothConfigWritePath(USER_PROJECT_CONFIG_JSON));
 }
 

package/src/core/Invocation.ts

@@ -1,16 +1,14 @@
 import type { Message } from '#src/modules/types.js';
 import { AIMessage, isAIMessage } from '@langchain/core/messages';
-import { SlothConfig } from '#src/config.js';
+import { getDefaultTools, SlothConfig } from '#src/config.js';
 import type { Connection } from '@langchain/mcp-adapters';
 import { MultiServerMCPClient } from '@langchain/mcp-adapters';
 import { createReactAgent } from '@langchain/langgraph/prebuilt';
 import { BaseCheckpointSaver, CompiledStateGraph } from '@langchain/langgraph';
-import { ProgressIndicator } from '#src/utils.js';
+import { formatToolCalls, ProgressIndicator } from '#src/utils.js';
 import { type RunnableConfig } from '@langchain/core/runnables';
 import { ToolCall } from '@langchain/core/messages/tool';
-import { StatusLevel } from '#src/core/types.js';
-import { formatToolCalls } from '#src/utils.js';
-import { getDefaultTools } from '#src/config.js';
+import { GthCommand, StatusLevel } from '#src/core/types.js';
 import { BaseToolkit, StructuredToolInterface } from '@langchain/core/tools';
 
 export type StatusUpdateCallback = (level: StatusLevel, message: string) => void;
@@ -32,7 +30,7 @@ export class Invocation {
   }
 
   async init(
-    command: 'ask' | 'pr' | 'review' | 'chat' | 'code' | undefined,
+    command: GthCommand | undefined,
     config: SlothConfig,
     checkpointSaver?: BaseCheckpointSaver | undefined
   ): Promise<void> {
@@ -41,23 +39,14 @@ export class Invocation {
     }
 
     // Merge command-specific filesystem config if provided
-    let effectiveConfig = config;
-    if (command && config.commands?.[command]?.filesystem !== undefined) {
-      effectiveConfig = {
-        ...config,
-        filesystem: config.commands[command].filesystem!,
-      };
-    }
-
-    this.config = effectiveConfig;
-    this.mcpClient = this.getMcpClient(effectiveConfig);
+    this.config = this.getEffectiveConfig(config, command);
+    this.mcpClient = this.getMcpClient(this.config);
 
     // Get default filesystem tools (filtered based on config)
-    const defaultTools = getDefaultTools(effectiveConfig.filesystem || 'none');
+    const defaultTools = getDefaultTools(this.config.filesystem || 'none');
 
     // Get user config tools
-    const configTools = effectiveConfig.tools || [];
-    const flattenedConfigTools = this.extractAndFlattenTools(configTools);
+    const flattenedConfigTools = this.extractAndFlattenTools(this.config.tools || []);
 
     // Get MCP tools
     const mcpTools = (await this.mcpClient?.getTools()) ?? [];
@@ -75,12 +64,26 @@ export class Invocation {
 
     // Create the React agent
     this.agent = createReactAgent({
-      llm: config.llm,
+      llm: this.config.llm,
       tools,
       checkpointSaver,
     });
   }
 
+  getEffectiveConfig(config: SlothConfig, command: GthCommand | undefined): SlothConfig {
+    const supportsTools = !!config.llm.bindTools;
+    if (!supportsTools) {
+      this.statusUpdate('warning', 'Model does not seem to support tools.');
+    }
+    return {
+      ...config,
+      filesystem:
+        command && config.commands?.[command]?.filesystem !== undefined
+          ? config.commands[command].filesystem!
+          : config.filesystem,
+    };
+  }
+
   async invoke(messages: Message[], runConfig?: RunnableConfig): Promise<string> {
     if (!this.agent || !this.config) {
       throw new Error('Invocation not initialized. Call init() first.');
@@ -89,7 +92,26 @@ export class Invocation {
     // Run the agent
     try {
       const output = { aiMessage: '' };
-      if (!this.config.streamOutput) {
+      if (this.config.streamOutput) {
+        // Streaming output
+        this.statusUpdate('info', '\nThinking...\n');
+        const stream = await this.agent.stream(
+          { messages },
+          { ...runConfig, streamMode: 'messages' }
+        );
+
+        for await (const [chunk, _metadata] of stream) {
+          if (isAIMessage(chunk)) {
+            this.statusUpdate('stream', chunk.text as string);
+            output.aiMessage += chunk.text;
+            const toolCalls = chunk.tool_calls?.filter((tc) => tc.name);
+            if (toolCalls && toolCalls.length > 0) {
+              this.statusUpdate('info', `\nUsed tools: ${formatToolCalls(toolCalls)}`);
+            }
+          }
+        }
+      } else {
+        // Not streaming output
         const progress = new ProgressIndicator('Thinking.');
         try {
           const response = await this.agent.invoke({ messages }, runConfig);
@@ -107,23 +129,6 @@ export class Invocation {
           progress.stop();
         }
         this.statusUpdate('display', output.aiMessage);
-      } else {
-        this.statusUpdate('info', '\nThinking...\n');
-        const stream = await this.agent.stream(
-          { messages },
-          { ...runConfig, streamMode: 'messages' }
-        );
-
-        for await (const [chunk, _metadata] of stream) {
-          if (isAIMessage(chunk)) {
-            this.statusUpdate('stream', chunk.text as string);
-            output.aiMessage += chunk.text;
-            const toolCalls = chunk.tool_calls?.filter((tc) => tc.name);
-            if (toolCalls && toolCalls.length > 0) {
-              this.statusUpdate('info', `Used tools: ${formatToolCalls(toolCalls)}`);
-            }
-          }
-        }
       }
 
       return output.aiMessage;

package/src/core/types.ts

@@ -1 +1,2 @@
 export type StatusLevel = 'info' | 'warning' | 'error' | 'success' | 'debug' | 'display' | 'stream';
+export type GthCommand = 'ask' | 'pr' | 'review' | 'chat' | 'code';

package/src/{configs → presets}/anthropic.ts

@@ -39,5 +39,8 @@ export function init(configFileName: string): void {
   }
 
   writeFileIfNotExistsWithMessages(configFileName, jsonContent);
-  displayWarning(`You need to update your ${configFileName} to add your Anthropic API key.`);
+  displayWarning(
+    `You need to update your ${configFileName} to add your Anthropic API key, ` +
+      'or define ANTHROPIC_API_KEY environment variable.'
+  );
 }

package/src/presets/deepseek.ts

@@ -0,0 +1,46 @@
+import path from 'node:path';
+import { displayWarning } from '#src/consoleUtils.js';
+import { env, getCurrentDir } from '#src/systemUtils.js';
+import { writeFileIfNotExistsWithMessages } from '#src/utils.js';
+import type {
+  BaseChatModel,
+  BaseChatModelParams,
+} from '@langchain/core/language_models/chat_models';
+import { ChatDeepSeekInput } from '@langchain/deepseek';
+
+// Function to process JSON config and create DeepSeek LLM instance
+export async function processJsonConfig(
+  llmConfig: ChatDeepSeekInput & BaseChatModelParams
+): Promise<BaseChatModel> {
+  const deepseek = await import('@langchain/deepseek');
+  // Use environment variable if available, otherwise use the config value
+  const deepseekApiKey = env.DEEPSEEK_API_KEY || llmConfig.apiKey;
+  return new deepseek.ChatDeepSeek({
+    ...llmConfig,
+    apiKey: deepseekApiKey,
+    model: llmConfig.model || 'deepseek-reasoner',
+  });
+}
+
+const jsonContent = `{
+  "llm": {
+    "type": "deepseek",
+    "model": "deepseek-reasoner"
+  }
+}`;
+
+export function init(configFileName: string): void {
+  const currentDir = getCurrentDir();
+  path.join(currentDir, configFileName);
+
+  // Determine which content to use based on file extension
+  if (!configFileName.endsWith('.json')) {
+    throw new Error('Only JSON config is supported.');
+  }
+
+  writeFileIfNotExistsWithMessages(configFileName, jsonContent);
+  displayWarning(
+    `You need to update your ${configFileName} to add your DeepSeek API key, ` +
+      'or define DEEPSEEK_API_KEY environment variable.'
+  );
+}

package/src/{configs → presets}/groq.ts

@@ -1,5 +1,5 @@
 import path from 'node:path';
-import { displayInfo, displayWarning } from '#src/consoleUtils.js';
+import { displayWarning } from '#src/consoleUtils.js';
 import { env, getCurrentDir } from '#src/systemUtils.js';
 import { writeFileIfNotExistsWithMessages } from '#src/utils.js';
 import { BaseChatModel } from '@langchain/core/language_models/chat_models';
@@ -20,8 +20,7 @@ export async function processJsonConfig(llmConfig: ChatGroqInput): Promise<BaseC
 const jsonContent = `{
   "llm": {
     "type": "groq",
-    "model": "deepseek-r1-distill-llama-70b",
-    "apiKey": "your-api-key-here"
+    "model": "deepseek-r1-distill-llama-70b"
   }
 }`;
 
@@ -35,8 +34,8 @@ export function init(configFileName: string): void {
   }
 
   writeFileIfNotExistsWithMessages(configFileName, jsonContent);
-  displayInfo(
-    `You can define GROQ_API_KEY environment variable with your Groq API key and it will work with default model.`
+  displayWarning(
+    `You need to edit your ${configFileName} to configure model, ` +
+      'or define GROQ_API_KEY environment variable.'
   );
-  displayWarning(`You need to edit your ${configFileName} to configure model.`);
 }

package/src/{configs → presets}/vertexai.ts

@@ -8,8 +8,7 @@ import type { BaseChatModel } from '@langchain/core/language_models/chat_models'
 const jsonContent = `{
   "llm": {
     "type": "vertexai",
-    "model": "gemini-2.5-pro-preview-05-06",
-    "temperature": 0
+    "model": "gemini-2.5-pro"
   }
 }`;
 
@@ -33,6 +32,6 @@ export async function processJsonConfig(llmConfig: ChatVertexAIInput): Promise<B
   const vertexAi = await import('@langchain/google-vertexai');
   return new vertexAi.ChatVertexAI({
     ...llmConfig,
-    model: llmConfig.model || 'gemini-2.5-pro-preview-05-06',
+    model: llmConfig.model || 'gemini-2.5-pro',
   });
 }

package/src/tools/GthFileSystemToolkit.ts

@@ -1,7 +1,7 @@
 import { BaseToolkit, StructuredToolInterface, tool } from '@langchain/core/tools';
 import { z } from 'zod';
 import fs from 'fs/promises';
-import path from 'path';
+import path from 'node:path';
 import os from 'os';
 import { createTwoFilesPatch } from 'diff';
 import { minimatch } from 'minimatch';

package/vitest-it.config.ts

@@ -5,7 +5,7 @@ export default defineConfig({
     include: ['integration-tests/**/*.it.ts'],
     environment: 'node',
     globals: true,
-    testTimeout: 1000 * 60 * 2,
+    testTimeout: 1000 * 60 * 3,
     maxWorkers: 1,
     fileParallelism: false,
   },

package/dist/configs/groq.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"groq.js","sourceRoot":"","sources":["../../src/configs/groq.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,gCAAgC,EAAE,MAAM,eAAe,CAAC;AAIjE,+DAA+D;AAC/D,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,SAAwB;IAC9D,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;IAC7C,wEAAwE;IACxE,MAAM,UAAU,GAAG,GAAG,CAAC,YAAY,IAAI,SAAS,CAAC,MAAM,CAAC;IACxD,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC;QACvB,GAAG,SAAS;QACZ,MAAM,EAAE,UAAU;QAClB,KAAK,EAAE,SAAS,CAAC,KAAK,IAAI,+BAA+B;KAC1D,CAAC,CAAC;AACL,CAAC;AAED,MAAM,WAAW,GAAG;;;;;;EAMlB,CAAC;AAEH,MAAM,UAAU,IAAI,CAAC,cAAsB;IACzC,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;IAEtC,yDAAyD;IACzD,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,gCAAgC,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;IAC9D,WAAW,CACT,8GAA8G,CAC/G,CAAC;IACF,cAAc,CAAC,yBAAyB,cAAc,sBAAsB,CAAC,CAAC;AAChF,CAAC"}

package/dist/configs/vertexai.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"vertexai.js","sourceRoot":"","sources":["../../src/configs/vertexai.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,gCAAgC,EAAE,MAAM,eAAe,CAAC;AAIjE,MAAM,WAAW,GAAG;;;;;;EAMlB,CAAC;AAEH,MAAM,UAAU,IAAI,CAAC,cAAsB;IACzC,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IACnC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;IAEtC,yDAAyD;IACzD,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,gCAAgC,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;IAC9D,cAAc,CACZ,+GAA+G,CAChH,CAAC;AACJ,CAAC;AAED,mEAAmE;AACnE,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,SAA4B;IAClE,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,4BAA4B,CAAC,CAAC;IAC5D,OAAO,IAAI,QAAQ,CAAC,YAAY,CAAC;QAC/B,GAAG,SAAS;QACZ,KAAK,EAAE,SAAS,CAAC,KAAK,IAAI,8BAA8B;KACzD,CAAC,CAAC;AACL,CAAC"}