@townco/agent 0.1.51 → 0.1.52

package/dist/telemetry/setup.js

@@ -1,11 +1,14 @@
 /**
  * OpenTelemetry provider setup for @townco/agent
- * Initializes the trace provider, exporter, and propagator
+ * Initializes the trace provider, log provider, exporters, and propagator
  */
 import { propagation } from "@opentelemetry/api";
+import { logs } from "@opentelemetry/api-logs";
 import { W3CTraceContextPropagator } from "@opentelemetry/core";
+import { OTLPLogExporter } from "@opentelemetry/exporter-logs-otlp-http";
 import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
 import { Resource } from "@opentelemetry/resources";
+import { BatchLogRecordProcessor, LoggerProvider, } from "@opentelemetry/sdk-logs";
 import { BatchSpanProcessor } from "@opentelemetry/sdk-trace-base";
 import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
 import { ATTR_SERVICE_NAME } from "@opentelemetry/semantic-conventions";
@@ -83,13 +86,38 @@ export function initializeOpenTelemetry(options = {}) {
     const batchProcessor = new BatchSpanProcessor(loggingExporter, {
         maxQueueSize: 100,
         maxExportBatchSize: 10,
-        scheduledDelayMillis: 5000, // Export every 5 seconds (default)
+        scheduledDelayMillis: 2000, // Export every 2 seconds
     });
     provider.addSpanProcessor(batchProcessor);
     // Register the provider globally
     provider.register();
     // Configure W3C Trace Context propagator for cross-process traces
     propagation.setGlobalPropagator(new W3CTraceContextPropagator());
+    // Set up LoggerProvider for OTLP log export
+    const logUrl = otlpEndpoint.endsWith("/")
+        ? `${otlpEndpoint}v1/logs`
+        : `${otlpEndpoint}/v1/logs`;
+    if (debug) {
+        console.log(`OTLP log URL: ${logUrl}`);
+    }
+    const logExporter = new OTLPLogExporter({
+        url: logUrl,
+    });
+    const loggerProvider = new LoggerProvider({
+        resource: new Resource({
+            [ATTR_SERVICE_NAME]: serviceName,
+        }),
+    });
+    loggerProvider.addLogRecordProcessor(new BatchLogRecordProcessor(logExporter, {
+        maxQueueSize: 100,
+        maxExportBatchSize: 10,
+        scheduledDelayMillis: 2000,
+    }));
+    // Register the logger provider globally
+    logs.setGlobalLoggerProvider(loggerProvider);
+    if (debug) {
+        console.log("✓ Log exporter configured");
+    }
     // Now configure our telemetry wrapper
     configureTelemetry({
         enabled: true,
@@ -106,6 +134,8 @@ export function initializeOpenTelemetry(options = {}) {
         try {
             await provider.forceFlush();
             await provider.shutdown();
+            await loggerProvider.forceFlush();
+            await loggerProvider.shutdown();
             if (debug) {
                 console.log("✓ Telemetry flushed");
             }
@@ -114,7 +144,7 @@ export function initializeOpenTelemetry(options = {}) {
             console.error("Error flushing telemetry:", error);
         }
     };
-    return { provider, shutdown };
+    return { provider, loggerProvider, shutdown };
 }
 /**
  * Initialize OpenTelemetry from environment variables and register shutdown handlers

package/dist/test-telemetry.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Simple test script to verify OpenTelemetry is working
+ * Run with: ENABLE_TELEMETRY=true DEBUG_TELEMETRY=true bun test-telemetry.ts
+ */
+export {};

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);