@koush/chatsh 1.0.12 → 1.0.15

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Koushik Dutta
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -2,34 +2,23 @@
 
 A terminal-based shell assistant that connects your shell session to an LLM for context-aware help.
 
-## What it does
+## Quick Start
 
-Chatsh wraps your shell session and maintains a transcript of all terminal activity. Use the `help` command to query an LLM with context about what you're doing.
-
-```
-$ help how do I find large files?
-```
-
-The LLM receives your entire shell transcript and can provide context-aware assistance.
-
-## Features
+```bash
+$ npx @koush/chatsh
+$ ls /nonexistent
+ls: /nonexistent: No such file or directory
 
-- **Shell wrapper** - Wraps zsh/bash/fish in a PTY session
-- **LLM integration** - Multiple provider support (OpenAI, Anthropic, Google, OpenAI-compatible)
-- **Transcript tracking** - Maintains full terminal history
-- **Smart clear detection** - Resets transcript on terminal clear
+$ help why did this command fail
 
-## Installation
+The command failed because the directory /nonexistent does not exist. 
+The ls command lists directory contents, but it cannot find a path 
+that doesn't exist on your filesystem. Try ls without arguments to 
+see your current directory, or use ls / to list the root directory.
 
-```bash
-npm install -g @koush/chatsh
 ```
 
-Or run directly with:
-
-```bash
-npx @koush/chatsh
-```
+The LLM sees your entire terminal transcript and provides context-aware assistance.
 
 ## Configuration
 
@@ -75,20 +64,12 @@ Create `~/.chatsh/chatsh.jsonc`:
 }
 ```
 
-## Usage
-
-Start chatsh:
-
-```bash
-chatsh
-```
-
-Use your shell normally. When you need help:
+## Features
 
-```bash
-$ help what does this error mean?
-$ help how do I grep recursively?
-```
+- Wraps zsh/bash/fish in a PTY session
+- Multiple LLM providers (OpenAI, Anthropic, Google, OpenAI-compatible)
+- Maintains full terminal history
+- Resets transcript on terminal clear
 
 ## How it works
 

package/dist/main.js

@@ -4,12 +4,48 @@ import { once } from 'node:events';
 import { readFileSync, existsSync } from 'node:fs';
 import { homedir } from 'node:os';
 import { join } from 'node:path';
+import { fileURLToPath } from 'node:url';
 import * as jsonc from 'jsonc-parser';
 import { createOpenAI } from '@ai-sdk/openai';
 import { createAnthropic } from '@ai-sdk/anthropic';
 import { createGoogleGenerativeAI } from '@ai-sdk/google';
 import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
 import { streamText } from 'ai';
