@observyze/sdk 0.1.0 → 0.1.3

package/dist/chunk-FQBYUOJB.mjs

@@ -0,0 +1,316 @@
+// src/types.ts
+import { SpanType, TraceStatus } from "@observyze/types";
+
+// src/trace.ts
+import { randomUUID } from "crypto";
+function generateId() {
+  return `${Date.now()}-${randomUUID().substring(0, 8)}`;
+}
+var Span = class {
+  data;
+  startTime;
+  constructor(name, type, parentSpanId) {
+    this.startTime = Date.now();
+    this.data = {
+      span_id: generateId(),
+      parent_span_id: parentSpanId,
+      name,
+      type,
+      start_time: new Date(this.startTime),
+      end_time: new Date(this.startTime),
+      // Will be updated on end()
+      duration_ms: 0,
+      input: null,
+      output: null,
+      metadata: {}
+    };
+  }
+  /**
+   * Set the input data for this span
+   */
+  setInput(input) {
+    this.data.input = input;
+    return this;
+  }
+  /**
+   * Set the output data for this span
+   */
+  setOutput(output) {
+    this.data.output = output;
+    return this;
+  }
+  /**
+   * Record an error that occurred during span execution
+   */
+  setError(error) {
+    this.data.error = {
+      message: error.message,
+      stack: error.stack,
+      code: error.code
+    };
+    return this;
+  }
+  /**
+   * Set metadata for this span
+   */
+  setMetadata(key, value) {
+    this.data.metadata[key] = value;
+    return this;
+  }
+  /**
+   * Set multiple metadata fields at once
+   */
+  setMetadataAll(metadata) {
+    this.data.metadata = { ...this.data.metadata, ...metadata };
+    return this;
+  }
+  /**
+   * Set token usage information
+   */
+  setTokens(tokens) {
+    this.data.tokens = tokens;
+    return this;
+  }
+  /**
+   * End the span and calculate duration
+   */
+  end() {
+    const endTime = Date.now();
+    this.data.end_time = new Date(endTime);
+    this.data.duration_ms = endTime - this.startTime;
+  }
+  /**
+   * Get the span ID
+   */
+  get id() {
+    return this.data.span_id;
+  }
+  /**
+   * Get the span data for serialization
+   */
+  toJSON() {
+    return { ...this.data };
+  }
+};
+var Trace = class {
+  data;
+  startTime;
+  spans = [];
+  ended = false;
+  constructor(name, organizationId, projectId) {
+    this.startTime = Date.now();
+    this.data = {
+      trace_id: generateId(),
+      organization_id: organizationId,
+      project_id: projectId,
+      name,
+      status: TraceStatus.RUNNING,
+      start_time: new Date(this.startTime),
+      end_time: new Date(this.startTime),
+      // Will be updated on end()
+      duration_ms: 0,
+      metadata: {},
+      spans: [],
+      tags: []
+    };
+  }
+  /**
+   * Start a new span within this trace
+   */
+  startSpan(name, type, parentSpanId) {
+    if (this.ended) {
+      throw new Error("Cannot start span on an ended trace");
+    }
+    const span = new Span(name, type, parentSpanId);
+    this.spans.push(span);
+    return span;
+  }
+  /**
+   * Add metadata to the trace
+   */
+  setMetadata(key, value) {
+    this.data.metadata[key] = value;
+    return this;
+  }
+  /**
+   * Set multiple metadata fields at once
+   */
+  setMetadataAll(metadata) {
+    this.data.metadata = { ...this.data.metadata, ...metadata };
+    return this;
+  }
+  /**
+   * Add tags to the trace
+   */
+  addTag(tag) {
+    if (!this.data.tags.includes(tag)) {
+      this.data.tags.push(tag);
+    }
+    return this;
+  }
+  /**
+   * Add multiple tags at once
+   */
+  addTags(tags) {
+    tags.forEach((tag) => this.addTag(tag));
+    return this;
+  }
+  /**
+   * Set the user ID associated with this trace
+   */
+  setUserId(userId) {
+    this.data.user_id = userId;
+    return this;
+  }
+  /**
+   * Set the session ID associated with this trace
+   */
+  setSessionId(sessionId) {
+    this.data.session_id = sessionId;
+    return this;
+  }
+  /**
+   * End the trace with a final status
+   */
+  end(status = TraceStatus.SUCCESS) {
+    if (this.ended) {
+      return;
+    }
+    const endTime = Date.now();
+    this.data.end_time = new Date(endTime);
+    this.data.duration_ms = endTime - this.startTime;
+    this.data.status = status;
+    this.data.spans = this.spans.map((span) => span.toJSON());
+    this.ended = true;
+  }
+  /**
+   * Get the trace ID
+   */
+  get id() {
+    return this.data.trace_id;
+  }
+  /**
+   * Check if the trace has ended
+   */
+  get isEnded() {
+    return this.ended;
+  }
+  /**
+   * Get the trace data for serialization
+   */
+  toJSON() {
+    return { ...this.data };
+  }
+};
+
+// src/opentelemetry/exporter.ts
+var ObservyzeSpanExporter = class {
+  client;
+  config;
+  constructor(client, config) {
+    this.client = client;
+    this.config = {
+      serviceName: config?.serviceName || "unknown-service",
+      projectId: config?.projectId || "",
+      defaultSpanType: config?.defaultSpanType || "llm",
+      headers: config?.headers || {}
+    };
+  }
+  /**
+   * Export spans — called by OTel SDK when spans are ready.
+   * Converts OTel spans to Observyze traces and buffers them.
+   */
+  async export(spans, resultCallback) {
+    if (!spans || spans.length === 0) {
+      resultCallback({ code: 0 });
+      return;
+    }
+    try {
+      const organizationId = this.client.config?.organizationId || "";
+      const projectId = this.config.projectId || this.client.config?.projectId || "";
+      for (const span of spans) {
+        const spanContext = span.spanContext();
+        const traceId = spanContext?.traceId || span.spanId();
+        const trace = new Trace(
+          span.name || "otel-span",
+          organizationId,
+          projectId
+        );
+        trace.setMetadata("source", "opentelemetry");
+        trace.setMetadata("otel.trace_id", traceId);
+        trace.setMetadata("otel.span_id", spanContext?.spanId || "");
+        trace.setMetadata("service.name", this.config.serviceName);
+        trace.setMetadataAll(span.attributes || {});
+        if (span.resource?.attributes) {
+          trace.setMetadataAll(span.resource.attributes);
+        }
+        const input = span.attributes?.["gen_ai.prompt.0.content"] || span.attributes?.["gen_ai.completion.0.content"] || span.attributes?.["llm.input"] || void 0;
+        const output = span.attributes?.["gen_ai.completion.0.content"] || span.attributes?.["llm.output"] || void 0;
+        const model = span.attributes?.["gen_ai.request.model"] || span.attributes?.["llm.model"] || void 0;
+        const provider = span.attributes?.["gen_ai.request.provider"] || span.attributes?.["llm.provider"] || void 0;
+        const inputTokens = typeof span.attributes?.["gen_ai.usage.input_tokens"] === "number" ? span.attributes["gen_ai.usage.input_tokens"] : void 0;
+        const outputTokens = typeof span.attributes?.["gen_ai.usage.output_tokens"] === "number" ? span.attributes["gen_ai.usage.output_tokens"] : void 0;
+        trace.setMetadata("provider", provider || "unknown");
+        trace.setMetadata("model", model || "unknown");
+        const nsDuration = span.duration;
+        const durationMs = nsDuration ? Math.round(nsDuration / 1e6) : 0;
+        trace.setMetadata("latency_ms", durationMs);
+        trace.setMetadata("otel.duration_ns", nsDuration);
+        const oSpan = trace.startSpan(span.name || "otel-operation", this.config.defaultSpanType);
+        if (input) oSpan.setInput(input);
+        if (output) oSpan.setOutput(output);
+        if (inputTokens || outputTokens) {
+          oSpan.setTokens({
+            input: inputTokens || 0,
+            output: outputTokens || 0,
+            total: (inputTokens || 0) + (outputTokens || 0)
+          });
+        }
+        if (model) oSpan.setMetadata("model", model);
+        if (provider) oSpan.setMetadata("provider", provider);
+        if (span.attributes) oSpan.setMetadataAll(span.attributes);
+        const status = span.status;
+        const statusCode = status?.code;
+        if (statusCode === 2) {
+          oSpan.setError(new Error(status?.message || "OTel span error"));
+          trace.end(TraceStatus.ERROR);
+        } else {
+          trace.end(TraceStatus.SUCCESS);
+        }
+      }
+      resultCallback({ code: 0 });
+    } catch (error) {
+      resultCallback({
+        code: 1,
+        error: error instanceof Error ? error : new Error(String(error))
+      });
+    }
+  }
+  /**
+   * Called when the exporter is shut down.
+   * Flushes any remaining buffered traces via the SDK client.
+   */
+  async shutdown() {
+    try {
+      await this.client.flush();
+    } catch {
+    }
+  }
+  /**
+   * Called by the OTel SDK to force-export buffered spans.
+   */
+  async forceFlush() {
+    try {
+      await this.client.flush();
+    } catch {
+    }
+  }
+};
+
+export {
+  SpanType,
+  TraceStatus,
+  Span,
+  Trace,
+  ObservyzeSpanExporter
+};

