@koush/chatsh 1.0.4

package/dist/config.js

@@ -0,0 +1,72 @@
+import { readFileSync, existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import * as jsonc from 'jsonc-parser';
+const CONFIG_PATH = join(homedir(), '.llmsh', 'llmsh.jsonc');
+export function loadConfig() {
+    if (!existsSync(CONFIG_PATH)) {
+        return null;
+    }
+    try {
+        const content = readFileSync(CONFIG_PATH, 'utf-8');
+        const config = jsonc.parse(content);
+        if (!config.provider || !config.model) {
+            return null;
+        }
+        const validProviders = ['openai', 'anthropic', 'google', 'openai-compatible'];
+        if (!validProviders.includes(config.provider)) {
+            return null;
+        }
+        return config;
+    }
+    catch {
+        return null;
+    }
+}
+export function getConfigPath() {
+    return CONFIG_PATH;
+}
+export function showConfigHelp() {
+    console.error(`Error: No valid config found at ${CONFIG_PATH}
+
+Create a config file at ~/.llmsh/llmsh.jsonc with one of the following formats:
+
+// OpenAI
+{
+  "provider": "openai",
+  "model": "gpt-4-turbo",
+  "options": {
+    "apiKey": "sk-..."  // or set OPENAI_API_KEY env var
+  }
+}
+
+// Anthropic (Claude)
+{
+  "provider": "anthropic", 
+  "model": "claude-sonnet-4-5",
+  "options": {
+    "apiKey": "sk-ant-..."  // or set ANTHROPIC_API_KEY env var
+  }
+}
+
+// Google (Gemini)
+{
+  "provider": "google",
+  "model": "gemini-2.5-flash",
+  "options": {
+    "apiKey": "..."  // or set GOOGLE_GENERATIVE_AI_API_KEY env var
+  }
+}
+
+// OpenAI-Compatible (Custom Endpoint)
+{
+  "provider": "openai-compatible",
+  "model": "glm-4.7",
+  "options": {
+    "name": "custom",
+    "baseURL": "http://localhost:8000/v1",
+    "apiKey": "your-api-key"
+  }
+}
+`);
+}

package/dist/main.js

@@ -0,0 +1,213 @@
+import { spawn } from '@scrypted/node-pty';
+import { createServer } from 'node:http';
+import { once } from 'node:events';
+import { readFileSync, existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+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 CONFIG_PATH = join(homedir(), '.llmsh', 'llmsh.jsonc');
+function loadConfig() {
+    if (!existsSync(CONFIG_PATH)) {
+        return null;
+    }
+    try {
+        const content = readFileSync(CONFIG_PATH, 'utf-8');
+        const config = jsonc.parse(content);
+        if (!config.provider || !config.model) {
+            return null;
+        }
+        const validProviders = ['openai', 'anthropic', 'google', 'openai-compatible'];
+        if (!validProviders.includes(config.provider)) {
+            return null;
+        }
+        return config;
+    }
+    catch {
+        return null;
+    }
+}
+function showConfigHelp() {
+    console.error(`Error: No valid config found at ${CONFIG_PATH}
+
+Create a config file at ~/.llmsh/llmsh.jsonc with one of the following formats:
+
+// OpenAI
+{
+  "provider": "openai",
+  "model": "gpt-4-turbo",
+  "options": {
+    "apiKey": "sk-..."  // or set OPENAI_API_KEY env var
+  }
+}
+
+// Anthropic (Claude)
+{
+  "provider": "anthropic", 
+  "model": "claude-sonnet-4-5",
+  "options": {
+    "apiKey": "sk-ant-..."  // or set ANTHROPIC_API_KEY env var
+  }
+}
+
+// Google (Gemini)
+{
+  "provider": "google",
+  "model": "gemini-2.5-flash",
+  "options": {
+    "apiKey": "..."  // or set GOOGLE_GENERATIVE_AI_API_KEY env var
+  }
+}
+
+// OpenAI-Compatible (Custom Endpoint)
+{
+  "provider": "openai-compatible",
+  "model": "glm-4.7",
+  "options": {
+    "name": "custom",
+    "baseURL": "http://localhost:8000/v1",
+    "apiKey": "your-api-key"
+  }
+}
+`);
+}
+function getEnvApiKey(provider) {
+    switch (provider) {
+        case 'openai':
+            return process.env.OPENAI_API_KEY;
+        case 'anthropic':
+            return process.env.ANTHROPIC_API_KEY;
+        case 'google':
+            return process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+        default:
+            return undefined;
+    }
+}
+function createProvider(config) {
+    const { provider, model, options = {} } = config;
+    const apiKey = options.apiKey || getEnvApiKey(provider);
+    const baseURL = options.baseURL;
+    switch (provider) {
+        case 'openai': {
+            return {
+                model: createOpenAI({
+                    apiKey,
+                    baseURL,
+                    organization: options.organization,
+                    project: options.project,
+                })(model),
+            };
+        }
+        case 'anthropic': {
+            return {
+                model: createAnthropic({
+                    apiKey,
+                    baseURL,
+                })(model),
+            };
+        }
+        case 'google': {
+            return {
+                model: createGoogleGenerativeAI({
+                    apiKey,
+                    baseURL,
+                })(model),
+            };
+        }
+        case 'openai-compatible': {
+            return {
+                model: createOpenAICompatible({
+                    name: options.name || 'custom',
+                    apiKey,
+                    baseURL: baseURL || 'http://localhost:8000/v1',
+                })(model),
+            };
+        }
+        default: {
+            throw new Error(`Unknown provider: ${provider}`);
+        }
+    }
+}
+async function main() {
+    const config = loadConfig();
+    if (!config) {
+        showConfigHelp();
+        process.exit(1);
+    }
+    const { model } = createProvider(config);
+    let transcript = '';
+    const server = createServer((req, res) => {
+        if (req.method === 'POST') {
+            let body = '';
+            req.on('data', (chunk) => body += chunk);
+            req.on('end', async () => {
+                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.',
+                        prompt: `${transcript}\n\n${body}`
+                    });
+                    res.writeHead(200, { 'Content-Type': 'text/plain' });
+                    res.write('\n\n');
+                    for await (const chunk of result.textStream) {
+                        res.write(chunk);
+                    }
+                    res.write('\n\n');
+                    res.end();
+                }
+                catch (error) {
+                    res.writeHead(500, { 'Content-Type': 'text/plain' });
+                    res.end('Error: ' + error.message);
+                }
+            });
+        }
+        else {
+            res.writeHead(200, { 'Content-Type': 'text/plain' });
+            res.end(transcript);
+        }
+    });
+    server.listen(0);
+    await once(server, 'listening');
+    const address = server.address();
+    const port = address && typeof address === 'object' ? address.port : 0;
+    const ptyProcess = spawn('zsh', [], {
+        name: process.env.TERM || 'xterm',
+        cols: process.stdout.columns || 80,
+        rows: process.stdout.rows || 24,
+        cwd: process.cwd(),
+        env: { ...process.env, LLMSH_PORT: String(port) }
+    });
+    process.stdin.setRawMode(true);
+    process.stdin.resume();
+    process.stdin.on('data', (data) => {
+        ptyProcess.write(data.toString());
+    });
+    ptyProcess.onData((data) => {
+        const clearSeq = '\x1b[2J';
+        const lastClearIndex = data.lastIndexOf(clearSeq);
+        if (lastClearIndex !== -1) {
+            transcript = '';
+            const afterClear = data.slice(lastClearIndex + clearSeq.length);
+            transcript += afterClear.replace(/^\x1b\[H/, '');
+        }
+        else {
+            transcript += data;
+        }
+        process.stdout.write(data);
+    });
+    ptyProcess.write('help() { curl -s -X POST -d "$*" http://localhost:$LLMSH_PORT }\n');
+    ptyProcess.onExit(({ exitCode }) => {
+        process.exit(exitCode);
+    });
+    process.on('exit', () => {
+        ptyProcess.kill();
+    });
+    process.stdout.on('resize', () => {
+        ptyProcess.resize(process.stdout.columns || 80, process.stdout.rows || 24);
+    });
+}
+main();

