usetraceforge-cli 0.1.6 → 0.1.8

package/dist/installers/gemini.js

@@ -50,19 +50,24 @@ export async function runGeminiAgent(apiKey, endpoint) {
     s.message("Agent: Analyzing architecture with Gemini...");
     try {
         const genAI = new GoogleGenerativeAI(geminiKey);
-        const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
+        // Use gemini-pro as a safe fallback since 1.5-flash threw a 404 on your key
+        const model = genAI.getGenerativeModel({ model: "gemini/gemini-3.1-flash-lite" });
         const prompt = `
 I am trying to install an SDK called "usetraceforge".
 I have the following files in my project:
 
 ${folderStructure}
 
-Please tell me EXACTLY how to configure TraceForge for my framework.
-If this is Next.js 15, tell me to create 'instrumentation.ts'.
-If this is Express, tell me to add 'TraceForge.expressErrorHandler()'.
-Keep your answer very short, professional, and provide the exact code snippets I need to copy and paste.
-The API Key they should use is: ${apiKey}
-The Endpoint is: ${endpoint}
+Here is the documentation for usetraceforge:
+- Next.js 15+ Server: Create 'instrumentation.ts' (or 'src/instrumentation.ts') exporting 'register' and 'onRequestError(err, request)' which calls 'TraceForge.captureException(err, { tags: { route: request.url } })'.
+- Next.js Client: Wrap your app with '<TraceForgeProvider>' from 'usetraceforge/react'.
+- Express: Call 'app.use(TraceForge.expressErrorHandler())' AFTER all routes but BEFORE custom error handlers.
+- Initialization: You must always call 'TraceForge.init({ apiKey, endpoint })' as early as possible.
+
+The user's API Key is: ${apiKey}
+The user's Endpoint is: ${endpoint}
+
+Based on the folder structure above, tell the developer exactly which files they need to edit and provide the exact code snippets they need to copy and paste to configure TraceForge. Be concise, accurate, and professional.
 `;
         const result = await model.generateContent(prompt);
         const response = result.response;

package/dist/installers/nextjs.js

@@ -2,6 +2,7 @@ import fs from "fs";
 import path from "path";
 import { spinner } from "@clack/prompts";
 import chalk from "chalk";
+import { Project, SyntaxKind } from "ts-morph";
 export async function installNextJs(apiKey, endpoint) {
     const s = spinner();
     s.start("Agent: Configuring Next.js Instrumentation...");
@@ -43,6 +44,48 @@ export function onRequestError(err: any, request: any) {
         else {
             console.log(chalk.yellow(`\nAgent: instrumentation.ts already exists. Please manually add TraceForge.`));
         }
+        // 3. Layout AST Injection (Client-side Provider)
+        const project = new Project();
+        const layoutPaths = [
+            path.resolve(process.cwd(), "app/layout.tsx"),
+            path.resolve(process.cwd(), "src/app/layout.tsx"),
+        ];
+        let targetLayout = null;
+        for (const p of layoutPaths) {
+            if (fs.existsSync(p)) {
+                targetLayout = p;
+                break;
+            }
+        }
+        if (targetLayout) {
+            const sourceFile = project.addSourceFileAtPath(targetLayout);
+            if (!sourceFile.getFullText().includes("TraceForgeProvider")) {
+                sourceFile.addImportDeclaration({
+                    namedImports: ["TraceForgeProvider"],
+                    moduleSpecifier: "usetraceforge/react"
+                });
+                // Find body tag
+                const jsxElements = sourceFile.getDescendantsOfKind(SyntaxKind.JsxElement);
+                let bodyTag = jsxElements.find(el => el.getOpeningElement().getTagNameNode().getText() === "body");
+                if (bodyTag) {
+                    const childrenText = bodyTag.getJsxChildren().map(c => c.getText()).join("");
+                    const opening = bodyTag.getOpeningElement().getText();
+                    const closing = bodyTag.getClosingElement().getText();
+                    bodyTag.replaceWithText(`${opening}\n        <TraceForgeProvider>\n          ${childrenText}\n        </TraceForgeProvider>\n      ${closing}`);
+                    sourceFile.saveSync();
+                    console.log(chalk.green(`\nAgent: Injected <TraceForgeProvider> into ${targetLayout.replace(process.cwd(), "")}`));
+                }
+                else {
+                    console.log(chalk.yellow("\nAgent: Could not find <body> tag in layout.tsx. Please add <TraceForgeProvider> manually."));
+                }
+            }
+            else {
+                console.log(chalk.blue(`\nAgent: <TraceForgeProvider> already exists in layout.tsx`));
+            }
+        }
+        else {
+            console.log(chalk.yellow("\nAgent: Could not find layout.tsx. Please add <TraceForgeProvider> manually."));
+        }
         s.stop(chalk.green("Agent: Next.js configuration complete!"));
     }
     catch (error) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "usetraceforge-cli",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "TraceForge CLI Wizard for 2-click installations",
   "bin": {
     "traceforge": "./dist/index.js"