@amplitude/ai 0.2.1 → 0.3.0

package/dist/mcp/generate-verify-test.js

@@ -0,0 +1,25 @@
+//#region src/mcp/generate-verify-test.ts
+function generateVerifyTest(scanResult) {
+	const lines = [];
+	lines.push("import { describe, it, expect } from 'vitest';", "import { MockAmplitudeAI } from '@amplitude/ai/testing';", "", "describe('Amplitude AI verification', () => {");
+	for (const agent of scanResult.agents) lines.push(`  it('${agent.inferred_id} emits correct event sequence', () => {`, "    const mock = new MockAmplitudeAI();", `    const agent = mock.agent('${agent.inferred_id}');`, "    const session = agent.session({ userId: 'verify-user', sessionId: 'verify-session' });", "    session.runSync((s) => {", "      s.trackUserMessage('test message');", "    });", "    const events = mock.getEvents();", "    expect(events.length).toBeGreaterThanOrEqual(2);", `    mock.assertEventTracked('[Agent] User Message', { '[Agent] Agent ID': '${agent.inferred_id}' });`, "    mock.assertSessionClosed('verify-session');", "  });", "");
+	if ((scanResult.is_multi_agent || scanResult.multi_agent_signals.length > 0) && scanResult.agents.length >= 2) {
+		const parentAgent = scanResult.agents[0];
+		const childAgent = scanResult.agents[1];
+		if (parentAgent && childAgent) {
+			lines.push("  it('multi-agent delegation via runAs', () => {", "    const mock = new MockAmplitudeAI();", `    const parent = mock.agent('${parentAgent.inferred_id}');`, `    const child = parent.child('${childAgent.inferred_id}');`, "    const session = parent.session({ userId: 'verify-user', sessionId: 'verify-multi' });", "    session.runSync((s) => {", "      s.runAsSync(child, (cs) => {", "        cs.trackUserMessage('delegated task');", "      });", "    });", `    const childEvents = mock.eventsForAgent('${childAgent.inferred_id}');`, "    expect(childEvents.length).toBeGreaterThan(0);", "    expect(childEvents[0].event_properties?.['[Agent] Session ID']).toBe('verify-multi');", "  });", "");
+			const childAgents = scanResult.agents.slice(0, 2);
+			lines.push("  it('parallel fan-out shares session', async () => {", "    const mock = new MockAmplitudeAI();", `    const orchestrator = mock.agent('${parentAgent.inferred_id}');`, "    const children = [");
+			for (const child of childAgents) lines.push(`      orchestrator.child('${child.inferred_id}'),`);
+			lines.push("    ];", "    const session = orchestrator.session({ userId: 'verify-user', sessionId: 'verify-parallel' });", "    await session.run(async (s) => {", "      await Promise.all(children.map((child) =>", "        s.runAs(child, async (cs) => {", "          cs.trackUserMessage('parallel task');", "        }),", "      ));", "    });", "    for (const child of children) {", "      const events = mock.eventsForAgent(child.agentId);", "      expect(events.length).toBeGreaterThan(0);", "      expect(events[0].event_properties?.['[Agent] Session ID']).toBe('verify-parallel');", "    }", "  });", "");
+			lines.push("  it('tool calls inside runAs are attributed to child agent', () => {", "    const mock = new MockAmplitudeAI();", `    const parent = mock.agent('${parentAgent.inferred_id}');`, `    const child = parent.child('${childAgent.inferred_id}');`, "    const session = parent.session({ userId: 'verify-user', sessionId: 'verify-tools' });", "    session.runSync((s) => {", "      s.runAsSync(child, (cs) => {", "        cs.trackToolCall('search_knowledge_base', 150, true);", "      });", "    });", "    const toolEvents = mock.getEvents().filter(e => e.event_type === '[Agent] Tool Call');", "    expect(toolEvents.length).toBe(1);", `    expect(toolEvents[0].event_properties?.['[Agent] Agent ID']).toBe('${childAgent.inferred_id}');`, `    expect(toolEvents[0].event_properties?.['[Agent] Parent Agent ID']).toBe('${parentAgent.inferred_id}');`, "  });", "");
+			lines.push("  it('runAs restores parent context after child throws', () => {", "    const mock = new MockAmplitudeAI();", `    const parent = mock.agent('${parentAgent.inferred_id}');`, `    const faultyChild = parent.child('${childAgent.inferred_id}-faulty');`, "    const session = parent.session({ userId: 'verify-user', sessionId: 'verify-error' });", "    session.runSync((s) => {", "      try {", "        s.runAsSync(faultyChild, () => {", "          throw new Error('child failed');", "        });", "      } catch {", "        s.trackUserMessage('recovering from child failure');", "      }", "    });", `    const parentEvents = mock.eventsForAgent('${parentAgent.inferred_id}');`, "    const recoveryMsg = parentEvents.find(e => e.event_type === '[Agent] User Message');", "    expect(recoveryMsg).toBeDefined();", `    expect(recoveryMsg?.event_properties?.['[Agent] Agent ID']).toBe('${parentAgent.inferred_id}');`, "  });", "");
+		}
+	}
+	lines.push("});", "");
+	return lines.join("\n");
+}
+
+//#endregion
+export { generateVerifyTest };
+//# sourceMappingURL=generate-verify-test.js.map