package/dist/chunk-YSYKPKKX.mjs

@@ -0,0 +1,357 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
+  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
+}) : x)(function(x) {
+  if (typeof require !== "undefined") return require.apply(this, arguments);
+  throw Error('Dynamic require of "' + x + '" is not supported');
+});
+var __esm = (fn, res) => function __init() {
+  return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
+};
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/types.ts
+import { SpanType, TraceStatus } from "@observyze/types";
+var init_types = __esm({
+  "src/types.ts"() {
+    "use strict";
+  }
+});
+
+// src/opentelemetry/exporter.ts
+init_types();
+
+// src/trace.ts
+init_types();
+import { randomUUID } from "crypto";
+function generateId() {
+  return `${Date.now()}-${randomUUID().substring(0, 8)}`;
+}
+var Span = class {
+  data;
+  startTime;
+  constructor(name, type, parentSpanId) {
+    this.startTime = Date.now();
+    this.data = {
+      span_id: generateId(),
+      parent_span_id: parentSpanId,
+      name,
+      type,
+      start_time: new Date(this.startTime),
+      end_time: new Date(this.startTime),
+      // Will be updated on end()
+      duration_ms: 0,
+      input: null,
+      output: null,
+      metadata: {}
+    };
+  }
+  /**
+   * Set the input data for this span
+   */
+  setInput(input) {
+    this.data.input = input;
+    return this;
+  }
+  /**
+   * Set the output data for this span
+   */
+  setOutput(output) {
+    this.data.output = output;
+    return this;
+  }
+  /**
+   * Record an error that occurred during span execution
+   */
+  setError(error) {
+    this.data.error = {
+      message: error.message,
+      stack: error.stack,
+      code: error.code
+    };
+    return this;
+  }
+  /**
+   * Set metadata for this span
+   */
+  setMetadata(key, value) {
+    this.data.metadata[key] = value;
+    return this;
+  }
+  /**
+   * Set multiple metadata fields at once
+   */
+  setMetadataAll(metadata) {
+    this.data.metadata = { ...this.data.metadata, ...metadata };
+    return this;
+  }
+  /**
+   * Set token usage information
+   */
+  setTokens(tokens) {
+    this.data.tokens = tokens;
+    return this;
+  }
+  /**
+   * End the span and calculate duration
+   */
+  end() {
+    const endTime = Date.now();
+    this.data.end_time = new Date(endTime);
+    this.data.duration_ms = endTime - this.startTime;
+  }
+  /**
+   * Get the span ID
+   */
+  get id() {
+    return this.data.span_id;
+  }
+  /**
+   * Get the span data for serialization
+   */
+  toJSON() {
+    return { ...this.data };
+  }
+};
+var Trace = class {
+  data;
+  startTime;
+  spans = [];
+  ended = false;
+  constructor(name, organizationId, projectId) {
+    this.startTime = Date.now();
+    this.data = {
+      trace_id: generateId(),
+      organization_id: organizationId,
+      project_id: projectId,
+      name,
+      status: TraceStatus.RUNNING,
+      start_time: new Date(this.startTime),
+      end_time: new Date(this.startTime),
+      // Will be updated on end()
+      duration_ms: 0,
+      metadata: {},
+      spans: [],
+      tags: []
+    };
+  }
+  /**
+   * Start a new span within this trace
+   */
+  startSpan(name, type, parentSpanId) {
+    if (this.ended) {
+      throw new Error("Cannot start span on an ended trace");
+    }
+    const span = new Span(name, type, parentSpanId);
+    this.spans.push(span);
+    return span;
+  }
+  /**
+   * Add metadata to the trace
+   */
+  setMetadata(key, value) {
+    this.data.metadata[key] = value;
+    return this;
+  }
+  /**
+   * Set multiple metadata fields at once
+   */
+  setMetadataAll(metadata) {
+    this.data.metadata = { ...this.data.metadata, ...metadata };
+    return this;
+  }
+  /**
+   * Add tags to the trace
+   */
+  addTag(tag) {
+    if (!this.data.tags.includes(tag)) {
+      this.data.tags.push(tag);
+    }
+    return this;
+  }
+  /**
+   * Add multiple tags at once
+   */
+  addTags(tags) {
+    tags.forEach((tag) => this.addTag(tag));
+    return this;
+  }
+  /**
+   * Set the user ID associated with this trace
+   */
+  setUserId(userId) {
+    this.data.user_id = userId;
+    return this;
+  }
+  /**
+   * Set the session ID associated with this trace
+   */
+  setSessionId(sessionId) {
+    this.data.session_id = sessionId;
+    return this;
+  }
+  /**
+   * End the trace with a final status
+   */
+  end(status = TraceStatus.SUCCESS) {
+    if (this.ended) {
+      return;
+    }
+    const endTime = Date.now();
+    this.data.end_time = new Date(endTime);
+    this.data.duration_ms = endTime - this.startTime;
+    this.data.status = status;
+    this.data.spans = this.spans.map((span) => span.toJSON());
+    this.ended = true;
+  }
+  /**
+   * Get the trace ID
+   */
+  get id() {
+    return this.data.trace_id;
+  }
+  /**
+   * Check if the trace has ended
+   */
+  get isEnded() {
+    return this.ended;
+  }
+  /**
+   * Get the trace data for serialization
+   */
+  toJSON() {
+    return { ...this.data };
+  }
+};
+
+// src/opentelemetry/exporter.ts
+var ObservyzeSpanExporter = class {
+  client;
+  config;
+  constructor(client, config) {
+    this.client = client;
+    this.config = {
+      serviceName: config?.serviceName || "unknown-service",
+      projectId: config?.projectId || "",
+      defaultSpanType: config?.defaultSpanType || "llm",
+      headers: config?.headers || {}
+    };
+  }
+  /**
+   * Export spans — called by OTel SDK when spans are ready.
+   * Converts OTel spans to Observyze traces and buffers them.
+   */
+  async export(spans, resultCallback) {
+    if (!spans || spans.length === 0) {
+      resultCallback({ code: 0 });
+      return;
+    }
+    try {
+      const organizationId = this.client.config?.organizationId || "";
+      const projectId = this.config.projectId || this.client.config?.projectId || "";
+      for (const span of spans) {
+        const spanContext = span.spanContext();
+        const traceId = spanContext?.traceId || span.spanId();
+        const trace = new Trace(
+          span.name || "otel-span",
+          organizationId,
+          projectId
+        );
+        trace.setMetadata("source", "opentelemetry");
+        trace.setMetadata("otel.trace_id", traceId);
+        trace.setMetadata("otel.span_id", spanContext?.spanId || "");
+        trace.setMetadata("service.name", this.config.serviceName);
+        trace.setMetadataAll(span.attributes || {});
+        if (span.resource?.attributes) {
+          trace.setMetadataAll(span.resource.attributes);
+        }
+        const input = span.attributes?.["gen_ai.prompt.0.content"] || span.attributes?.["gen_ai.completion.0.content"] || span.attributes?.["llm.input"] || void 0;
+        const output = span.attributes?.["gen_ai.completion.0.content"] || span.attributes?.["llm.output"] || void 0;
+        const model = span.attributes?.["gen_ai.request.model"] || span.attributes?.["llm.model"] || void 0;
+        const provider = span.attributes?.["gen_ai.request.provider"] || span.attributes?.["llm.provider"] || void 0;
+        const inputTokens = typeof span.attributes?.["gen_ai.usage.input_tokens"] === "number" ? span.attributes["gen_ai.usage.input_tokens"] : void 0;
+        const outputTokens = typeof span.attributes?.["gen_ai.usage.output_tokens"] === "number" ? span.attributes["gen_ai.usage.output_tokens"] : void 0;
+        trace.setMetadata("provider", provider || "unknown");
+        trace.setMetadata("model", model || "unknown");
+        const nsDuration = span.duration;
+        const durationMs = nsDuration ? Math.round(nsDuration / 1e6) : 0;
+        trace.setMetadata("latency_ms", durationMs);
+        trace.setMetadata("otel.duration_ns", nsDuration);
+        const oSpan = trace.startSpan(span.name || "otel-operation", this.config.defaultSpanType);
+        if (input) oSpan.setInput(input);
+        if (output) oSpan.setOutput(output);
+        if (inputTokens || outputTokens) {
+          oSpan.setTokens({
+            input: inputTokens || 0,
+            output: outputTokens || 0,
+            total: (inputTokens || 0) + (outputTokens || 0)
+          });
+        }
+        if (model) oSpan.setMetadata("model", model);
+        if (provider) oSpan.setMetadata("provider", provider);
+        if (span.attributes) oSpan.setMetadataAll(span.attributes);
+        const status = span.status;
+        const statusCode = status?.code;
+        if (statusCode === 2) {
+          oSpan.setError(new Error(status?.message || "OTel span error"));
+          trace.end(TraceStatus.ERROR);
+        } else {
+          trace.end(TraceStatus.SUCCESS);
+        }
+      }
+      resultCallback({ code: 0 });
+    } catch (error) {
+      resultCallback({
+        code: 1,
+        error: error instanceof Error ? error : new Error(String(error))
+      });
+    }
+  }
+  /**
+   * Called when the exporter is shut down.
+   * Flushes any remaining buffered traces via the SDK client.
+   */
+  async shutdown() {
+    try {
+      await this.client.flush();
+    } catch {
+    }
+  }
+  /**
+   * Called by the OTel SDK to force-export buffered spans.
+   */
+  async forceFlush() {
+    try {
+      await this.client.flush();
+    } catch {
+    }
+  }
+};
+
+export {
+  __require,
+  __esm,
+  __export,
+  __toCommonJS,
+  SpanType,
+  TraceStatus,
+  init_types,
+  Span,
+  Trace,
+  ObservyzeSpanExporter
+};