package/dist/provider.js

@@ -0,0 +1,61 @@
+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';
+export function createProvider(config) {
+    const { provider, model, options = {} } = config;
+    const apiKey = options.apiKey || getEnvApiKey(provider);
+    const baseURL = options.baseURL;
+    switch (provider) {
+        case 'openai': {
+            return {
+                model: createOpenAI({
+                    apiKey,
+                    baseURL,
+                    organization: options.organization,
+                    project: options.project,
+                })(model),
+            };
+        }
+        case 'anthropic': {
+            return {
+                model: createAnthropic({
+                    apiKey,
+                    baseURL,
+                })(model),
+            };
+        }
+        case 'google': {
+            return {
+                model: createGoogleGenerativeAI({
+                    apiKey,
+                    baseURL,
+                })(model),
+            };
+        }
+        case 'openai-compatible': {
+            return {
+                model: createOpenAICompatible({
+                    name: options.name || 'custom',
+                    apiKey,
+                    baseURL: baseURL || 'http://localhost:8000/v1',
+                })(model),
+            };
+        }
+        default: {
+            throw new Error(`Unknown provider: ${provider}`);
+        }
+    }
+}
+function getEnvApiKey(provider) {
+    switch (provider) {
+        case 'openai':
+            return process.env.OPENAI_API_KEY;
+        case 'anthropic':
+            return process.env.ANTHROPIC_API_KEY;
+        case 'google':
+            return process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+        default:
+            return undefined;
+    }
+}

