shabti 1.2.0 → 1.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shabti",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "description": "Demo CLI tool — showcasing npm-publishable CLI structure",
   "type": "module",
   "bin": {
@@ -37,6 +37,8 @@
     "chalk": "^5.6.2",
     "cli-table3": "^0.6.5",
     "commander": "^14.0.3",
+    "dotenv": "^17.3.1",
+    "openai": "^6.22.0",
     "ora": "^9.3.0"
   },
   "lint-staged": {

package/src/commands/chat.js

@@ -0,0 +1,84 @@
+import { createInterface } from "readline";
+import OpenAI from "openai";
+import chalk from "chalk";
+import { error, info } from "../utils/style.js";
+
+const SYSTEM_PROMPT = `You are Shabti, a helpful assistant accessed via CLI. Be concise and direct.`;
+
+export function registerChat(program) {
+  program
+    .command("chat")
+    .description("Start an interactive chat session with GPT")
+    .option("-m, --model <model>", "OpenAI model to use", "gpt-4o-mini")
+    .option("-s, --system <prompt>", "Custom system prompt", SYSTEM_PROMPT)
+    .action(async (opts) => {
+      await import("dotenv/config");
+      const apiKey = process.env.OPENAI_API_KEY;
+      if (!apiKey || apiKey === "your-api-key-here") {
+        error("OPENAI_API_KEY is not set. Add it to .env in the project root.");
+        process.exit(1);
+      }
+
+      const client = new OpenAI({ apiKey });
+      const messages = [{ role: "system", content: opts.system }];
+
+      console.log();
+      info(`shabti chat — model: ${chalk.cyan(opts.model)}`);
+      console.log(chalk.dim("  Type your message. Ctrl+C or 'exit' to quit.\n"));
+
+      const rl = createInterface({
+        input: process.stdin,
+        output: process.stdout,
+      });
+
+      const prompt = () => {
+        rl.question(chalk.green("You: "), async (input) => {
+          const trimmed = input.trim();
+          if (!trimmed) return prompt();
+          if (trimmed.toLowerCase() === "exit") {
+            console.log(chalk.dim("\nBye.\n"));
+            rl.close();
+            return;
+          }
+
+          messages.push({ role: "user", content: trimmed });
+
+          try {
+            process.stdout.write(chalk.cyan("shabti: "));
+
+            const stream = await client.chat.completions.create({
+              model: opts.model,
+              messages,
+              stream: true,
+            });
+
+            let reply = "";
+            for await (const chunk of stream) {
+              const delta = chunk.choices[0]?.delta?.content || "";
+              reply += delta;
+              process.stdout.write(delta);
+            }
+            console.log("\n");
+
+            messages.push({ role: "assistant", content: reply });
+          } catch (err) {
+            console.log();
+            if (err.status === 401) {
+              error("Invalid API key. Check your .env file.");
+            } else if (err.status === 429) {
+              error("Rate limited. Wait a moment and try again.");
+            } else {
+              error(`API error: ${err.message}`);
+            }
+            console.log();
+          }
+
+          prompt();
+        });
+      };
+
+      prompt();
+
+      rl.on("close", () => process.exit(0));
+    });
+}

package/src/index.js

@@ -2,6 +2,7 @@
 
 import chalk from "chalk";
 import { Command } from "commander";
+import { registerChat } from "./commands/chat.js";
 import { registerHello } from "./commands/hello.js";
 import { registerList } from "./commands/list.js";
 import { registerSpin } from "./commands/spin.js";
@@ -41,6 +42,7 @@ program
   .version("1.0.0");
 
 // Register commands
+registerChat(program);
 registerHello(program);
 registerList(program);
 registerSpin(program);