@cloudbase/agent-observability 1.0.1-alpha.19 → 1.0.1-alpha.23

package/dist/langchain.d.mts

@@ -9,9 +9,9 @@ import { SpanContext } from '@opentelemetry/api';
 import { Logger } from '@cloudbase/agent-shared';
 
 /**
- * LangChain Callback Handler for AG-Kit Observability
+ * LangChain Callback Handler for Observability
  *
- * Converts LangChain callback events into AG-Kit observations with OpenInference semantics.
+ * Converts LangChain callback events into observations with OpenInference semantics.
  */
 
 /**
@@ -49,10 +49,10 @@ type AnonymousLlmMessage = {
     additional_kwargs?: BaseMessageFields["additional_kwargs"];
 };
 /**
- * LangChain Callback Handler for AG-Kit Observability.
+ * LangChain Callback Handler for Observability.
  *
  * This handler intercepts LangChain callbacks and converts them into
- * AG-Kit observations following OpenInference semantic conventions.
+ * Observations following OpenInference semantic conventions.
  *
  * @public
  */

package/dist/langchain.d.ts

@@ -9,9 +9,9 @@ import { SpanContext } from '@opentelemetry/api';
 import { Logger } from '@cloudbase/agent-shared';
 
 /**
- * LangChain Callback Handler for AG-Kit Observability
+ * LangChain Callback Handler for Observability
  *
- * Converts LangChain callback events into AG-Kit observations with OpenInference semantics.
+ * Converts LangChain callback events into observations with OpenInference semantics.
  */
 
 /**
@@ -49,10 +49,10 @@ type AnonymousLlmMessage = {
     additional_kwargs?: BaseMessageFields["additional_kwargs"];
 };
 /**
- * LangChain Callback Handler for AG-Kit Observability.
+ * LangChain Callback Handler for Observability.
  *
  * This handler intercepts LangChain callbacks and converts them into
- * AG-Kit observations following OpenInference semantic conventions.
+ * Observations following OpenInference semantic conventions.
  *
  * @public
  */

package/dist/langchain.js

