inconvo 1.2.1 → 1.4.0

package/README.md

@@ -20,6 +20,8 @@ Options:
 
 By default the installer writes everything under `src/`, including the tool components and their supporting `src/lib/inconvo/types.ts`. It also installs external dependencies such as `recharts` and `@tanstack/react-table` unless you opt out with `--skip-install`.
 
+After the files are copied, the CLI attempts to update `src/components/assistant-ui/thread.tsx` by importing `InconvoTools` and rendering it inside `ThreadPrimitive.Root`. If the file is missing or has been customized heavily, you'll see a warning and can complete the insertion manually.
+
 ## Development
 
 ```bash

package/dist/commands/add.js

@@ -4,6 +4,7 @@ import fs from "node:fs/promises";
 import { logger } from "../lib/logger.js";
 import { confirmPrompt } from "../lib/prompt.js";
 import { installDependencies } from "../lib/install-dependencies.js";
+import { ensureThreadHasTools } from "../lib/thread-injector.js";
 import { componentPacks, getPack } from "../templates/packs.js";
 export const addCommand = new Command()
     .name("add")
@@ -70,6 +71,7 @@ export const addCommand = new Command()
     if (replaced.size) {
         logger.info(`Overwrote ${replaced.size} file(s).`);
     }
+    await ensureThreadHasTools(cwd);
     if (pack.dependencies) {
         const depList = Object.entries(pack.dependencies).map(([name, version]) => `${name}@${version}`);
         if (!depList.length)

package/dist/lib/thread-injector.js

@@ -0,0 +1,86 @@
+import fs from "node:fs/promises";
+import path from "node:path";
+import { logger } from "./logger.js";
+const TOOLS_IMPORT = `import { InconvoTools } from "./tools/inconvo-tools";`;
+const THREAD_PATH = ["src", "components", "assistant-ui", "thread.tsx"];
+export async function ensureThreadHasTools(projectRoot) {
+    const threadPath = path.join(projectRoot, ...THREAD_PATH);
+    if (!(await fileExists(threadPath))) {
+        logger.warn(`Could not find ${path.relative(projectRoot, threadPath)}. Skipping thread update.`);
+        return;
+    }
+    let content = await fs.readFile(threadPath, "utf8");
+    let updated = false;
+    if (!hasToolsImport(content)) {
+        content = injectImport(content);
+        updated = true;
+    }
+    if (!content.includes("<InconvoTools")) {
+        const nextContent = injectComponentUsage(content);
+        if (nextContent) {
+            content = nextContent;
+            updated = true;
+        }
+        else {
+            logger.warn("Could not locate </ThreadPrimitive.Root> in thread.tsx. Please insert <InconvoTools /> manually.");
+        }
+    }
+    if (updated) {
+        await fs.writeFile(threadPath, content, "utf8");
+        logger.success("Updated thread component to render <InconvoTools />.");
+    }
+    else {
+        logger.info("Thread component already renders <InconvoTools />.");
+    }
+}
+function hasToolsImport(content) {
+    const namedImportRegex = /import\s+{[^}]*InconvoTools[^}]*}\s+from\s+["'][^"']+["']/;
+    const defaultImportRegex = /import\s+InconvoTools\s+from\s+["'][^"']+["']/;
+    return (namedImportRegex.test(content) || defaultImportRegex.test(content));
+}
+function injectImport(content) {
+    const lines = content.split("\n");
+    let lastImportIndex = -1;
+    for (let i = 0; i < lines.length; i++) {
+        if (lines[i].startsWith("import ")) {
+            lastImportIndex = i;
+        }
+    }
+    if (lastImportIndex >= 0) {
+        lines.splice(lastImportIndex + 1, 0, TOOLS_IMPORT);
+    }
+    else {
+        lines.unshift(TOOLS_IMPORT);
+    }
+    return lines.join("\n");
+}
+function injectComponentUsage(content) {
+    const lines = content.split("\n");
+    const rootIndex = lines.findIndex((line) => line.includes("<ThreadPrimitive.Root"));
+    if (rootIndex === -1) {
+        return null;
+    }
+    let insertIndex = rootIndex + 1;
+    for (let i = rootIndex; i < lines.length; i++) {
+        if (lines[i].includes(">")) {
+            insertIndex = i + 1;
+            break;
+        }
+    }
+    const baseIndent = lines[rootIndex].match(/^\s*/)?.[0] ?? "";
+    const componentLine = `${baseIndent}  <InconvoTools />`;
+    lines.splice(insertIndex, 0, componentLine);
+    return lines.join("\n");
+}
+async function fileExists(target) {
+    try {
+        await fs.stat(target);
+        return true;
+    }
+    catch (error) {
+        if (error && error.code === "ENOENT") {
+            return false;
+        }
+        throw error;
+    }
+}

package/dist/templates/packs.js

@@ -2,7 +2,7 @@ import { resolveFromRoot } from "../lib/paths.js";
 export const componentPacks = {
     "assistant-ui-tool-components": {
         name: "assistant-ui-tool-components",
-        description: "Inconvo-flavored assistant-ui tools and supporting components.",
+        description: "Inconvo-flavored tool output components for assistant-ui.",
         templateDir: resolveFromRoot("templates", "assistant-ui", "src"),
         targetDir: "src",
         dependencies: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "inconvo",
-  "version": "1.2.1",
+  "version": "1.4.0",
   "description": "CLI for installing Inconvo assistant-ui tool components into any project.",
   "license": "MIT",
   "type": "module",

package/templates/assistant-ui/src/components/assistant-ui/tools/inconvo-hidden-tools.tsx

@@ -0,0 +1,30 @@
+"use client";
+
+import {
+  makeAssistantToolUI,
+  type ToolCallMessagePartComponent,
+} from "@assistant-ui/react";
+
+type GenericArgs = Record<string, unknown>;
+type GenericResult = unknown;
+
+const HiddenToolRender: ToolCallMessagePartComponent<
+  GenericArgs,
+  GenericResult
+> = () => null;
+
+export const GetDataSummaryTool = makeAssistantToolUI<
+  GenericArgs,
+  GenericResult
+>({
+  toolName: "get_data_summary",
+  render: HiddenToolRender,
+});
+
+export const StartDataAnalystConversationTool = makeAssistantToolUI<
+  GenericArgs,
+  GenericResult
+>({
+  toolName: "start_data_analyst_conversation",
+  render: HiddenToolRender,
+});

package/templates/assistant-ui/src/components/assistant-ui/tools/inconvo-tools.tsx

@@ -1,7 +1,17 @@
 "use client";
 
 import { MessageDataAnalystTool } from "~/components/assistant-ui/tools/message-data-analyst-tool";
+import {
+  GetDataSummaryTool,
+  StartDataAnalystConversationTool,
+} from "~/components/assistant-ui/tools/inconvo-hidden-tools";
 
 export const InconvoTools = () => {
-  return <MessageDataAnalystTool />;
+  return (
+    <>
+      <MessageDataAnalystTool />
+      <GetDataSummaryTool />
+      <StartDataAnalystConversationTool />
+    </>
+  );
 };