package/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "@koush/chatsh",
+  "version": "1.0.4",
+  "type": "module",
+  "main": "dist/main.js",
+  "scripts": {
+    "build": "tsc",
+    "start": "node --experimental-strip-types src/main.ts",
+    "prepublishOnly": "npm version patch && npm run build"
+  },
+  "author": "",
+  "license": "ISC",
+  "description": "",
+  "devDependencies": {
+    "@types/node": "^20.10.0",
+    "typescript": "^5.3.0"
+  },
+  "dependencies": {
+    "@ai-sdk/anthropic": "^3.0.58",
+    "@ai-sdk/google": "^3.0.43",
+    "@ai-sdk/openai": "^3.0.41",
+    "@ai-sdk/openai-compatible": "^2.0.35",
+    "@scrypted/node-pty": "^1.0.25",
+    "ai": "^6.0.116",
+    "jsonc-parser": "^3.3.1"
+  }
+}

package/src/@scrypted__node-pty.d.ts

@@ -0,0 +1,49 @@
+declare module '@scrypted/node-pty' {
+  export function spawn(file: string, args: string[] | string, options: IPtyForkOptions | IWindowsPtyForkOptions): IPty;
+
+  export interface IBasePtyForkOptions {
+    name?: string;
+    cols?: number;
+    rows?: number;
+    cwd?: string;
+    env?: { [key: string]: string | undefined };
+    encoding?: string | null;
+    handleFlowControl?: boolean;
+    flowControlPause?: string;
+    flowControlResume?: string;
+  }
+
+  export interface IPtyForkOptions extends IBasePtyForkOptions {
+    uid?: number;
+    gid?: number;
+  }
+
+  export interface IWindowsPtyForkOptions extends IBasePtyForkOptions {
+    useConpty?: boolean;
+    conptyInheritCursor?: boolean;
+  }
+
+  export interface IPty {
+    readonly pid: number;
+    readonly cols: number;
+    readonly rows: number;
+    readonly process: string;
+    handleFlowControl: boolean;
+    readonly onData: IEvent<string>;
+    readonly onExit: IEvent<{ exitCode: number, signal?: number }>;
+    resize(columns: number, rows: number): void;
+    clear(): void;
+    write(data: string): void;
+    kill(signal?: string): void;
+    pause(): void;
+    resume(): void;
+  }
+
+  export interface IDisposable {
+    dispose(): void;
+  }
+
+  export interface IEvent<T> {
+    (listener: (e: T) => any): IDisposable;
+  }
+}

package/src/main.ts