+const __filename = fileURLToPath(import.meta.url);
+function createEscapeConverter() {
+    let buffer = '';
+    const patterns = ['\\033[', '\\x1b[', '\\e[', '\\u001b['];
+    return {
+        convert(chunk) {
+            buffer += chunk;
+            let result = '';
+            for (const pattern of patterns) {
+                while (buffer.includes(pattern)) {
+                    const idx = buffer.indexOf(pattern);
+                    result += buffer.substring(0, idx);
+                    result += '\x1b[';
+                    buffer = buffer.substring(idx + pattern.length);
+                }
+            }
+            // Keep potential partial matches in buffer
+            let safeIdx = buffer.length;
+            for (const pattern of patterns) {
+                const partialMatch = pattern.substring(0, pattern.length - 1);
+                if (buffer.endsWith(partialMatch)) {
+                    safeIdx = Math.min(safeIdx, buffer.length - partialMatch.length);
+                }
+            }
+            result += buffer.substring(0, safeIdx);
+            buffer = buffer.substring(safeIdx);
+            return result;
+        },
+        flush() {
+            const remaining = buffer;
+            buffer = '';
+            return remaining;
+        }
+    };
+}
 const CONFIG_PATH = join(homedir(), '.chatsh', 'chatsh.jsonc');
 function loadConfig() {
     if (!existsSync(CONFIG_PATH)) {
@@ -148,14 +184,16 @@ async function main() {
                 try {
                     const result = streamText({
                         model,
-                        system: 'You are a helpful assistant running in a terminal session. You receive a shell transcript from the user and an associated user query, if any.\n\nCRITICAL: Do NOT use any markdown formatting whatsoever. This means:\n- No bullet points (- or *)\n- No backticks (`) for code\n- No headers (#)\n- No bold/italic markers (** or _)\n- No markdown links\n\nUse plain text only. If you need to emphasize something or format code examples, use ANSI escape codes instead (e.g., \\x1b[1m for bold, \\x1b[32m for green, \\x1b[0m to reset).\n\nHelp the user based on the transcript context.',
+                        system: 'You are a helpful assistant running in a terminal session. You receive a shell transcript from the user and an associated user query, if any.\n\nCRITICAL: Do NOT use any markdown formatting whatsoever. This means:\n- No bullet points (- or *)\n- No backticks (`) for code\n- No headers (#)\n- No bold/italic markers (** or _)\n- No markdown links\n\nUse plain text only. When you need to emphasize something or format output, use ANSI escape codes. They WILL be converted and rendered properly in the terminal:\n\nWrite escape codes using \\033 notation:\n- \\033[1m for bold\n- \\033[32m for green\n- \\033[31m for red\n- \\033[33m for yellow\n- \\033[34m for blue\n- \\033[36m for cyan\n- \\033[35m for magenta\n- \\033[0m to reset formatting\n\nExample: To print "Hello" in green, write: \\033[32mHello\\033[0m\n\nThe escape codes will be automatically converted and rendered as colors in the terminal.\n\nHelp the user based on the transcript context.',
                         prompt: `${transcript}\n\n${body}`
                     });
                     res.writeHead(200, { 'Content-Type': 'text/plain' });
                     res.write('\n\n');
+                    const converter = createEscapeConverter();
                     for await (const chunk of result.textStream) {
-                        res.write(chunk);
+                        res.write(converter.convert(chunk));
                     }
+                    res.write(converter.flush());
                     res.write('\n\n');
                     res.end();
                 }
@@ -180,7 +218,13 @@ async function main() {
         cols: process.stdout.columns || 80,
         rows: process.stdout.rows || 24,
         cwd: process.cwd(),
-        env: { ...process.env, CHATSH_PORT: String(port) }
+        env: {
+            ...process.env,
+            CHATSH_PORT: String(port),
+            CHATSH_NODE: process.execPath,
+            CHATSH_SCRIPT: __filename,
+            CHATSH_TS: __filename.endsWith('.ts') ? '--experimental-strip-types' : ''
+        }
     });
     process.stdin.setRawMode(true);
     process.stdin.resume();
@@ -200,7 +244,7 @@ async function main() {
         }
         process.stdout.write(data);
     });
-    ptyProcess.write('help() { curl -s -X POST -d "$*" http://localhost:$CHATSH_PORT }\n');
+    ptyProcess.write('help() { "$CHATSH_NODE" --no-warnings $CHATSH_TS "$CHATSH_SCRIPT" --help "$*"; }\n');
     if (shell.includes('zsh')) {
         ptyProcess.write('bindkey "^R" history-incremental-search-backward\n');
     }
@@ -214,4 +258,23 @@ async function main() {
         ptyProcess.resize(process.stdout.columns || 80, process.stdout.rows || 24);
     });
 }
-main();
+// Handle --help flag for shell help command
+if (process.argv[2] === '--help') {
+    const query = process.argv.slice(3).join(' ');
+    const port = process.env.CHATSH_PORT;
+    if (!port) {
+        console.error('CHATSH_PORT not set');
+        process.exit(1);
+    }
+    const response = await fetch(`http://localhost:${port}`, {
+        method: 'POST',
+        body: query
+    });
+    for await (const chunk of response.body) {
+        process.stdout.write(chunk);
+    }
+    process.exit(0);
+}
+else {
+    main();
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@koush/chatsh",
-  "version": "1.0.12",
+  "version": "1.0.15",
   "type": "module",
   "main": "dist/main.js",
   "bin": {
@@ -14,8 +14,8 @@
     "start": "node --experimental-strip-types src/main.ts",
     "prepublishOnly": "npm version patch && npm run build"
   },
-  "author": "",
-  "license": "ISC",
+  "author": "Koushik Dutta",
+  "license": "MIT",
   "description": "",
   "devDependencies": {
     "@types/node": "^20.10.0",