jeo-code 0.1.0 → 0.4.5

package/src/commands/chat.ts

@@ -1,31 +1,60 @@
-import { createModelManager } from "../ai/model-manager";
+import { createModelManager, thinkingMaxTokens } from "../ai/model-manager";
+import { friendlyProviderError } from "../util/provider-error";
 
 /**
- * `joc chat "<message>"` — a quick, single-shot streaming chat (no tools).
+ * `jeo chat "<message>"` — a quick, single-shot streaming chat (no tools).
  * Renders the reply token-by-token via the provider streaming path
- * (`ModelManager.stream`), complementary to the tool-loop `joc launch`.
+ * (`ModelManager.stream`), complementary to the tool-loop `jeo launch`.
  */
+/** Pull chat-only flags out of args; the rest joins into the message.
+ *  Previously flags were swallowed into the message text, so `jeo chat --model X "hi"`
+ *  silently chatted with the DEFAULT model — the worst kind of wrong. */
+export function parseChatArgs(args: string[]): { model?: string; thinking?: string; maxTokens?: number; message: string } {
+  let model: string | undefined;
+  let thinking: string | undefined;
+  let maxTokens: number | undefined;
+  const rest: string[] = [];
+  const parsePositiveInt = (value: string | undefined): number | undefined => {
+    const n = Number(value);
+    return Number.isInteger(n) && n > 0 ? n : undefined;
+  };
+  for (let i = 0; i < args.length; i++) {
+    const a = args[i] ?? "";
+    if (a === "--model" || a === "-m") { model = args[++i]; continue; }
+    if (a.startsWith("--model=")) { model = a.slice("--model=".length); continue; }
+    if (a === "--thinking") { thinking = args[++i]; continue; }
+    if (a.startsWith("--thinking=")) { thinking = a.slice("--thinking=".length); continue; }
+    if (a === "--max-tokens") { maxTokens = parsePositiveInt(args[++i]); continue; }
+    if (a.startsWith("--max-tokens=")) { maxTokens = parsePositiveInt(a.slice("--max-tokens=".length)); continue; }
+    rest.push(a);
+  }
+  return { model, thinking, maxTokens, message: rest.join(" ").trim() };
+}
+
 export async function runChatCommand(args: string[] = []): Promise<void> {
-  let message = args.join(" ").trim();
+  const { model, thinking, maxTokens: explicitMaxTokens, message: parsed } = parseChatArgs(args);
+  let message = parsed;
   if (!message && !process.stdin.isTTY) message = (await Bun.stdin.text()).trim();
   if (!message) {
-    console.log('Usage: joc chat "<message>"   (streams the reply token-by-token)');
+    console.log('Usage: jeo chat [--model <id>] [--thinking <level>] [--max-tokens <n>] "<message>"   (streams the reply token-by-token)');
     process.exitCode = 1;
     return;
   }
 
   const manager = createModelManager();
-  process.stdout.write("joc> ");
+  process.stdout.write("jeo> ");
   let any = false;
   let usage: { inputTokens?: number; outputTokens?: number; durationMs?: number } | undefined;
   try {
-    for await (const chunk of manager.stream([{ role: "user", content: message }], { onUsage: u => { usage = u; } })) {
+    const effort = (thinking === "xhigh" ? "high" : thinking) as "minimal" | "low" | "medium" | "high" | undefined;
+    const maxTokens = explicitMaxTokens ?? (thinking ? thinkingMaxTokens(thinking as "minimal" | "low" | "medium" | "high" | "xhigh") : undefined);
+    for await (const chunk of manager.stream([{ role: "user", content: message }], { model, maxTokens, reasoningEffort: effort, onUsage: u => { usage = u; } })) {
       process.stdout.write(chunk);
       any = true;
     }
   } catch (err) {
     process.stdout.write("\n");
-    console.log(`! ${(err as Error).message}`);
+    console.log(`! ${friendlyProviderError(err)}`);
     process.exitCode = 1;
     return;
   }