@cloudbase/agent-observability 0.0.16 → 0.0.18

package/dist/server.js

@@ -1348,15 +1348,133 @@ var init_esm4 = __esm({
 var server_exports = {};
 __export(server_exports, {
   ExporterType: () => ExporterType,
+  SingleLineConsoleSpanExporter: () => SingleLineConsoleSpanExporter,
+  isSingleLineConsoleExporterEnabled: () => isSingleLineConsoleExporterEnabled,
   setupObservability: () => setupObservability
 });
 module.exports = __toCommonJS(server_exports);
 
+// src/core/constants.ts
+var import_openinference_semantic_conventions = require("@arizeai/openinference-semantic-conventions");
+var OBSERVABILITY_TRACER_NAME = "agui-tracer";
+var OtelSpanAttributes = {
+  // OpenInference - re-export all standard conventions
+  ...import_openinference_semantic_conventions.SemanticConventions,
+  // Trace attributes (non-standard)
+  TRACE_NAME: "trace.name",
+  TRACE_TAGS: "trace.tags",
+  TRACE_PUBLIC: "trace.public",
+  TRACE_METADATA: "trace.metadata",
+  TRACE_INPUT: "trace.input",
+  TRACE_OUTPUT: "trace.output",
+  // Observation attributes (non-standard)
+  OBSERVATION_TYPE: "observation.type",
+  OBSERVATION_LEVEL: "observation.level",
+  OBSERVATION_STATUS_MESSAGE: "observation.status_message",
+  OBSERVATION_INPUT: "observation.input",
+  OBSERVATION_OUTPUT: "observation.output",
+  OBSERVATION_METADATA: "observation.metadata",
+  // LLM-specific (non-standard)
+  LLM_COMPLETION_START_TIME: "llm.completion_start_time",
+  LLM_MODEL_PARAMETERS: "llm.model_parameters",
+  LLM_USAGE_DETAILS: "llm.usage_details",
+  LLM_COST_DETAILS: "llm.cost_details",
+  // Retriever-specific (non-standard)
+  RETRIEVER_NAME: "retriever.name",
+  RETRIEVER_QUERY: "retriever.query",
+  RETRIEVER_INDEX_ID: "retriever.index_id",
+  RETRIEVER_TOP_K: "retriever.top_k",
+  // General (non-standard)
+  ENVIRONMENT: "environment",
+  RELEASE: "release",
+  VERSION: "version"
+};
+
+// src/server/SingleLineConsoleSpanExporter.ts
+var SingleLineConsoleSpanExporter = class {
+  /**
+   * Export spans as single-line JSON.
+   */
+  export(spans, resultCallback) {
+    try {
+      for (const span of spans) {
+        const spanDict = this.spanToDict(span);
+        const jsonLine = JSON.stringify(spanDict);
+        console.log(jsonLine);
+      }
+      resultCallback({ code: 0 /* SUCCESS */ });
+    } catch (error) {
+      resultCallback({ code: 1 /* FAILED */, error });
+    }
+  }
+  /**
+   * Shutdown the exporter.
+   */
+  shutdown() {
+    return Promise.resolve();
+  }
+  /**
+   * Force flush the exporter.
+   */
+  forceFlush() {
+    return Promise.resolve();
+  }
+  /**
+   * Convert a ReadableSpan to a dictionary for JSON serialization.
+   */
+  spanToDict(span) {
+    try {
+      const context = span.spanContext();
+      const parentId = span.parentSpanId || span.parentSpanContext?.spanId;
+      const result = {
+        name: span.name,
+        context: {
+          trace_id: context.traceId,
+          span_id: context.spanId,
+          trace_flags: context.traceFlags
+        },
+        kind: span.kind,
+        parent_id: parentId,
+        start_time: span.startTime,
+        end_time: span.endTime,
+        status: {
+          status_code: span.status.code,
+          description: span.status.message
+        },
+        attributes: span.attributes,
+        events: span.events.map((event) => ({
+          name: event.name,
+          timestamp: event.time,
+          attributes: event.attributes
+        })),
+        links: span.links.map((link) => ({
+          context: {
+            trace_id: link.context.traceId,
+            span_id: link.context.spanId
+          },
+          attributes: link.attributes
+        })),
+        resource: {
+          attributes: span.resource.attributes
+        }
+      };
+      result["_log_from"] = OBSERVABILITY_TRACER_NAME;
+      return result;
+    } catch {
+      return {};
+    }
+  }
+};
+function isSingleLineConsoleExporterEnabled() {
+  const value = process.env.CONSOLE_EXPORTER_SINGLE_LINE?.toLowerCase() || "true";
+  return ["true", "1", "yes", "on"].includes(value);
+}
+
 // src/server/setup.ts
 var TRUTHY_ENV_VALUES = /* @__PURE__ */ new Set(["true", "1", "yes", "on"]);
 var DEFAULT_BATCH_CONFIG = {
   maxExportBatchSize: 100,
-  scheduledDelayMillis: 5e3,
+  scheduledDelayMillis: 1e3,
   maxQueueSize: 2048,
   exportTimeoutMillis: 3e4
 };
@@ -1404,33 +1522,29 @@ async function setupConsoleExporter(config) {
   const { trace } = await import("@opentelemetry/api");
   const { resourceFromAttributes } = await import("@opentelemetry/resources");
   const { NodeTracerProvider } = await import("@opentelemetry/sdk-trace-node");
-  const { ConsoleSpanExporter, BatchSpanProcessor } = await import("@opentelemetry/sdk-trace-base");
-  const batchConfig = resolveBatchConfig(config.batch);
+  const { ConsoleSpanExporter, SimpleSpanProcessor } = await import("@opentelemetry/sdk-trace-base");
+  const useSingleLine = isSingleLineConsoleExporterEnabled();
+  const exporter = useSingleLine ? new SingleLineConsoleSpanExporter() : new ConsoleSpanExporter();
+  const exporterType = useSingleLine ? "single-line" : "multi-line";
   let provider = trace.getTracerProvider();
   const isRealProvider = "addSpanProcessor" in provider;
+  const processor = new SimpleSpanProcessor(exporter);
   if (isRealProvider) {
-    const exporter = new ConsoleSpanExporter();
-    const processor = new BatchSpanProcessor(exporter, batchConfig);
     provider.addSpanProcessor(processor);
-    console.info(
-      `[Observability] Console exporter configured (batch=${batchConfig.maxExportBatchSize}, delay=${batchConfig.scheduledDelayMillis}ms)`
-    );
   } else {
     const resource = resourceFromAttributes({
       "service.name": process.env.OTEL_SERVICE_NAME || "ag-ui-server",
       "service.version": "1.0.0"
     });
-    const exporter = new ConsoleSpanExporter();
-    const processor = new BatchSpanProcessor(exporter, batchConfig);
     const tracerProvider = new NodeTracerProvider({
       resource,
       spanProcessors: [processor]
     });
     tracerProvider.register();
-    console.info(
-      `[Observability] Console exporter configured (batch=${batchConfig.maxExportBatchSize}, delay=${batchConfig.scheduledDelayMillis}ms)`
-    );
   }
+  console.info(
+    `[Observability] Console exporter configured (${exporterType}, simple processor)`
+  );
 }
 async function setupOTLPExporter(config) {
   const { trace } = await import("@opentelemetry/api");
@@ -1525,6 +1639,8 @@ var ExporterType = {
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ExporterType,
+  SingleLineConsoleSpanExporter,
+  isSingleLineConsoleExporterEnabled,
   setupObservability
 });
 //# sourceMappingURL=server.js.map