@@ -21,39 +21,42 @@ var __copyProps = (to, from, except, desc) => {
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/core/constants.ts
-var import_openinference_semantic_conventions, OtelSpanAttributes;
+var import_openinference_semantic_conventions, OBSERVABILITY_TRACER_NAME, OBSERVABILITY_SDK_NAME, OBSERVABILITY_SDK_VERSION, OtelSpanAttributes;
 var init_constants = __esm({
   "src/core/constants.ts"() {
     "use strict";
     import_openinference_semantic_conventions = require("@arizeai/openinference-semantic-conventions");
+    OBSERVABILITY_TRACER_NAME = "agui-tracer";
+    OBSERVABILITY_SDK_NAME = "observability";
+    OBSERVABILITY_SDK_VERSION = "0.1.0";
     OtelSpanAttributes = {
       // OpenInference - re-export all standard conventions
       ...import_openinference_semantic_conventions.SemanticConventions,
-      // AG-Kit Trace attributes (non-standard)
+      // 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",
-      // AG-Kit Observation attributes (non-standard)
+      // 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",
-      // AG-Kit LLM-specific (non-standard)
+      // 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",
-      // AG-Kit Retriever-specific (non-standard)
+      // Retriever-specific (non-standard)
       RETRIEVER_NAME: "retriever.name",
       RETRIEVER_QUERY: "retriever.query",
       RETRIEVER_INDEX_ID: "retriever.index_id",
       RETRIEVER_TOP_K: "retriever.top_k",
-      // AG-Kit General (non-standard)
+      // General (non-standard)
       ENVIRONMENT: "environment",
       RELEASE: "release",
       VERSION: "version"
@@ -106,19 +109,20 @@ function createObservationAttributes(type, attributes) {
     modelParameters,
     usageDetails
   } = attributes;
+  const spanKind = import_openinference_semantic_conventions.OpenInferenceSpanKind[type.toUpperCase()] || "UNKNOWN";
   const otelAttributes = {
-    [import_openinference_semantic_conventions2.SemanticConventions.OPENINFERENCE_SPAN_KIND]: type.toUpperCase(),
+    [import_openinference_semantic_conventions2.SemanticConventions.OPENINFERENCE_SPAN_KIND]: spanKind,
     [OtelSpanAttributes.OBSERVATION_TYPE]: type,
     [OtelSpanAttributes.OBSERVATION_LEVEL]: level,
     [OtelSpanAttributes.OBSERVATION_STATUS_MESSAGE]: statusMessage,
     [OtelSpanAttributes.VERSION]: version,
     // Use OpenInference input.value convention
     [import_openinference_semantic_conventions2.SemanticConventions.INPUT_VALUE]: _serialize(input),
-    // Also set legacy agkit.observation.input for compatibility
+    // Also set observation.input for compatibility
     [OtelSpanAttributes.OBSERVATION_INPUT]: _serialize(input),
     // Use OpenInference output.value convention
     [import_openinference_semantic_conventions2.SemanticConventions.OUTPUT_VALUE]: _serialize(output),
-    // Also set legacy agkit.observation.output for compatibility
+    // Also set observation.output for compatibility
     [OtelSpanAttributes.OBSERVATION_OUTPUT]: _serialize(output)
   };
   if (type === "llm") {
@@ -411,22 +415,22 @@ function getTracer() {
     OBSERVABILITY_SDK_VERSION
   );
 }
-var import_api, OBSERVABILITY_GLOBAL_SYMBOL, OBSERVABILITY_SDK_NAME, OBSERVABILITY_SDK_VERSION;
+var import_api, OBSERVABILITY_GLOBAL_SYMBOL;
 var init_tracerProvider = __esm({
   "src/core/tracerProvider.ts"() {
     "use strict";
     import_api = require("@opentelemetry/api");
+    init_constants();
     OBSERVABILITY_GLOBAL_SYMBOL = /* @__PURE__ */ Symbol.for("observability");
-    OBSERVABILITY_SDK_NAME = "ag-kit-observability";
-    OBSERVABILITY_SDK_VERSION = "0.1.0";
   }
 });
 
 // src/core/spanWrapper.ts
-var BaseObservation, ObservationSpan, ObservationLLM, ObservationEmbedding, ObservationAgent, ObservationTool, ObservationChain, ObservationRetriever, ObservationReranker, ObservationEvaluator, ObservationGuardrail;
+var import_api2, BaseObservation, ObservationSpan, ObservationLLM, ObservationEmbedding, ObservationAgent, ObservationTool, ObservationChain, ObservationRetriever, ObservationReranker, ObservationEvaluator, ObservationGuardrail;
 var init_spanWrapper = __esm({
   "src/core/spanWrapper.ts"() {
     "use strict";
+    import_api2 = require("@opentelemetry/api");
     init_attributes();
     init_tracerProvider();
     BaseObservation = class {
@@ -449,7 +453,7 @@ var init_spanWrapper = __esm({
           );
         }
       }
-      /** Gets the AG-Kit OpenTelemetry tracer instance */
+      /** Gets the OpenTelemetry tracer instance */
       get tracer() {
         return getTracer();
       }
@@ -461,6 +465,27 @@ var init_spanWrapper = __esm({
       end(endTime) {
         this.otelSpan.end(endTime);
       }
+      /**
+       * Sets the span status.
+       *
+       * @param status - The status to set on the span
+       */
+      setStatus(status) {
+        this.otelSpan.setStatus(status);
+      }
+      /**
+       * Sets the span status to ERROR.
+       *
+       * Convenience method for marking the span as failed.
+       *
+       * @param message - Error description message
+       */
+      setErrorStatus(message) {
+        this.otelSpan.setStatus({
+          code: import_api2.SpanStatusCode.ERROR,
+          message
+        });
+      }
       /**
        * Updates the OTEL span attributes.
        *
@@ -587,6 +612,9 @@ var init_spanWrapper = __esm({
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
+  OBSERVABILITY_SDK_NAME: () => OBSERVABILITY_SDK_NAME,
+  OBSERVABILITY_SDK_VERSION: () => OBSERVABILITY_SDK_VERSION,
+  OBSERVABILITY_TRACER_NAME: () => OBSERVABILITY_TRACER_NAME,
   ObservationAgent: () => ObservationAgent,
   ObservationChain: () => ObservationChain,
   ObservationEmbedding: () => ObservationEmbedding,
@@ -619,7 +647,7 @@ function createOtelSpan(params) {
 }
 function createParentContext(parentSpanContext) {
   if (!parentSpanContext) return;
-  return import_api2.trace.setSpanContext(import_api2.context.active(), parentSpanContext);
+  return import_api3.trace.setSpanContext(import_api3.context.active(), parentSpanContext);
 }
 function startObservation(name, attributes, options) {
   const { asType = "span", ...observationOptions } = options || {};
@@ -682,7 +710,7 @@ function startObservation(name, attributes, options) {
   }
 }
 function updateActiveTrace(attributes) {
-  const span = import_api2.trace.getActiveSpan();
+  const span = import_api3.trace.getActiveSpan();
   if (!span) {
     console.warn(
       "[Observability] No active OTEL span in context. Skipping trace update."
@@ -692,10 +720,10 @@ function updateActiveTrace(attributes) {
   span.setAttributes(createTraceAttributes(attributes));
 }
 function getActiveTraceId() {
-  return import_api2.trace.getActiveSpan()?.spanContext().traceId;
+  return import_api3.trace.getActiveSpan()?.spanContext().traceId;
 }
 function getActiveSpanId() {
-  return import_api2.trace.getActiveSpan()?.spanContext().spanId;
+  return import_api3.trace.getActiveSpan()?.spanContext().spanId;
 }
 function wrapPromise(promise, span, endOnExit) {
   return promise.then(
@@ -707,7 +735,7 @@ function wrapPromise(promise, span, endOnExit) {
     },
     (err) => {
       span.setStatus({
-        code: import_api2.SpanStatusCode.ERROR,
+        code: import_api3.SpanStatusCode.ERROR,
         message: err instanceof Error ? err.message : "Unknown error"
       });
       if (endOnExit !== false) {
@@ -722,7 +750,7 @@ function startActiveObservation(name, fn, options) {
   return getTracer().startActiveSpan(
     name,
     { startTime: observationOptions?.startTime },
-    createParentContext(observationOptions?.parentSpanContext) ?? import_api2.context.active(),
+    createParentContext(observationOptions?.parentSpanContext) ?? import_api3.context.active(),
     (span) => {
       try {
         let observation;
@@ -773,7 +801,7 @@ function startActiveObservation(name, fn, options) {
         }
       } catch (err) {
         span.setStatus({
-          code: import_api2.SpanStatusCode.ERROR,
+          code: import_api3.SpanStatusCode.ERROR,
           message: err instanceof Error ? err.message : "Unknown error"
         });
         if (endOnExit !== false) {
@@ -785,7 +813,7 @@ function startActiveObservation(name, fn, options) {
   );
 }
 function updateActiveObservation(attributes) {
-  const span = import_api2.trace.getActiveSpan();
+  const span = import_api3.trace.getActiveSpan();
   if (!span) {
     console.warn(
       "[Observability] No active OTEL span in context. Skipping observation update."
@@ -817,8 +845,8 @@ function observe(fn, options = {}) {
         asType
       }
     );
-    const activeContext = import_api2.trace.setSpan(import_api2.context.active(), observation.otelSpan);
-    const result = import_api2.context.with(activeContext, () => fn.apply(this, args));
+    const activeContext = import_api3.trace.setSpan(import_api3.context.active(), observation.otelSpan);
+    const result = import_api3.context.with(activeContext, () => fn.apply(this, args));
     if (result instanceof Promise) {
       return result.then(
         (value) => {
@@ -848,16 +876,17 @@ function observe(fn, options = {}) {
   Object.defineProperty(wrappedFunction, "length", { value: fn.length });
   return wrappedFunction;
 }
-var import_api2;
+var import_api3;
 var init_src = __esm({
   "src/index.ts"() {
     "use strict";
-    import_api2 = require("@opentelemetry/api");
+    import_api3 = require("@opentelemetry/api");
     init_attributes();
     init_spanWrapper();
     init_tracerProvider();
     init_attributes();
     init_tracerProvider();
+    init_constants();
   }
 });
 
@@ -1317,11 +1346,7 @@ var CallbackHandler = class extends import_base.BaseCallbackHandler {
     const statusMessage = attributes.statusMessage;
     if (level === "ERROR") {
       try {
-        const { SpanStatusCode: SpanStatusCode2 } = require("@opentelemetry/api");
-        observation.otelSpan.setStatus({
-          code: SpanStatusCode2.ERROR,
-          message: statusMessage || "Unknown error"
-        });
+        observation.setErrorStatus(statusMessage || "Unknown error");
       } catch (e) {
         this.logger.debug?.("Failed to set span error status:", e);
       }