@@ -0,0 +1,252 @@
+import { spawn } from '@scrypted/node-pty';
+import { createServer } from 'node:http';
+import { once } from 'node:events';
+import { readFileSync, existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+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';
+
+type ProviderType = 'openai' | 'anthropic' | 'google' | 'openai-compatible';
+
+interface LlmshConfig {
+  provider: ProviderType;
+  model: string;
+  options?: {
+    apiKey?: string;
+    baseURL?: string;
+    name?: string;
+    organization?: string;
+    project?: string;
+  };
+}
+
+const CONFIG_PATH = join(homedir(), '.llmsh', 'llmsh.jsonc');
+
+function loadConfig(): LlmshConfig | null {
+  if (!existsSync(CONFIG_PATH)) {
+    return null;
+  }
+
+  try {
+    const content = readFileSync(CONFIG_PATH, 'utf-8');
+    const config = jsonc.parse(content) as LlmshConfig;
+    
+    if (!config.provider || !config.model) {
+      return null;
+    }
+
+    const validProviders: ProviderType[] = ['openai', 'anthropic', 'google', 'openai-compatible'];
+    if (!validProviders.includes(config.provider)) {
+      return null;
+    }
+
+    return config;
+  } catch {
+    return null;
+  }
+}
+
+function showConfigHelp(): void {
+  console.error(`Error: No valid config found at ${CONFIG_PATH}
+
+Create a config file at ~/.llmsh/llmsh.jsonc with one of the following formats:
+
+// OpenAI
+{
+  "provider": "openai",
+  "model": "gpt-4-turbo",
+  "options": {
+    "apiKey": "sk-..."  // or set OPENAI_API_KEY env var
+  }
+}
+
+// Anthropic (Claude)
+{
+  "provider": "anthropic", 
+  "model": "claude-sonnet-4-5",
+  "options": {
+    "apiKey": "sk-ant-..."  // or set ANTHROPIC_API_KEY env var
+  }
+}
+
+// Google (Gemini)
+{
+  "provider": "google",
+  "model": "gemini-2.5-flash",
+  "options": {
+    "apiKey": "..."  // or set GOOGLE_GENERATIVE_AI_API_KEY env var
+  }
+}
+
+// OpenAI-Compatible (Custom Endpoint)
+{
+  "provider": "openai-compatible",
+  "model": "glm-4.7",
+  "options": {
+    "name": "custom",
+    "baseURL": "http://localhost:8000/v1",
+    "apiKey": "your-api-key"
+  }
+}
+`);
+}
+
+function getEnvApiKey(provider: string): string | undefined {
+  switch (provider) {
+    case 'openai':
+      return process.env.OPENAI_API_KEY;
+    case 'anthropic':
+      return process.env.ANTHROPIC_API_KEY;
+    case 'google':
+      return process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    default:
+      return undefined;
+  }
+}
+
+function createProvider(config: LlmshConfig) {
+  const { provider, model, options = {} } = config;
+  const apiKey = options.apiKey || getEnvApiKey(provider);
+  const baseURL = options.baseURL;
+
+  switch (provider) {
+    case 'openai': {
+      return {
+        model: createOpenAI({
+          apiKey,
+          baseURL,
+          organization: options.organization,
+          project: options.project,
+        })(model),
+      };
+    }
+
+    case 'anthropic': {
+      return {
+        model: createAnthropic({
+          apiKey,
+          baseURL,
+        })(model),
+      };
+    }
+
+    case 'google': {
+      return {
+        model: createGoogleGenerativeAI({
+          apiKey,
+          baseURL,
+        })(model),
+      };
+    }
+
+    case 'openai-compatible': {
+      return {
+        model: createOpenAICompatible({
+          name: options.name || 'custom',
+          apiKey,
+          baseURL: baseURL || 'http://localhost:8000/v1',
+        })(model),
+      };
+    }
+
+    default: {
+      throw new Error(`Unknown provider: ${provider}`);
+    }
+  }
+}
+
+async function main() {
+  const config = loadConfig();
+  
+  if (!config) {
+    showConfigHelp();
+    process.exit(1);
+  }
+
+  const { model } = createProvider(config);
+  let transcript = '';
+
+  const server = createServer((req, res) => {
+    if (req.method === 'POST') {
+      let body = '';
+      req.on('data', (chunk) => body += chunk);
+      req.on('end', async () => {
+        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.',
+            prompt: `${transcript}\n\n${body}`
+          });
+
+          res.writeHead(200, { 'Content-Type': 'text/plain' });
+          res.write('\n\n');
+          for await (const chunk of result.textStream) {
+            res.write(chunk);
+          }
+          res.write('\n\n');
+          res.end();
+        } catch (error) {
+          res.writeHead(500, { 'Content-Type': 'text/plain' });
+          res.end('Error: ' + (error as Error).message);
+        }
+      });
+    } else {
+      res.writeHead(200, { 'Content-Type': 'text/plain' });
+      res.end(transcript);
+    }
+  });
+  server.listen(0);
+  await once(server, 'listening');
+  const address = server.address();
+  const port = address && typeof address === 'object' ? address.port : 0;
+
+  const ptyProcess = spawn('zsh', [], {
+    name: process.env.TERM || 'xterm',
+    cols: process.stdout.columns || 80,
+    rows: process.stdout.rows || 24,
+    cwd: process.cwd(),
+    env: { ...process.env, LLMSH_PORT: String(port) } as { [key: string]: string }
+  });
+
+  process.stdin.setRawMode(true);
+  process.stdin.resume();
+
+  process.stdin.on('data', (data: Buffer) => {
+    ptyProcess.write(data.toString());
+  });
+
+  ptyProcess.onData((data: string) => {
+    const clearSeq = '\x1b[2J';
+    const lastClearIndex = data.lastIndexOf(clearSeq);
+    
+    if (lastClearIndex !== -1) {
+      transcript = '';
+      const afterClear = data.slice(lastClearIndex + clearSeq.length);
+      transcript += afterClear.replace(/^\x1b\[H/, '');
+    } else {
+      transcript += data;
+    }
+    process.stdout.write(data);
+  });
+
+  ptyProcess.write('help() { curl -s -X POST -d "$*" http://localhost:$LLMSH_PORT }\n');
+
+  ptyProcess.onExit(({ exitCode }: { exitCode: number }) => {
+    process.exit(exitCode);
+  });
+
+  process.on('exit', () => {
+    ptyProcess.kill();
+  });
+
+  process.stdout.on('resize', () => {
+    ptyProcess.resize(process.stdout.columns || 80, process.stdout.rows || 24);
+  });
+}
+
+main();

package/tsconfig.json

@@ -0,0 +1,15 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "verbatimModuleSyntax": true
+  },
+  "include": ["src/**/*"]
+}