package/dist/mcp/generate-verify-test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-verify-test.js","names":["lines: string[]"],"sources":["../../src/mcp/generate-verify-test.ts"],"sourcesContent":["import type { ScanResult } from './scan-project.js';\n\nexport function generateVerifyTest(scanResult: ScanResult): string {\n  const lines: string[] = [];\n\n  lines.push(\n    \"import { describe, it, expect } from 'vitest';\",\n    \"import { MockAmplitudeAI } from '@amplitude/ai/testing';\",\n    '',\n    \"describe('Amplitude AI verification', () => {\",\n  );\n\n  for (const agent of scanResult.agents) {\n    lines.push(\n      `  it('${agent.inferred_id} emits correct event sequence', () => {`,\n      '    const mock = new MockAmplitudeAI();',\n      `    const agent = mock.agent('${agent.inferred_id}');`,\n      \"    const session = agent.session({ userId: 'verify-user', sessionId: 'verify-session' });\",\n      '    session.runSync((s) => {',\n      \"      s.trackUserMessage('test message');\",\n      '    });',\n      '    const events = mock.getEvents();',\n      '    expect(events.length).toBeGreaterThanOrEqual(2);',\n      `    mock.assertEventTracked('[Agent] User Message', { '[Agent] Agent ID': '${agent.inferred_id}' });`,\n      \"    mock.assertSessionClosed('verify-session');\",\n      '  });',\n      '',\n    );\n  }\n\n  const shouldEmitMultiAgentTests =\n    (scanResult.is_multi_agent || scanResult.multi_agent_signals.length > 0) &&\n    scanResult.agents.length >= 2;\n\n  if (shouldEmitMultiAgentTests) {\n    const parentAgent = scanResult.agents[0];\n    const childAgent = scanResult.agents[1];\n\n    if (parentAgent && childAgent) {\n      lines.push(\n        \"  it('multi-agent delegation via runAs', () => {\",\n        '    const mock = new MockAmplitudeAI();',\n        `    const parent = mock.agent('${parentAgent.inferred_id}');`,\n        `    const child = parent.child('${childAgent.inferred_id}');`,\n        \"    const session = parent.session({ userId: 'verify-user', sessionId: 'verify-multi' });\",\n        '    session.runSync((s) => {',\n        '      s.runAsSync(child, (cs) => {',\n        \"        cs.trackUserMessage('delegated task');\",\n        '      });',\n        '    });',\n        `    const childEvents = mock.eventsForAgent('${childAgent.inferred_id}');`,\n        '    expect(childEvents.length).toBeGreaterThan(0);',\n        \"    expect(childEvents[0].event_properties?.['[Agent] Session ID']).toBe('verify-multi');\",\n        '  });',\n        '',\n      );\n\n      const childAgents = scanResult.agents.slice(0, 2);\n      lines.push(\n        \"  it('parallel fan-out shares session', async () => {\",\n        '    const mock = new MockAmplitudeAI();',\n        `    const orchestrator = mock.agent('${parentAgent.inferred_id}');`,\n        '    const children = [',\n      );\n      for (const child of childAgents) {\n        lines.push(\n          `      orchestrator.child('${child.inferred_id}'),`,\n        );\n      }\n      lines.push(\n        '    ];',\n        \"    const session = orchestrator.session({ userId: 'verify-user', sessionId: 'verify-parallel' });\",\n        '    await session.run(async (s) => {',\n        '      await Promise.all(children.map((child) =>',\n        '        s.runAs(child, async (cs) => {',\n        \"          cs.trackUserMessage('parallel task');\",\n        '        }),',\n        '      ));',\n        '    });',\n        '    for (const child of children) {',\n        '      const events = mock.eventsForAgent(child.agentId);',\n        '      expect(events.length).toBeGreaterThan(0);',\n        \"      expect(events[0].event_properties?.['[Agent] Session ID']).toBe('verify-parallel');\",\n        '    }',\n        '  });',\n        '',\n      );\n\n      lines.push(\n        \"  it('tool calls inside runAs are attributed to child agent', () => {\",\n        '    const mock = new MockAmplitudeAI();',\n        `    const parent = mock.agent('${parentAgent.inferred_id}');`,\n        `    const child = parent.child('${childAgent.inferred_id}');`,\n        \"    const session = parent.session({ userId: 'verify-user', sessionId: 'verify-tools' });\",\n        '    session.runSync((s) => {',\n        '      s.runAsSync(child, (cs) => {',\n        \"        cs.trackToolCall('search_knowledge_base', 150, true);\",\n        '      });',\n        '    });',\n        \"    const toolEvents = mock.getEvents().filter(e => e.event_type === '[Agent] Tool Call');\",\n        '    expect(toolEvents.length).toBe(1);',\n        `    expect(toolEvents[0].event_properties?.['[Agent] Agent ID']).toBe('${childAgent.inferred_id}');`,\n        `    expect(toolEvents[0].event_properties?.['[Agent] Parent Agent ID']).toBe('${parentAgent.inferred_id}');`,\n        '  });',\n        '',\n      );\n\n      lines.push(\n        \"  it('runAs restores parent context after child throws', () => {\",\n        '    const mock = new MockAmplitudeAI();',\n        `    const parent = mock.agent('${parentAgent.inferred_id}');`,\n        `    const faultyChild = parent.child('${childAgent.inferred_id}-faulty');`,\n        \"    const session = parent.session({ userId: 'verify-user', sessionId: 'verify-error' });\",\n        '    session.runSync((s) => {',\n        '      try {',\n        '        s.runAsSync(faultyChild, () => {',\n        \"          throw new Error('child failed');\",\n        '        });',\n        '      } catch {',\n        \"        s.trackUserMessage('recovering from child failure');\",\n        '      }',\n        '    });',\n        `    const parentEvents = mock.eventsForAgent('${parentAgent.inferred_id}');`,\n        \"    const recoveryMsg = parentEvents.find(e => e.event_type === '[Agent] User Message');\",\n        '    expect(recoveryMsg).toBeDefined();',\n        `    expect(recoveryMsg?.event_properties?.['[Agent] Agent ID']).toBe('${parentAgent.inferred_id}');`,\n        '  });',\n        '',\n      );\n    }\n  }\n\n  lines.push('});', '');\n\n  return lines.join('\\n');\n}\n"],"mappings":";AAEA,SAAgB,mBAAmB,YAAgC;CACjE,MAAMA,QAAkB,EAAE;AAE1B,OAAM,KACJ,kDACA,4DACA,IACA,gDACD;AAED,MAAK,MAAM,SAAS,WAAW,OAC7B,OAAM,KACJ,SAAS,MAAM,YAAY,0CAC3B,2CACA,iCAAiC,MAAM,YAAY,MACnD,8FACA,gCACA,6CACA,WACA,wCACA,wDACA,8EAA8E,MAAM,YAAY,QAChG,mDACA,SACA,GACD;AAOH,MAHG,WAAW,kBAAkB,WAAW,oBAAoB,SAAS,MACtE,WAAW,OAAO,UAAU,GAEC;EAC7B,MAAM,cAAc,WAAW,OAAO;EACtC,MAAM,aAAa,WAAW,OAAO;AAErC,MAAI,eAAe,YAAY;AAC7B,SAAM,KACJ,oDACA,2CACA,kCAAkC,YAAY,YAAY,MAC1D,mCAAmC,WAAW,YAAY,MAC1D,6FACA,gCACA,sCACA,kDACA,aACA,WACA,gDAAgD,WAAW,YAAY,MACvE,sDACA,6FACA,SACA,GACD;GAED,MAAM,cAAc,WAAW,OAAO,MAAM,GAAG,EAAE;AACjD,SAAM,KACJ,yDACA,2CACA,wCAAwC,YAAY,YAAY,MAChE,yBACD;AACD,QAAK,MAAM,SAAS,YAClB,OAAM,KACJ,6BAA6B,MAAM,YAAY,KAChD;AAEH,SAAM,KACJ,UACA,sGACA,wCACA,mDACA,0CACA,mDACA,eACA,aACA,WACA,uCACA,4DACA,mDACA,6FACA,SACA,SACA,GACD;AAED,SAAM,KACJ,yEACA,2CACA,kCAAkC,YAAY,YAAY,MAC1D,mCAAmC,WAAW,YAAY,MAC1D,6FACA,gCACA,sCACA,iEACA,aACA,WACA,8FACA,0CACA,0EAA0E,WAAW,YAAY,MACjG,iFAAiF,YAAY,YAAY,MACzG,SACA,GACD;AAED,SAAM,KACJ,oEACA,2CACA,kCAAkC,YAAY,YAAY,MAC1D,yCAAyC,WAAW,YAAY,aAChE,6FACA,gCACA,eACA,4CACA,8CACA,eACA,mBACA,gEACA,WACA,WACA,iDAAiD,YAAY,YAAY,MACzE,4FACA,0CACA,yEAAyE,YAAY,YAAY,MACjG,SACA,GACD;;;AAIL,OAAM,KAAK,OAAO,GAAG;AAErB,QAAO,MAAM,KAAK,KAAK"}

package/dist/mcp/index.d.ts

@@ -1,4 +1,5 @@
 import { GENERATED_FILES, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, MToolName } from "./contract.js";
+import { ScanResult, scanProject } from "./scan-project.js";
 import { IntegrationPattern, getIntegrationPatterns } from "./patterns.js";
 import { buildInstrumentationGuidance, createServer, runMcpServer } from "./server.js";
-export { GENERATED_FILES, IntegrationPattern, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, MToolName, buildInstrumentationGuidance, createServer, getIntegrationPatterns, runMcpServer };
+export { GENERATED_FILES, IntegrationPattern, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, MToolName, type ScanResult, buildInstrumentationGuidance, createServer, getIntegrationPatterns, runMcpServer, scanProject };

