@townco/agent 0.1.50 → 0.1.52

package/dist/test-telemetry.js

@@ -0,0 +1,88 @@
+/**
+ * Simple test script to verify OpenTelemetry is working
+ * Run with: ENABLE_TELEMETRY=true DEBUG_TELEMETRY=true bun test-telemetry.ts
+ */
+import { makeRunnerFromDefinition } from "./runner/index.js";
+import { configureTelemetry } from "./telemetry/index.js";
+// Initialize OpenTelemetry
+console.log("🔧 Initializing OpenTelemetry...");
+const { NodeTracerProvider } = await import("@opentelemetry/sdk-trace-node");
+const { BatchSpanProcessor } = await import("@opentelemetry/sdk-trace-base");
+const { OTLPTraceExporter } = await import("@opentelemetry/exporter-trace-otlp-grpc");
+const { Resource } = await import("@opentelemetry/resources");
+const { ATTR_SERVICE_NAME } = await import("@opentelemetry/semantic-conventions");
+const serviceName = "test-agent";
+const otlpEndpoint = process.env.OTEL_EXPORTER_OTLP_ENDPOINT ?? "http://localhost:4317";
+console.log(`📡 Service: ${serviceName}`);
+console.log(`📡 Endpoint: ${otlpEndpoint}`);
+const provider = new NodeTracerProvider({
+    resource: new Resource({
+        [ATTR_SERVICE_NAME]: serviceName,
+    }),
+});
+const exporter = new OTLPTraceExporter({
+    url: otlpEndpoint,
+});
+provider.addSpanProcessor(new BatchSpanProcessor(exporter));
+provider.register();
+configureTelemetry({
+    enabled: true,
+    serviceName,
+    attributes: {
+        "test.run": "true",
+    },
+});
+console.log("✅ OpenTelemetry initialized\n");
+// Create a simple test agent
+console.log("🤖 Creating test agent...");
+const runner = makeRunnerFromDefinition({
+    model: "claude-sonnet-4-5-20250929",
+    systemPrompt: "You are a helpful assistant. Keep responses brief.",
+    tools: ["get_weather"],
+});
+console.log("✅ Agent created\n");
+// Run a simple test invocation
+console.log("🚀 Running test invocation...");
+const result = runner.invoke({
+    prompt: [
+        {
+            type: "text",
+            text: "What's the weather like in San Francisco?",
+        },
+    ],
+    sessionId: "test-session-123",
+    messageId: "test-msg-1",
+});
+console.log("📨 Processing response...\n");
+let tokenCount = 0;
+let toolCalls = 0;
+for await (const chunk of result) {
+    if (chunk.sessionUpdate === "tool_call") {
+        console.log(`🔧 Tool called: ${chunk.title}`);
+        toolCalls++;
+    }
+    else if (chunk.sessionUpdate === "agent_message_chunk") {
+        if (chunk.content.type === "text") {
+            process.stdout.write(chunk.content.text);
+        }
+        if (chunk._meta &&
+            "tokenUsage" in chunk._meta &&
+            chunk._meta.tokenUsage &&
+            typeof chunk._meta.tokenUsage === "object" &&
+            "outputTokens" in chunk._meta.tokenUsage) {
+            tokenCount += chunk._meta.tokenUsage.outputTokens ?? 0;
+        }
+    }
+}
+console.log("\n\n✅ Test completed!");
+console.log(`📊 Stats: ${toolCalls} tool calls, ~${tokenCount} output tokens`);
+// Force flush before exit
+console.log("\n⏳ Flushing telemetry data...");
+await provider.forceFlush();
+console.log("✅ Telemetry flushed!");
+console.log("\n🎯 Check Jaeger UI at: http://localhost:16686");
+console.log(`   Service: ${serviceName}`);
+console.log(`   Look for "agent.invoke" spans`);
+// Give a moment for final flush
+await new Promise((resolve) => setTimeout(resolve, 1000));
+process.exit(0);