youmd 0.3.3 → 0.4.1

package/dist/commands/chat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../src/commands/chat.ts"],"names":[],"mappings":"AA+sBA,wBAAsB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAwXjD"}
+{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../src/commands/chat.ts"],"names":[],"mappings":"AA25BA,wBAAsB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAihBjD"}

package/dist/commands/chat.js

@@ -42,12 +42,127 @@ const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 const chalk_1 = __importDefault(require("chalk"));
 const config_1 = require("../lib/config");
+const project_1 = require("../lib/project");
 const compiler_1 = require("../lib/compiler");
 const api_1 = require("../lib/api");
 const render_1 = require("../lib/render");
 const onboarding_1 = require("../lib/onboarding");
 // ─── URL Detection + Scraping (mirrors web useYouAgent) ──────────────
 const CONVEX_SITE_URL = "https://kindly-cassowary-600.convex.site";
+const STREAM_URL = `${CONVEX_SITE_URL}/api/v1/chat/stream`;
+// ─── Streaming LLM client ─────────────────────────────────────────────
+async function streamLLM(_apiKey, messages, onToken) {
+    const res = await fetch(STREAM_URL, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ messages }),
+        signal: AbortSignal.timeout(120000),
+    });
+    if (!res.ok) {
+        const body = await res.text();
+        throw new Error(`Stream error (${res.status}): ${body}`);
+    }
+    if (!res.body) {
+        throw new Error("No response body from stream endpoint");
+    }
+    const reader = res.body.getReader();
+    const decoder = new TextDecoder();
+    let fullText = "";
+    let buffer = "";
+    while (true) {
+        const { done, value } = await reader.read();
+        if (done)
+            break;
+        buffer += decoder.decode(value, { stream: true });
+        // Process complete SSE lines
+        const lines = buffer.split("\n");
+        // Keep the last potentially incomplete line in the buffer
+        buffer = lines.pop() || "";
+        for (const line of lines) {
+            const trimmed = line.trim();
+            if (!trimmed)
+                continue;
+            if (trimmed.startsWith("data: ")) {
+                const data = trimmed.slice(6);
+                if (data === "[DONE]") {
+                    continue;
+                }
+                try {
+                    const parsed = JSON.parse(data);
+                    if (parsed.text) {
+                        fullText += parsed.text;
+                        onToken(parsed.text);
+                    }
+                }
+                catch {
+                    // Skip malformed JSON chunks
+                }
+            }
+        }
+    }
+    // Process any remaining buffer
+    if (buffer.trim()) {
+        const trimmed = buffer.trim();
+        if (trimmed.startsWith("data: ")) {
+            const data = trimmed.slice(6);
+            if (data !== "[DONE]") {
+                try {
+                    const parsed = JSON.parse(data);
+                    if (parsed.text) {
+                        fullText += parsed.text;
+                        onToken(parsed.text);
+                    }
+                }
+                catch {
+                    // Skip
+                }
+            }
+        }
+    }
+    return fullText;
+}
+/**
+ * Call LLM with streaming, falling back to blocking callLLM on failure.
+ * Returns the full response text.
+ */
+async function callLLMWithStreaming(apiKey, messages, spinnerLabel) {
+    const thinkSpinner = new render_1.BrailleSpinner(spinnerLabel);
+    thinkSpinner.start();
+    try {
+        let firstToken = true;
+        const response = await streamLLM(apiKey, messages, (token) => {
+            if (firstToken) {
+                // Clear the spinner line before writing streamed text
+                thinkSpinner.stop();
+                process.stdout.write("  ");
+                firstToken = false;
+            }
+            process.stdout.write(token);
+        });
+        if (!firstToken) {
+            // We streamed something -- add trailing newline
+            process.stdout.write("\n");
+        }
+        else {
+            // No tokens received -- clear spinner
+            thinkSpinner.stop();
+        }
+        return response;
+    }
+    catch {
+        // Streaming failed -- fall back to blocking call
+        thinkSpinner.update("streaming unavailable, waiting for response");
+        try {
+            const response = await (0, onboarding_1.callLLM)(apiKey, messages);
+            thinkSpinner.stop();
+            return response;
+        }
+        catch (err) {
+            thinkSpinner.fail(err instanceof Error ? err.message : "failed");
+            throw err;
+        }
+    }
+}
 function detectSourcesInMessage(text) {
     const sources = [];
     const seen = new Set();
@@ -212,6 +327,51 @@ function parsePrivateUpdates(text) {
     return updates;
 }
 const scrapedSources = new Set();
+// ─── Image/File handling ──────────────────────────────────────────────
+function detectFilePath(input) {
+    const trimmed = input.trim();
+    // Detect dragged file paths (terminals add quotes or escape spaces)
+    // Strip surrounding quotes
+    const unquoted = trimmed.replace(/^['"]|['"]$/g, "");
+    // Check if it looks like a file path
+    if ((unquoted.startsWith("/") || unquoted.startsWith("~") || unquoted.startsWith("./")) &&
+        fs.existsSync(unquoted)) {
+        return unquoted;
+    }
+    return null;
+}
+function isImageFile(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    return [".png", ".jpg", ".jpeg", ".gif", ".webp", ".bmp", ".svg"].includes(ext);
+}
+function fileToBase64DataUrl(filePath) {
+    try {
+        const buffer = fs.readFileSync(filePath);
+        const ext = path.extname(filePath).toLowerCase();
+        const mimeMap = {
+            ".png": "image/png",
+            ".jpg": "image/jpeg",
+            ".jpeg": "image/jpeg",
+            ".gif": "image/gif",
+            ".webp": "image/webp",
+            ".bmp": "image/bmp",
+            ".svg": "image/svg+xml",
+        };
+        const mime = mimeMap[ext] || "application/octet-stream";
+        return `data:${mime};base64,${buffer.toString("base64")}`;
+    }
+    catch {
+        return null;
+    }
+}
+function readTextFile(filePath) {
+    try {
+        return fs.readFileSync(filePath, "utf-8");
+    }
+    catch {
+        return null;
+    }
+}
 // ─── Constants ────────────────────────────────────────────────────────
 const CHAT_SYSTEM_PROMPT = `you are the you.md agent — the first AI that truly knows people. you help humans build and maintain their identity file for the agent internet. not a chatbot. not an assistant. an identity specialist with a personality.
 
@@ -269,7 +429,16 @@ your job:
 5. be proactive: "looks like your projects section could use an update — want to add that?"
 6. when they share something sensitive, ask: "want me to keep that private or add it to your public profile?"
 
-rules: each section starts with YAML frontmatter. real markdown, not placeholders. output FULL section content each time. be substantive — write from what you actually know.`;
+rules: each section starts with YAML frontmatter. real markdown, not placeholders. output FULL section content each time. be substantive — write from what you actually know.
+
+--- project context updates ---
+
+if the user is working in a project (you'll see a [PROJECT CONTEXT] block), you can update project files. when you learn something about the project — a decision made, a task completed, a feature shipped, a new requirement — output:
+\`\`\`json
+{"project_updates": [{"file": "context/todo.md", "content": "updated content..."}]}
+\`\`\`
+allowed files: context/todo.md, context/features.md, context/changelog.md, context/decisions.md, context/prd.md, agent/instructions.md, agent/memory.json, private/notes.md
+only output project_updates when something actually changed. the system will write these files and show a notice to the user.`;
 const SLASH_COMMANDS = {
     "/status": "show bundle status",
     "/preview": "show profile preview",
@@ -280,6 +449,7 @@ const SLASH_COMMANDS = {
     "/memory": "show memory summary + stats",
     "/recall": "show recent memories (or /recall query)",
     "/private": "show private context (notes, links, projects)",
+    "/image <path>": "attach an image or file",
     "/rebuild": "recompile the bundle",
     "/help": "show available commands",
     "/done": "exit chat",
@@ -625,12 +795,74 @@ async function chatCommand() {
     const bundleDir = (0, config_1.getLocalBundleDir)();
     const apiKey = (0, onboarding_1.getOpenRouterKey)();
     const rl = createRL();
+    // Detect project context (legacy detection from config.ts)
+    const projectCtx = (0, config_1.detectProjectContext)();
+    let projectContextBlock = "";
+    let activeProjectDir = null;
+    if (projectCtx) {
+        console.log("");
+        console.log("  " + chalk_1.default.hex("#C46A3A")("project:") + " " + chalk_1.default.white(projectCtx.name) +
+            chalk_1.default.dim(` (${projectCtx.root})`));
+        // Try the new file-system project context first
+        const projectsRoot = (0, project_1.findProjectsRoot)();
+        if (projectsRoot) {
+            const detected = (0, project_1.detectCurrentProject)(projectsRoot);
+            if (detected) {
+                activeProjectDir = (0, project_1.getProjectDir)(projectsRoot, detected);
+                const injection = (0, project_1.buildProjectContextInjection)(activeProjectDir);
+                if (injection) {
+                    projectContextBlock = `\n\n--- project context ---\n${injection}`;
+                }
+            }
+        }
+        // Fallback to legacy project context if new system didn't produce anything
+        if (!projectContextBlock) {
+            const projectNotes = (0, config_1.readProjectPrivateNotes)(projectCtx.name);
+            const parts = [];
+            parts.push(`the user is currently working in project: ${projectCtx.name} at ${projectCtx.root}`);
+            if (projectCtx.youmdProject?.description) {
+                parts.push(`project description: ${projectCtx.youmdProject.description}`);
+            }
+            if (projectNotes) {
+                parts.push(`project-specific private notes:\n${projectNotes}`);
+            }
+            projectContextBlock = `\n\n--- project context ---\n${parts.join("\n")}`;
+        }
+    }
     console.log("");
     console.log("  " + chalk_1.default.bold("you.md chat"));
     console.log(chalk_1.default.dim("  talk to update your profile. /help for commands."));
     console.log("");
     // Load current profile as context
     const currentBundle = loadCurrentBundle(bundleDir);
+    // Load agent directives from you.json if available
+    let directivesContext = "";
+    const youJsonPath = path.join(bundleDir, "you.json");
+    if (fs.existsSync(youJsonPath)) {
+        try {
+            const youJson = JSON.parse(fs.readFileSync(youJsonPath, "utf-8"));
+            const directives = youJson.agent_directives;
+            if (directives) {
+                const parts = [];
+                if (directives.communication_style)
+                    parts.push(`communication style: ${directives.communication_style}`);
+                if (directives.default_stack)
+                    parts.push(`default stack: ${directives.default_stack}`);
+                if (directives.current_goal)
+                    parts.push(`current goal: ${directives.current_goal}`);
+                if (directives.decision_framework)
+                    parts.push(`decision framework: ${directives.decision_framework}`);
+                if (directives.negative_prompts && directives.negative_prompts.length > 0)
+                    parts.push(`never do: ${directives.negative_prompts.join("; ")}`);
+                if (parts.length > 0) {
+                    directivesContext = `\n\n--- agent directives (follow these when interacting with me) ---\n${parts.join("\n")}`;
+                }
+            }
+        }
+        catch {
+            // non-fatal — skip directives if you.json is malformed
+        }
+    }
     // Extract profile details for a personalized greeting prompt
     const profileHint = extractProfileHint(bundleDir);
     let greetingInstruction = "greet me briefly and ask what i'd like to update or work on. keep it short.";
@@ -641,25 +873,21 @@ async function chatCommand() {
         { role: "system", content: CHAT_SYSTEM_PROMPT },
         {
             role: "user",
-            content: `here is my current identity bundle:\n\n${currentBundle}\n\n${greetingInstruction}`,
+            content: `here is my current identity bundle:\n\n${currentBundle}${directivesContext}${projectContextBlock}\n\n${greetingInstruction}`,
         },
     ];
     // Initial greeting from agent
-    const spinner = new onboarding_1.Spinner((0, onboarding_1.randomThinking)());
-    spinner.start();
     let response;
     try {
-        response = await (0, onboarding_1.callLLM)(apiKey, messages);
+        response = await callLLMWithStreaming(apiKey, messages, (0, onboarding_1.randomThinking)());
     }
     catch (err) {
-        spinner.stop();
         console.log(chalk_1.default.red(`  failed to connect: ${err instanceof Error ? err.message : String(err)}`));
         console.log(chalk_1.default.dim("  chat requires the AI service. try again later."));
         console.log("");
         rl.close();
         return;
     }
-    spinner.stop();
     messages.push({ role: "assistant", content: response });
     const initial = (0, onboarding_1.parseUpdatesFromResponse)(response);
     // Write any updates (unlikely on greeting, but handle it)
@@ -670,6 +898,9 @@ async function chatCommand() {
         console.log(chalk_1.default.cyan(`  [updated: ${initial.updates.map((u) => (0, onboarding_1.sectionLabel)(u.section)).join(", ")}]`));
         console.log("");
     }
+    // Only print via rich renderer if we didn't stream (streaming already wrote output)
+    // But we still need to display parsed output for non-streamed fallback
+    // Since streaming writes raw text, print formatted version for updates parsing
     printAgentMessage(initial.display);
     // ── Conversation loop ──────────────────────────────────────────────
     while (true) {
@@ -803,20 +1034,16 @@ async function chatCommand() {
             if (!researchOk)
                 continue;
             // After research, get an LLM response with the injected context
-            const researchSpinner = new onboarding_1.Spinner((0, onboarding_1.randomThinking)());
-            researchSpinner.start();
             try {
-                response = await (0, onboarding_1.callLLM)(apiKey, messages);
+                response = await callLLMWithStreaming(apiKey, messages, (0, onboarding_1.randomThinking)());
             }
             catch (err) {
-                researchSpinner.stop();
                 console.log(chalk_1.default.red(`  AI error: ${err instanceof Error ? err.message : String(err)}`));
                 console.log(chalk_1.default.dim("  try again."));
                 console.log("");
                 messages.pop();
                 continue;
             }
-            researchSpinner.stop();
             messages.push({ role: "assistant", content: response });
             const researchParsed = (0, onboarding_1.parseUpdatesFromResponse)(response);
             if (researchParsed.updates.length > 0) {
@@ -833,6 +1060,93 @@ async function chatCommand() {
             handleRebuild(bundleDir);
             continue;
         }
+        // ── Handle /image command ──
+        if (lower.startsWith("/image ")) {
+            const imgPath = userInput.slice(7).trim().replace(/^['"]|['"]$/g, "");
+            if (!fs.existsSync(imgPath)) {
+                console.log(chalk_1.default.hex("#C46A3A")(`  file not found: ${imgPath}`));
+                console.log("");
+                continue;
+            }
+            if (isImageFile(imgPath)) {
+                const dataUrl = fileToBase64DataUrl(imgPath);
+                if (dataUrl) {
+                    console.log(chalk_1.default.green(`  ✓`) + chalk_1.default.dim(` attached image: ${path.basename(imgPath)}`));
+                    messages.push({
+                        role: "user",
+                        content: `[USER ATTACHED IMAGE: ${path.basename(imgPath)}]\nthe user attached an image file. describe or use it as needed.\n![${path.basename(imgPath)}](${dataUrl})`,
+                    });
+                }
+            }
+            else {
+                const text = readTextFile(imgPath);
+                if (text) {
+                    console.log(chalk_1.default.green(`  ✓`) + chalk_1.default.dim(` attached file: ${path.basename(imgPath)} (${text.length} chars)`));
+                    messages.push({
+                        role: "user",
+                        content: `[USER ATTACHED FILE: ${path.basename(imgPath)}]\n\`\`\`\n${text.slice(0, 10000)}\n\`\`\``,
+                    });
+                }
+            }
+            try {
+                response = await callLLMWithStreaming(apiKey, messages, (0, onboarding_1.randomThinking)());
+            }
+            catch (err) {
+                console.log(chalk_1.default.red(`  ${err instanceof Error ? err.message : "failed"}`));
+                messages.pop();
+                continue;
+            }
+            messages.push({ role: "assistant", content: response });
+            printAgentMessage((0, onboarding_1.parseUpdatesFromResponse)(response).display);
+            continue;
+        }
+        // ── Detect dragged/pasted file paths ──
+        const detectedFile = detectFilePath(userInput);
+        if (detectedFile) {
+            if (isImageFile(detectedFile)) {
+                const dataUrl = fileToBase64DataUrl(detectedFile);
+                if (dataUrl) {
+                    console.log(chalk_1.default.green(`  ✓`) + chalk_1.default.dim(` detected image: ${path.basename(detectedFile)}`));
+                    messages.push({
+                        role: "user",
+                        content: `[USER DROPPED IMAGE: ${path.basename(detectedFile)}]\nthe user dropped an image file into the chat.\n![${path.basename(detectedFile)}](${dataUrl})`,
+                    });
+                    try {
+                        response = await callLLMWithStreaming(apiKey, messages, (0, onboarding_1.randomThinking)());
+                    }
+                    catch (err) {
+                        console.log(chalk_1.default.red(`  ${err instanceof Error ? err.message : "failed"}`));
+                        messages.pop();
+                        continue;
+                    }
+                    messages.push({ role: "assistant", content: response });
+                    printAgentMessage((0, onboarding_1.parseUpdatesFromResponse)(response).display);
+                    continue;
+                }
+            }
+            else {
+                // Text file — inject content
+                const text = readTextFile(detectedFile);
+                if (text) {
+                    console.log(chalk_1.default.green(`  ✓`) + chalk_1.default.dim(` detected file: ${path.basename(detectedFile)} (${text.length} chars)`));
+                    messages.push({
+                        role: "user",
+                        content: `[USER DROPPED FILE: ${path.basename(detectedFile)}]\n\`\`\`\n${text.slice(0, 10000)}\n\`\`\`\n\nreview this file and suggest how it relates to my identity or profile.`,
+                    });
+                    try {
+                        response = await callLLMWithStreaming(apiKey, messages, (0, onboarding_1.randomThinking)());
+                    }
+                    catch (err) {
+                        console.log(chalk_1.default.red(`  ${err instanceof Error ? err.message : "failed"}`));
+                        messages.pop();
+                        continue;
+                    }
+                    messages.push({ role: "assistant", content: response });
+                    printAgentMessage((0, onboarding_1.parseUpdatesFromResponse)(response).display);
+                    continue;
+                }
+            }
+        }
         // ── Auto-detect URLs and scrape before sending to LLM ──
         const detectedSources = detectSourcesInMessage(userInput);
         const newSources = detectedSources.filter((s) => !scrapedSources.has(`${s.platform}:${s.username || s.url}`));
@@ -863,13 +1177,11 @@ async function chatCommand() {
                 });
             }
         }
-        const thinkSpinner = new render_1.BrailleSpinner((0, onboarding_1.randomThinking)());
-        thinkSpinner.start();
         try {
-            response = await (0, onboarding_1.callLLM)(apiKey, messages);
+            response = await callLLMWithStreaming(apiKey, messages, (0, onboarding_1.randomThinking)());
         }
         catch (err) {
-            thinkSpinner.fail(err instanceof Error ? err.message : "failed");
+            console.log(chalk_1.default.red(`  ${err instanceof Error ? err.message : "failed"}`));
             console.log(chalk_1.default.dim("  try again."));
             console.log("");
             messages.pop();
@@ -877,7 +1189,6 @@ async function chatCommand() {
                 messages.pop(); // remove scrape context too
             continue;
         }
-        thinkSpinner.stop();
         messages.push({ role: "assistant", content: response });
         const parsed = (0, onboarding_1.parseUpdatesFromResponse)(response);
         // Write section updates
@@ -923,6 +1234,21 @@ async function chatCommand() {
                 }
             }
         }
+        // Handle project context updates
+        if (activeProjectDir) {
+            const projUpdates = (0, project_1.parseProjectUpdates)(response);
+            if (projUpdates.length > 0) {
+                for (const pu of projUpdates) {
+                    try {
+                        (0, project_1.updateProjectFile)(activeProjectDir, pu.file, pu.content);
+                        console.log(chalk_1.default.hex("#C46A3A")(`  [updated project context: ${pu.file}]`));
+                    }
+                    catch {
+                        // non-fatal
+                    }
+                }
+            }
+        }
         printAgentMessage(parsed.display);
     }
     rl.close();