package/dist/mcp/index.js

@@ -1,5 +1,6 @@
 import { GENERATED_FILES, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS } from "./contract.js";
 import { getIntegrationPatterns } from "./patterns.js";
+import { scanProject } from "./scan-project.js";
 import { buildInstrumentationGuidance, createServer, runMcpServer } from "./server.js";
 
-export { GENERATED_FILES, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, buildInstrumentationGuidance, createServer, getIntegrationPatterns, runMcpServer };
+export { GENERATED_FILES, MCP_PROMPTS, MCP_RESOURCES, MCP_SERVER_NAME, MCP_TOOLS, buildInstrumentationGuidance, createServer, getIntegrationPatterns, runMcpServer, scanProject };

package/dist/mcp/instrument-file.d.ts

@@ -0,0 +1,14 @@
+//#region src/mcp/instrument-file.d.ts
+interface InstrumentFileOptions {
+  source: string;
+  filePath: string;
+  tier: 'quick_start' | 'standard' | 'advanced';
+  bootstrapImportPath: string;
+  agentId: string;
+  description?: string | null;
+  providers: string[];
+}
+declare function instrumentFile(opts: InstrumentFileOptions): string;
+//#endregion
+export { InstrumentFileOptions, instrumentFile };
+//# sourceMappingURL=instrument-file.d.ts.map

package/dist/mcp/instrument-file.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrument-file.d.ts","names":[],"sources":["../../src/mcp/instrument-file.ts"],"sourcesContent":[],"mappings":";UAAiB,qBAAA;EAAA,MAAA,EAAA,MAAA;EA0KD,QAAA,EAAA,MAAA;;;;;;;iBAAA,cAAA,OAAqB"}

package/dist/mcp/instrument-file.js

@@ -0,0 +1,139 @@
+//#region src/mcp/instrument-file.ts
+const PROVIDER_IMPORT_MAP = {
+	openai: {
+		module: "openai",
+		defaultExport: "OpenAI",
+		namedExport: "openai"
+	},
+	"@anthropic-ai/sdk": {
+		module: "@anthropic-ai/sdk",
+		defaultExport: "Anthropic",
+		namedExport: "anthropic"
+	},
+	"@google/generative-ai": {
+		module: "@google/generative-ai",
+		defaultExport: "GoogleGenerativeAI",
+		namedExport: "gemini"
+	},
+	"@google/genai": {
+		module: "@google/genai",
+		defaultExport: "GoogleGenAI",
+		namedExport: "genai"
+	},
+	"@mistralai/mistralai": {
+		module: "@mistralai/mistralai",
+		defaultExport: "Mistral",
+		namedExport: "mistral"
+	},
+	"@azure/openai": {
+		module: "@azure/openai",
+		defaultExport: "AzureOpenAI",
+		namedExport: "azureOpenai"
+	},
+	"cohere-ai": {
+		module: "cohere-ai",
+		defaultExport: "CohereClient",
+		namedExport: "cohere"
+	}
+};
+function matchConstructor(source, constructorName) {
+	const results = [];
+	const re = new RegExp(`new\\s+${constructorName}\\s*\\(`, "g");
+	for (const m of source.matchAll(re)) {
+		const openIdx = (m.index ?? 0) + m[0].length - 1;
+		let depth = 1;
+		let i = openIdx + 1;
+		while (i < source.length && depth > 0) {
+			if (source[i] === "(") depth++;
+			else if (source[i] === ")") depth--;
+			i++;
+		}
+		results.push({
+			start: m.index ?? 0,
+			end: i,
+			fullMatch: source.slice(m.index ?? 0, i)
+		});
+	}
+	return results;
+}
+const ROUTE_HANDLER_RE = /export\s+async\s+function\s+(?:POST|GET|PUT|DELETE)\b/;
+const EXPRESS_HANDLER_RE = /(?:app|router)\.\s*(?:get|post|put|delete)\s*\(\s*['"][^'"]+['"]\s*,\s*(?:async\s+)?\(/;
+const HONO_HANDLER_RE = /(?:app|router)\.\s*(?:get|post|put|delete)\s*\(\s*['"][^'"]+['"]\s*,\s*(?:async\s+)?\(\s*c\b/;
+function replaceProviderImports(source, providers, bootstrapImportPath) {
+	let result = source;
+	const namedImports = [];
+	for (const provider of providers) {
+		const mapping = PROVIDER_IMPORT_MAP[provider];
+		if (!mapping) continue;
+		const defaultImportRe = /* @__PURE__ */ new RegExp(`import\\s+${mapping.defaultExport}\\s+from\\s+['"]${mapping.module}['"];?`);
+		const namedImportRe = /* @__PURE__ */ new RegExp(`import\\s*\\{\\s*${mapping.defaultExport}\\s*\\}\\s*from\\s+['"]${mapping.module}['"];?`);
+		if (defaultImportRe.test(result)) {
+			result = result.replace(defaultImportRe, "");
+			namedImports.push(mapping.namedExport);
+		} else if (namedImportRe.test(result)) {
+			result = result.replace(namedImportRe, "");
+			namedImports.push(mapping.namedExport);
+		}
+		const matches = matchConstructor(result, mapping.defaultExport);
+		for (let i = matches.length - 1; i >= 0; i--) {
+			const match = matches[i];
+			if (match) result = result.slice(0, match.start) + mapping.namedExport + result.slice(match.end);
+		}
+	}
+	if (namedImports.length > 0) result = `import { ${namedImports.join(", ")} } from '${bootstrapImportPath}';\n` + result;
+	return result;
+}
+function addSessionWrapping(source, agentId, bootstrapImportPath) {
+	let result = source;
+	const existingImportMatch = (/* @__PURE__ */ new RegExp(`import\\s*\\{([^}]*)\\}\\s*from\\s*['"]${bootstrapImportPath.replace(/[.*+?^${}()|[\]\\]/g, "\\$&")}['"]`)).exec(result);
+	if (existingImportMatch) {
+		const existingNames = existingImportMatch[1] ?? "";
+		if (!existingNames.split(",").map((s) => s.trim()).includes("ai")) {
+			const newNames = existingNames.trim() ? `ai, ${existingNames.trim()}` : "ai";
+			result = result.replace(existingImportMatch[0], `import { ${newNames} } from '${bootstrapImportPath}'`);
+		}
+	} else if (!result.includes(`from '${bootstrapImportPath}'`) && !result.includes(`from "${bootstrapImportPath}"`)) result = `import { ai } from '${bootstrapImportPath}';\n${result}`;
+	const agentLine = `const agent = ai.agent('${agentId}');\n`;
+	if (ROUTE_HANDLER_RE.test(result)) {
+		const handlerMatch = result.match(/export\s+async\s+function\s+(?:POST|GET|PUT|DELETE)\s*\([^)]*\)\s*\{/);
+		result = result.replace(/(export\s+async\s+function\s+(?:POST|GET|PUT|DELETE)\s*\([^)]*\)\s*\{)/, `$1\n  ${agentLine.trim()}\n  const { messages, userId, sessionId } = await req.json();\n  return agent.session({ userId, sessionId }).run(async (s) => {`);
+		if (handlerMatch?.index != null) {
+			const openBraceIdx = result.indexOf("{", handlerMatch.index);
+			if (openBraceIdx >= 0) {
+				let depth = 1;
+				let i = openBraceIdx + 1;
+				while (i < result.length && depth > 0) {
+					if (result[i] === "{") depth++;
+					else if (result[i] === "}") depth--;
+					i++;
+				}
+				const closingBraceIdx = i - 1;
+				result = `${result.slice(0, closingBraceIdx)}  });\n${result.slice(closingBraceIdx)}`;
+			}
+		}
+	} else if (EXPRESS_HANDLER_RE.test(result) || HONO_HANDLER_RE.test(result)) result = result.replace(/((?:app|router)\.\s*(?:get|post|put|delete)\s*\(\s*['"][^'"]+['"]\s*,\s*(?:async\s+)?\([^)]*\)\s*(?:=>)?\s*\{)/, `$1\n    ${agentLine.trim()}\n    return agent.session({ userId: 'todo-extract-user-id', sessionId: 'todo-extract-session-id' }).run(async (s) => {`);
+	return result;
+}
+function addUserMessageTracking(source) {
+	const match = /(const\s+\{[^}]*\}\s*=\s*(?:await\s+)?(?:req\.body|request\.json\(\)|await\s+request\.json\(\)))/.exec(source);
+	if (match) return source.replace(match[0], `${match[0]};\n    // TODO: extract user message and call s.trackUserMessage(userMessage)`);
+	return source;
+}
+function addFlushBeforeReturn(source) {
+	return source.replace(/(\n)([ \t]*)(return\s+(?:Response\.json|new\s+Response|NextResponse|res\.json|res\.send)\s*\()/g, "$1$2await ai.flush();\n$2$3");
+}
+function instrumentFile(opts) {
+	if (opts.tier === "quick_start") return opts.source;
+	let result = opts.source;
+	result = replaceProviderImports(result, opts.providers, opts.bootstrapImportPath);
+	if (opts.tier === "advanced") {
+		result = addSessionWrapping(result, opts.agentId, opts.bootstrapImportPath);
+		result = addUserMessageTracking(result);
+		result = addFlushBeforeReturn(result);
+	}
+	return result;
+}
+
+//#endregion
+export { instrumentFile };
+//# sourceMappingURL=instrument-file.js.map

package/dist/mcp/instrument-file.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrument-file.js","names":["PROVIDER_IMPORT_MAP: Record<string, { module: string; defaultExport: string; namedExport: string }>","results: Array<{ start: number; end: number; fullMatch: string }>","namedImports: string[]"],"sources":["../../src/mcp/instrument-file.ts"],"sourcesContent":["export interface InstrumentFileOptions {\n  source: string;\n  filePath: string;\n  tier: 'quick_start' | 'standard' | 'advanced';\n  bootstrapImportPath: string;\n  agentId: string;\n  description?: string | null;\n  providers: string[];\n}\n\nconst PROVIDER_IMPORT_MAP: Record<string, { module: string; defaultExport: string; namedExport: string }> = {\n  openai: { module: 'openai', defaultExport: 'OpenAI', namedExport: 'openai' },\n  '@anthropic-ai/sdk': { module: '@anthropic-ai/sdk', defaultExport: 'Anthropic', namedExport: 'anthropic' },\n  '@google/generative-ai': { module: '@google/generative-ai', defaultExport: 'GoogleGenerativeAI', namedExport: 'gemini' },\n  '@google/genai': { module: '@google/genai', defaultExport: 'GoogleGenAI', namedExport: 'genai' },\n  '@mistralai/mistralai': { module: '@mistralai/mistralai', defaultExport: 'Mistral', namedExport: 'mistral' },\n  '@azure/openai': { module: '@azure/openai', defaultExport: 'AzureOpenAI', namedExport: 'azureOpenai' },\n  'cohere-ai': { module: 'cohere-ai', defaultExport: 'CohereClient', namedExport: 'cohere' },\n};\n\n// Balanced-paren constructor matcher: handles nested parens like new OpenAI({ apiKey: getKey() })\nfunction matchConstructor(source: string, constructorName: string): Array<{ start: number; end: number; fullMatch: string }> {\n  const results: Array<{ start: number; end: number; fullMatch: string }> = [];\n  const re = new RegExp(`new\\\\s+${constructorName}\\\\s*\\\\(`, 'g');\n  for (const m of source.matchAll(re)) {\n    const openIdx = (m.index ?? 0) + m[0].length - 1;\n    let depth = 1;\n    let i = openIdx + 1;\n    while (i < source.length && depth > 0) {\n      if (source[i] === '(') depth++;\n      else if (source[i] === ')') depth--;\n      i++;\n    }\n    results.push({ start: m.index ?? 0, end: i, fullMatch: source.slice(m.index ?? 0, i) });\n  }\n  return results;\n}\n\nconst ROUTE_HANDLER_RE =\n  /export\\s+async\\s+function\\s+(?:POST|GET|PUT|DELETE)\\b/;\nconst EXPRESS_HANDLER_RE =\n  /(?:app|router)\\.\\s*(?:get|post|put|delete)\\s*\\(\\s*['\"][^'\"]+['\"]\\s*,\\s*(?:async\\s+)?\\(/;\nconst HONO_HANDLER_RE =\n  /(?:app|router)\\.\\s*(?:get|post|put|delete)\\s*\\(\\s*['\"][^'\"]+['\"]\\s*,\\s*(?:async\\s+)?\\(\\s*c\\b/;\n\nfunction replaceProviderImports(\n  source: string,\n  providers: string[],\n  bootstrapImportPath: string,\n): string {\n  let result = source;\n  const namedImports: string[] = [];\n\n  for (const provider of providers) {\n    const mapping = PROVIDER_IMPORT_MAP[provider];\n    if (!mapping) continue;\n\n    // Match default import: import OpenAI from 'openai'\n    const defaultImportRe = new RegExp(\n      `import\\\\s+${mapping.defaultExport}\\\\s+from\\\\s+['\"]${mapping.module}['\"];?`,\n    );\n    // Match named import: import { AzureOpenAI } from '@azure/openai'\n    const namedImportRe = new RegExp(\n      `import\\\\s*\\\\{\\\\s*${mapping.defaultExport}\\\\s*\\\\}\\\\s*from\\\\s+['\"]${mapping.module}['\"];?`,\n    );\n    if (defaultImportRe.test(result)) {\n      result = result.replace(defaultImportRe, '');\n      namedImports.push(mapping.namedExport);\n    } else if (namedImportRe.test(result)) {\n      result = result.replace(namedImportRe, '');\n      namedImports.push(mapping.namedExport);\n    }\n\n    // Replace constructors with pre-wrapped named imports using balanced-paren matching\n    const matches = matchConstructor(result, mapping.defaultExport);\n    for (let i = matches.length - 1; i >= 0; i--) {\n      const match = matches[i];\n      if (match) {\n        result = result.slice(0, match.start) + mapping.namedExport + result.slice(match.end);\n      }\n    }\n  }\n\n  if (namedImports.length > 0) {\n    const importLine = `import { ${namedImports.join(', ')} } from '${bootstrapImportPath}';\\n`;\n    result = importLine + result;\n  }\n\n  return result;\n}\n\nfunction addSessionWrapping(\n  source: string,\n  agentId: string,\n  bootstrapImportPath: string,\n): string {\n  let result = source;\n\n  const importFromPath = new RegExp(\n    `import\\\\s*\\\\{([^}]*)\\\\}\\\\s*from\\\\s*['\"]${bootstrapImportPath.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&')}['\"]`,\n  );\n  const existingImportMatch = importFromPath.exec(result);\n  if (existingImportMatch) {\n    const existingNames = existingImportMatch[1] ?? '';\n    const importedNames = existingNames.split(',').map(s => s.trim());\n    if (!importedNames.includes('ai')) {\n      const newNames = existingNames.trim() ? `ai, ${existingNames.trim()}` : 'ai';\n      result = result.replace(existingImportMatch[0], `import { ${newNames} } from '${bootstrapImportPath}'`);\n    }\n  } else if (!result.includes(`from '${bootstrapImportPath}'`) &&\n      !result.includes(`from \"${bootstrapImportPath}\"`)) {\n    result = `import { ai } from '${bootstrapImportPath}';\\n${result}`;\n  }\n\n  const agentLine = `const agent = ai.agent('${agentId}');\\n`;\n\n  // Wrap route handler body inside session.run()\n  if (ROUTE_HANDLER_RE.test(result)) {\n    const handlerMatch = result.match(\n      /export\\s+async\\s+function\\s+(?:POST|GET|PUT|DELETE)\\s*\\([^)]*\\)\\s*\\{/,\n    );\n    result = result.replace(\n      /(export\\s+async\\s+function\\s+(?:POST|GET|PUT|DELETE)\\s*\\([^)]*\\)\\s*\\{)/,\n      `$1\\n  ${agentLine.trim()}\\n  const { messages, userId, sessionId } = await req.json();\\n  return agent.session({ userId, sessionId }).run(async (s) => {`,\n    );\n    // Find the handler's closing brace using balanced-brace matching from the handler opening\n    if (handlerMatch?.index != null) {\n      const openBraceIdx = result.indexOf('{', handlerMatch.index);\n      if (openBraceIdx >= 0) {\n        let depth = 1;\n        let i = openBraceIdx + 1;\n        while (i < result.length && depth > 0) {\n          if (result[i] === '{') depth++;\n          else if (result[i] === '}') depth--;\n          i++;\n        }\n        const closingBraceIdx = i - 1;\n        result = `${result.slice(0, closingBraceIdx)}  });\\n${result.slice(closingBraceIdx)}`;\n      }\n    }\n  } else if (EXPRESS_HANDLER_RE.test(result) || HONO_HANDLER_RE.test(result)) {\n    result = result.replace(\n      /((?:app|router)\\.\\s*(?:get|post|put|delete)\\s*\\(\\s*['\"][^'\"]+['\"]\\s*,\\s*(?:async\\s+)?\\([^)]*\\)\\s*(?:=>)?\\s*\\{)/,\n      `$1\\n    ${agentLine.trim()}\\n    return agent.session({ userId: 'todo-extract-user-id', sessionId: 'todo-extract-session-id' }).run(async (s) => {`,\n    );\n  }\n\n  return result;\n}\n\nfunction addUserMessageTracking(source: string): string {\n  const requestBodyRe = /(const\\s+\\{[^}]*\\}\\s*=\\s*(?:await\\s+)?(?:req\\.body|request\\.json\\(\\)|await\\s+request\\.json\\(\\)))/;\n  const match = requestBodyRe.exec(source);\n  if (match) {\n    return source.replace(\n      match[0],\n      `${match[0]};\\n    // TODO: extract user message and call s.trackUserMessage(userMessage)`,\n    );\n  }\n  return source;\n}\n\nfunction addFlushBeforeReturn(source: string): string {\n  // Insert await ai.flush() before return statements in route handlers\n  return source.replace(\n    /(\\n)([ \\t]*)(return\\s+(?:Response\\.json|new\\s+Response|NextResponse|res\\.json|res\\.send)\\s*\\()/g,\n    '$1$2await ai.flush();\\n$2$3',\n  );\n}\n\nexport function instrumentFile(opts: InstrumentFileOptions): string {\n  if (opts.tier === 'quick_start') {\n    return opts.source;\n  }\n\n  let result = opts.source;\n\n  result = replaceProviderImports(result, opts.providers, opts.bootstrapImportPath);\n\n  if (opts.tier === 'advanced') {\n    result = addSessionWrapping(result, opts.agentId, opts.bootstrapImportPath);\n    result = addUserMessageTracking(result);\n    result = addFlushBeforeReturn(result);\n  }\n\n  return result;\n}\n"],"mappings":";AAUA,MAAMA,sBAAsG;CAC1G,QAAQ;EAAE,QAAQ;EAAU,eAAe;EAAU,aAAa;EAAU;CAC5E,qBAAqB;EAAE,QAAQ;EAAqB,eAAe;EAAa,aAAa;EAAa;CAC1G,yBAAyB;EAAE,QAAQ;EAAyB,eAAe;EAAsB,aAAa;EAAU;CACxH,iBAAiB;EAAE,QAAQ;EAAiB,eAAe;EAAe,aAAa;EAAS;CAChG,wBAAwB;EAAE,QAAQ;EAAwB,eAAe;EAAW,aAAa;EAAW;CAC5G,iBAAiB;EAAE,QAAQ;EAAiB,eAAe;EAAe,aAAa;EAAe;CACtG,aAAa;EAAE,QAAQ;EAAa,eAAe;EAAgB,aAAa;EAAU;CAC3F;AAGD,SAAS,iBAAiB,QAAgB,iBAAmF;CAC3H,MAAMC,UAAoE,EAAE;CAC5E,MAAM,KAAK,IAAI,OAAO,UAAU,gBAAgB,UAAU,IAAI;AAC9D,MAAK,MAAM,KAAK,OAAO,SAAS,GAAG,EAAE;EACnC,MAAM,WAAW,EAAE,SAAS,KAAK,EAAE,GAAG,SAAS;EAC/C,IAAI,QAAQ;EACZ,IAAI,IAAI,UAAU;AAClB,SAAO,IAAI,OAAO,UAAU,QAAQ,GAAG;AACrC,OAAI,OAAO,OAAO,IAAK;YACd,OAAO,OAAO,IAAK;AAC5B;;AAEF,UAAQ,KAAK;GAAE,OAAO,EAAE,SAAS;GAAG,KAAK;GAAG,WAAW,OAAO,MAAM,EAAE,SAAS,GAAG,EAAE;GAAE,CAAC;;AAEzF,QAAO;;AAGT,MAAM,mBACJ;AACF,MAAM,qBACJ;AACF,MAAM,kBACJ;AAEF,SAAS,uBACP,QACA,WACA,qBACQ;CACR,IAAI,SAAS;CACb,MAAMC,eAAyB,EAAE;AAEjC,MAAK,MAAM,YAAY,WAAW;EAChC,MAAM,UAAU,oBAAoB;AACpC,MAAI,CAAC,QAAS;EAGd,MAAM,kCAAkB,IAAI,OAC1B,aAAa,QAAQ,cAAc,kBAAkB,QAAQ,OAAO,QACrE;EAED,MAAM,gCAAgB,IAAI,OACxB,oBAAoB,QAAQ,cAAc,yBAAyB,QAAQ,OAAO,QACnF;AACD,MAAI,gBAAgB,KAAK,OAAO,EAAE;AAChC,YAAS,OAAO,QAAQ,iBAAiB,GAAG;AAC5C,gBAAa,KAAK,QAAQ,YAAY;aAC7B,cAAc,KAAK,OAAO,EAAE;AACrC,YAAS,OAAO,QAAQ,eAAe,GAAG;AAC1C,gBAAa,KAAK,QAAQ,YAAY;;EAIxC,MAAM,UAAU,iBAAiB,QAAQ,QAAQ,cAAc;AAC/D,OAAK,IAAI,IAAI,QAAQ,SAAS,GAAG,KAAK,GAAG,KAAK;GAC5C,MAAM,QAAQ,QAAQ;AACtB,OAAI,MACF,UAAS,OAAO,MAAM,GAAG,MAAM,MAAM,GAAG,QAAQ,cAAc,OAAO,MAAM,MAAM,IAAI;;;AAK3F,KAAI,aAAa,SAAS,EAExB,UADmB,YAAY,aAAa,KAAK,KAAK,CAAC,WAAW,oBAAoB,QAChE;AAGxB,QAAO;;AAGT,SAAS,mBACP,QACA,SACA,qBACQ;CACR,IAAI,SAAS;CAKb,MAAM,uCAHiB,IAAI,OACzB,0CAA0C,oBAAoB,QAAQ,uBAAuB,OAAO,CAAC,MACtG,EAC0C,KAAK,OAAO;AACvD,KAAI,qBAAqB;EACvB,MAAM,gBAAgB,oBAAoB,MAAM;AAEhD,MAAI,CADkB,cAAc,MAAM,IAAI,CAAC,KAAI,MAAK,EAAE,MAAM,CAAC,CAC9C,SAAS,KAAK,EAAE;GACjC,MAAM,WAAW,cAAc,MAAM,GAAG,OAAO,cAAc,MAAM,KAAK;AACxE,YAAS,OAAO,QAAQ,oBAAoB,IAAI,YAAY,SAAS,WAAW,oBAAoB,GAAG;;YAEhG,CAAC,OAAO,SAAS,SAAS,oBAAoB,GAAG,IACxD,CAAC,OAAO,SAAS,SAAS,oBAAoB,GAAG,CACnD,UAAS,uBAAuB,oBAAoB,MAAM;CAG5D,MAAM,YAAY,2BAA2B,QAAQ;AAGrD,KAAI,iBAAiB,KAAK,OAAO,EAAE;EACjC,MAAM,eAAe,OAAO,MAC1B,uEACD;AACD,WAAS,OAAO,QACd,0EACA,SAAS,UAAU,MAAM,CAAC,iIAC3B;AAED,MAAI,cAAc,SAAS,MAAM;GAC/B,MAAM,eAAe,OAAO,QAAQ,KAAK,aAAa,MAAM;AAC5D,OAAI,gBAAgB,GAAG;IACrB,IAAI,QAAQ;IACZ,IAAI,IAAI,eAAe;AACvB,WAAO,IAAI,OAAO,UAAU,QAAQ,GAAG;AACrC,SAAI,OAAO,OAAO,IAAK;cACd,OAAO,OAAO,IAAK;AAC5B;;IAEF,MAAM,kBAAkB,IAAI;AAC5B,aAAS,GAAG,OAAO,MAAM,GAAG,gBAAgB,CAAC,SAAS,OAAO,MAAM,gBAAgB;;;YAG9E,mBAAmB,KAAK,OAAO,IAAI,gBAAgB,KAAK,OAAO,CACxE,UAAS,OAAO,QACd,kHACA,WAAW,UAAU,MAAM,CAAC,yHAC7B;AAGH,QAAO;;AAGT,SAAS,uBAAuB,QAAwB;CAEtD,MAAM,QADgB,mGACM,KAAK,OAAO;AACxC,KAAI,MACF,QAAO,OAAO,QACZ,MAAM,IACN,GAAG,MAAM,GAAG,+EACb;AAEH,QAAO;;AAGT,SAAS,qBAAqB,QAAwB;AAEpD,QAAO,OAAO,QACZ,mGACA,8BACD;;AAGH,SAAgB,eAAe,MAAqC;AAClE,KAAI,KAAK,SAAS,cAChB,QAAO,KAAK;CAGd,IAAI,SAAS,KAAK;AAElB,UAAS,uBAAuB,QAAQ,KAAK,WAAW,KAAK,oBAAoB;AAEjF,KAAI,KAAK,SAAS,YAAY;AAC5B,WAAS,mBAAmB,QAAQ,KAAK,SAAS,KAAK,oBAAoB;AAC3E,WAAS,uBAAuB,OAAO;AACvC,WAAS,qBAAqB,OAAO;;AAGvC,QAAO"}

package/dist/mcp/scan-project.d.ts

@@ -0,0 +1,52 @@
+//#region src/mcp/scan-project.d.ts
+interface ScanResult {
+  project_name: string | null;
+  runtime: 'node';
+  language: 'typescript' | 'javascript';
+  framework: string | null;
+  providers: string[];
+  agent_frameworks: string[];
+  package_manager: string | null;
+  existing_instrumentation: {
+    has_amplitude_ai: boolean;
+    has_patch: boolean;
+    has_wrappers: boolean;
+    has_session_context: boolean;
+  };
+  agents: Array<{
+    inferred_id: string;
+    file: string;
+    call_sites: number;
+    is_instrumented: boolean;
+    is_route_handler: boolean;
+    inferred_description: string | null;
+    call_site_details: Array<{
+      line: number;
+      provider: string;
+      api: string;
+      instrumented: boolean;
+      containing_function: string | null;
+      code_context: string;
+    }>;
+    tool_definitions: string[];
+    function_definitions: string[];
+  }>;
+  total_call_sites: number;
+  instrumented_call_sites: number;
+  uninstrumented_call_sites: number;
+  is_multi_agent: boolean;
+  multi_agent_signals: string[];
+  has_streaming: boolean;
+  has_vercel_ai_sdk: boolean;
+  has_edge_runtime: boolean;
+  has_assistants_api: boolean;
+  has_langgraph: boolean;
+  message_queue_deps: string[];
+  has_frontend_deps: boolean;
+  recommendations: string[];
+  recommended_tier: 'quick_start' | 'standard' | 'advanced';
+}
+declare function scanProject(rootPath: string): ScanResult;
+//#endregion
+export { ScanResult, scanProject };
+//# sourceMappingURL=scan-project.d.ts.map

package/dist/mcp/scan-project.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"scan-project.d.ts","names":[],"sources":["../../src/mcp/scan-project.ts"],"sourcesContent":[],"mappings":";UASiB,UAAA;EAAA,YAAA,EAAU,MAAA,GAAA,IAqBJ;EAsLP,OAAA,EAAA,MAAW;;;;;;;;;;;;UA7LjB;;;;;;;uBAOa;;;;;;;;;;;;;;;;;;;;;;;;;;iBAsLP,WAAA,oBAA+B"}

package/dist/mcp/scan-project.js

@@ -0,0 +1,309 @@
+import { analyzeFileInstrumentation } from "./validate-file.js";
+import { existsSync, readFileSync, readdirSync, statSync } from "node:fs";
+import { basename, dirname, join, relative } from "node:path";
+
+//#region src/mcp/scan-project.ts
+const SOURCE_EXTENSIONS = new Set([
+	".ts",
+	".tsx",
+	".js",
+	".jsx"
+]);
+const SKIP_DIRS = new Set([
+	"node_modules",
+	"dist",
+	".next",
+	".git",
+	"__tests__",
+	"test"
+]);
+const MAX_DEPTH = 5;
+const FRAMEWORK_DEPS = [
+	["next", "nextjs"],
+	["express", "express"],
+	["fastify", "fastify"],
+	["hono", "hono"],
+	["@remix-run/node", "remix"]
+];
+const PROVIDER_DEPS = [
+	"openai",
+	"@anthropic-ai/sdk",
+	"@google/generative-ai",
+	"@google/genai",
+	"@aws-sdk/client-bedrock-runtime",
+	"@mistralai/mistralai",
+	"@azure/openai",
+	"cohere-ai"
+];
+const AGENT_FRAMEWORK_DEPS = [
+	"langchain",
+	"@langchain/core",
+	"llamaindex",
+	"@openai/agents",
+	"crewai"
+];
+const MESSAGE_QUEUE_DEPS = [
+	"bullmq",
+	"bull",
+	"ioredis",
+	"amqplib",
+	"@aws-sdk/client-sqs",
+	"kafkajs",
+	"@google-cloud/pubsub"
+];
+const FRONTEND_DEPS = [
+	"react",
+	"vue",
+	"svelte",
+	"@sveltejs/kit",
+	"angular",
+	"@angular/core"
+];
+const STREAMING_RE = /streamText\s*\(|useChat\s*\(|stream\s*:\s*true/;
+const VERCEL_AI_SDK_DEPS = [
+	"ai",
+	"@ai-sdk/openai",
+	"@ai-sdk/anthropic",
+	"@ai-sdk/google",
+	"@ai-sdk/mistral"
+];
+const VERCEL_AI_SDK_RE = /\b(?:streamText|generateText|streamObject|generateObject|useChat|useCompletion|useAssistant)\s*\(/;
+const EDGE_RUNTIME_RE = /runtime\s*=\s*['"]edge['"]/;
+const ASSISTANTS_API_RE = /\.beta\.(?:threads|assistants)\./;
+const LANGGRAPH_DEPS = ["@langchain/langgraph"];
+const MULTI_AGENT_CODE_RE = /\.child\s*\(|\.runAs\s*\(|\.runAsSync\s*\(/;
+const ROUTE_HANDLER_RE = /export\s+async\s+function\s+(?:POST|GET|PUT|DELETE)\b|app\.\s*(?:get|post|put|delete)\s*\(|router\./;
+function collectSourceFiles(dir, rootPath, depth) {
+	if (depth > MAX_DEPTH) return [];
+	let entries;
+	try {
+		entries = readdirSync(dir);
+	} catch {
+		return [];
+	}
+	const files = [];
+	for (const entry of entries) {
+		if (SKIP_DIRS.has(entry)) continue;
+		const fullPath = join(dir, entry);
+		let stat;
+		try {
+			stat = statSync(fullPath);
+		} catch {
+			continue;
+		}
+		if (stat.isDirectory()) files.push(...collectSourceFiles(fullPath, rootPath, depth + 1));
+		else if (stat.isFile()) {
+			const ext = entry.slice(entry.lastIndexOf("."));
+			if (SOURCE_EXTENSIONS.has(ext)) files.push(fullPath);
+		}
+	}
+	return files;
+}
+function inferAgentId(filePath) {
+	const base = basename(filePath).replace(/\.[^.]+$/, "");
+	if (base === "route" || base === "index") return basename(dirname(filePath)) || base;
+	return base;
+}
+function inferDescription(filePath, isRouteHandler) {
+	const agentId = inferAgentId(filePath);
+	if (isRouteHandler) return `Route handler agent: ${agentId}`;
+	if (/agent|worker|service/i.test(filePath)) return `Background agent: ${agentId}`;
+	return null;
+}
+function readPackageJson(rootPath) {
+	const pkgPath = join(rootPath, "package.json");
+	if (!existsSync(pkgPath)) return {
+		name: null,
+		allDeps: /* @__PURE__ */ new Set()
+	};
+	try {
+		const raw = readFileSync(pkgPath, "utf8");
+		const pkg = JSON.parse(raw);
+		const name = typeof pkg.name === "string" ? pkg.name : null;
+		const deps = pkg.dependencies ?? {};
+		const devDeps = pkg.devDependencies ?? {};
+		return {
+			name,
+			allDeps: new Set([...Object.keys(deps), ...Object.keys(devDeps)])
+		};
+	} catch {
+		return {
+			name: null,
+			allDeps: /* @__PURE__ */ new Set()
+		};
+	}
+}
+function scanProject(rootPath) {
+	const { name: projectName, allDeps } = readPackageJson(rootPath);
+	let framework = null;
+	for (const [dep, label] of FRAMEWORK_DEPS) if (allDeps.has(dep)) {
+		framework = label;
+		break;
+	}
+	const providers = PROVIDER_DEPS.filter((dep) => allDeps.has(dep));
+	const agentFrameworks = AGENT_FRAMEWORK_DEPS.filter((dep) => allDeps.has(dep));
+	let packageManager = null;
+	if (existsSync(join(rootPath, "pnpm-lock.yaml"))) packageManager = "pnpm";
+	else if (existsSync(join(rootPath, "yarn.lock"))) packageManager = "yarn";
+	else if (existsSync(join(rootPath, "package-lock.json"))) packageManager = "npm";
+	const isTypeScript = existsSync(join(rootPath, "tsconfig.json"));
+	const hasAmplitudeAiDep = allDeps.has("@amplitude/ai");
+	const sourceFiles = collectSourceFiles(rootPath, rootPath, 0);
+	let totalCallSites = 0;
+	let instrumentedCallSites = 0;
+	let uninstrumentedCallSites = 0;
+	let globalHasPatch = false;
+	let globalHasSessionContext = false;
+	let globalHasAmplitudeImport = hasAmplitudeAiDep;
+	let hasStreaming = false;
+	let hasVercelAiSdkUsage = false;
+	let hasEdgeRuntime = false;
+	let hasAssistantsApi = false;
+	let hasMultiAgentCodePatterns = false;
+	const globalWrappedProviders = /* @__PURE__ */ new Set();
+	const CONSTRUCTOR_TO_PROVIDER = {
+		OpenAI: "openai",
+		Anthropic: "anthropic",
+		Gemini: "gemini",
+		GoogleGenerativeAI: "gemini",
+		GoogleGenAI: "gemini",
+		AzureOpenAI: "azure-openai",
+		Bedrock: "bedrock",
+		Mistral: "mistral",
+		CohereClient: "cohere"
+	};
+	const agents = [];
+	const filesWithCallSites = /* @__PURE__ */ new Set();
+	const providerImportsPerFile = /* @__PURE__ */ new Map();
+	for (const filePath of sourceFiles) {
+		let content;
+		try {
+			content = readFileSync(filePath, "utf8");
+		} catch {
+			continue;
+		}
+		const analysis = analyzeFileInstrumentation(content);
+		if (analysis.has_amplitude_import) globalHasAmplitudeImport = true;
+		if (analysis.has_session_context) globalHasSessionContext = true;
+		if (/\bpatch\s*\(\s*\{/.test(content)) globalHasPatch = true;
+		if (STREAMING_RE.test(content) && analysis.total_call_sites > 0) hasStreaming = true;
+		if (VERCEL_AI_SDK_RE.test(content)) hasVercelAiSdkUsage = true;
+		if (EDGE_RUNTIME_RE.test(content)) hasEdgeRuntime = true;
+		if (ASSISTANTS_API_RE.test(content)) hasAssistantsApi = true;
+		if (MULTI_AGENT_CODE_RE.test(content)) hasMultiAgentCodePatterns = true;
+		if (analysis.has_amplitude_import || /\bamplitude\s*:/.test(content)) {
+			for (const m of content.matchAll(/new\s+(OpenAI|Anthropic|Gemini|AzureOpenAI|Bedrock|Mistral|GoogleGenerativeAI|GoogleGenAI|CohereClient)\s*\(/g)) {
+				const provLabel = CONSTRUCTOR_TO_PROVIDER[m[1] ?? ""];
+				if (provLabel && /\bamplitude\s*:/.test(content)) globalWrappedProviders.add(provLabel);
+			}
+			if (/\.wrap\s*\(/.test(content)) for (const site of analysis.call_sites) globalWrappedProviders.add(site.provider);
+		}
+		const fileProviders = /* @__PURE__ */ new Set();
+		for (const site of analysis.call_sites) fileProviders.add(site.provider);
+		if (fileProviders.size > 0) {
+			const relPath = relative(rootPath, filePath);
+			providerImportsPerFile.set(relPath, fileProviders);
+		}
+		if (analysis.total_call_sites > 0) {
+			const relPath = relative(rootPath, filePath);
+			filesWithCallSites.add(relPath);
+			totalCallSites += analysis.total_call_sites;
+			instrumentedCallSites += analysis.instrumented;
+			uninstrumentedCallSites += analysis.uninstrumented;
+			const isRouteHandler = ROUTE_HANDLER_RE.test(content);
+			const inferredId = inferAgentId(filePath);
+			const allInstrumented = analysis.uninstrumented === 0;
+			agents.push({
+				inferred_id: inferredId,
+				file: relPath,
+				call_sites: analysis.total_call_sites,
+				is_instrumented: allInstrumented,
+				is_route_handler: isRouteHandler,
+				inferred_description: inferDescription(filePath, isRouteHandler),
+				call_site_details: analysis.call_sites.map((cs) => ({
+					line: cs.line,
+					provider: cs.provider,
+					api: cs.api,
+					instrumented: cs.instrumented,
+					containing_function: cs.containing_function,
+					code_context: cs.code_context
+				})),
+				tool_definitions: analysis.tool_definitions,
+				function_definitions: analysis.function_definitions
+			});
+		}
+	}
+	const globalHasWrappers = globalWrappedProviders.size > 0;
+	if (globalHasWrappers && globalHasAmplitudeImport) {
+		for (const agent of agents) if (!agent.is_instrumented) {
+			if ([...new Set(agent.call_site_details.map((cs) => cs.provider))].every((p) => globalWrappedProviders.has(p))) {
+				agent.is_instrumented = true;
+				const delta = agent.call_sites;
+				uninstrumentedCallSites -= delta;
+				instrumentedCallSites += delta;
+			}
+		}
+	}
+	const multiAgentSignals = [];
+	if (filesWithCallSites.size > 1) multiAgentSignals.push(`LLM call sites found in ${filesWithCallSites.size} separate files: ${[...filesWithCallSites].join(", ")}`);
+	if (agentFrameworks.length > 0) multiAgentSignals.push(`Agent framework dependencies detected: ${agentFrameworks.join(", ")}`);
+	const allProviders = /* @__PURE__ */ new Set();
+	for (const provSet of providerImportsPerFile.values()) for (const p of provSet) allProviders.add(p);
+	if (allProviders.size > 1) multiAgentSignals.push(`Multiple LLM providers in use: ${[...allProviders].join(", ")}`);
+	for (const agent of agents) {
+		const toolNames = agent.tool_definitions;
+		const callFunctions = agent.call_site_details.map((cs) => cs.containing_function).filter(Boolean);
+		const overlapping = toolNames.filter((t) => callFunctions.includes(t));
+		if (overlapping.length > 0) multiAgentSignals.push(`File ${agent.file}: tool definitions ${overlapping.join(", ")} also contain LLM calls — possible delegation-as-tools pattern`);
+	}
+	if (hasMultiAgentCodePatterns) multiAgentSignals.push("Amplitude AI SDK delegation APIs detected: .child() or .runAs() — multi-agent confirmed");
+	const isMultiAgent = hasMultiAgentCodePatterns;
+	const messageQueueDeps = MESSAGE_QUEUE_DEPS.filter((dep) => allDeps.has(dep));
+	const hasFrontendDeps = FRONTEND_DEPS.some((dep) => allDeps.has(dep));
+	const hasVercelAiSdk = VERCEL_AI_SDK_DEPS.some((dep) => allDeps.has(dep)) || hasVercelAiSdkUsage;
+	const hasLanggraph = LANGGRAPH_DEPS.some((dep) => allDeps.has(dep));
+	const recommendedTier = "advanced";
+	const recommendations = [];
+	if (hasStreaming) recommendations.push("Streaming detected: keep sessions open until stream is fully consumed. Use session.run() with an awaited stream, not a fire-and-forget pattern.");
+	if (messageQueueDeps.length > 0) recommendations.push(`Message queue deps detected (${messageQueueDeps.join(", ")}): enable propagateContext in the bootstrap file and use injectContext/extractContext to correlate events across services.`);
+	if (hasFrontendDeps) recommendations.push("Frontend framework detected alongside backend: pass browserSessionId and deviceId from frontend request headers to agent.session() for session replay linking.");
+	if (hasVercelAiSdk) recommendations.push("Vercel AI SDK detected. Provider wrappers instrument the underlying provider SDK (openai, @anthropic-ai/sdk), not the Vercel AI SDK abstraction layer (streamText, generateText). If you also have the underlying provider as a direct dependency, wrappers will work because Vercel AI SDK delegates to them internally. Otherwise, use Tier 1 (patch) which intercepts at the transport level, or add the underlying provider SDK as a direct dependency.");
+	if (hasEdgeRuntime) recommendations.push("Edge Runtime detected. session.run() relies on AsyncLocalStorage which may not be available in Edge Runtime or Cloudflare Workers. Use explicit context passing instead of session.run(): call agent.trackUserMessage() and agent.trackAiMessage() directly with sessionId parameter.");
+	if (hasAssistantsApi) recommendations.push("OpenAI Assistants API detected (client.beta.threads/assistants). Provider wrappers do not auto-instrument the Assistants API. Use manual tracking with trackUserMessage/trackAiMessage, or migrate to the OpenAI Agents SDK which supports AmplitudeTracingProcessor.");
+	if (hasLanggraph) recommendations.push("LangGraph detected. LLM calls within LangGraph are captured via the LangChain AmplitudeCallbackHandler, but graph orchestration events (node transitions, checkpoints, human-in-the-loop) are not yet instrumented.");
+	if (globalHasAmplitudeImport && globalHasWrappers && globalHasSessionContext) recommendations.push("Project already has @amplitude/ai instrumentation with wrappers and session context. No re-instrumentation needed. Consider upgrading contentMode tiers, adding scoring, or expanding multi-agent coverage if applicable.");
+	return {
+		project_name: projectName,
+		runtime: "node",
+		language: isTypeScript ? "typescript" : "javascript",
+		framework,
+		providers,
+		agent_frameworks: agentFrameworks,
+		package_manager: packageManager,
+		existing_instrumentation: {
+			has_amplitude_ai: globalHasAmplitudeImport,
+			has_patch: globalHasPatch,
+			has_wrappers: globalHasWrappers,
+			has_session_context: globalHasSessionContext
+		},
+		agents,
+		total_call_sites: totalCallSites,
+		instrumented_call_sites: instrumentedCallSites,
+		uninstrumented_call_sites: uninstrumentedCallSites,
+		is_multi_agent: isMultiAgent,
+		multi_agent_signals: multiAgentSignals,
+		has_streaming: hasStreaming,
+		has_vercel_ai_sdk: hasVercelAiSdk,
+		has_edge_runtime: hasEdgeRuntime,
+		has_assistants_api: hasAssistantsApi,
+		has_langgraph: hasLanggraph,
+		message_queue_deps: messageQueueDeps,
+		has_frontend_deps: hasFrontendDeps,
+		recommendations,
+		recommended_tier: recommendedTier
+	};
+}
+
+//#endregion
+export { scanProject };
+//# sourceMappingURL=scan-project.js.map