@sensu-ai/sdk 0.1.5 → 0.5.0

package/dist/client.d.ts

@@ -0,0 +1,128 @@
+import type { TelemetryEvent } from '@sensu-ai/shared';
+import type { SensuClientOptions, StartRunOptions, StartStepOptions, TrackLlmOptions, TrackStreamingLlmOptions, TrackToolOptions, RawLlmCallOptions, TrackRetrievalOptions, RawRetrievalOptions, TrackEmbeddingOptions, RawEmbeddingOptions, RecordFeedbackOptions, RecordEvalScoreOptions, HandoffOptions, SpawnRunOptions, TrackGuardrailOptions, RawGuardrailOptions, RecordPromptRenderOptions, DeployPromptVersionOptions, StartSessionOptions, ResumeSessionOptions } from './types.js';
+export declare class StepHandle {
+    private readonly client;
+    readonly stepId: string;
+    readonly runId: string;
+    readonly sessionId: string;
+    readonly agentId: string;
+    readonly orgId: string;
+    readonly traceId: string;
+    readonly spanId: string;
+    private sequence;
+    private stepName?;
+    private ended;
+    constructor(client: SensuClient, opts: {
+        stepId: string;
+        runId: string;
+        sessionId: string;
+        agentId: string;
+        orgId: string;
+        traceId: string;
+        spanId: string;
+        sequence: number;
+        name?: string;
+    });
+    private base;
+    /** Track an LLM call — wraps fn(), measures latency, emits event */
+    trackLlm(opts: TrackLlmOptions): Promise<unknown>;
+    /**
+     * Track a streaming LLM call — consumes the async iterable, measures TTFT,
+     * and emits stream.token.received events as tokens arrive.
+     * Returns the full concatenated text.
+     */
+    trackStreamingLlm(opts: TrackStreamingLlmOptions): Promise<string>;
+    /** Emit a raw LLM call event (when you have the stats already) */
+    recordLlm(opts: RawLlmCallOptions): void;
+    /** Track a tool call — wraps fn(), measures latency */
+    trackTool(opts: TrackToolOptions): Promise<unknown>;
+    /** Track a retrieval call — wraps fn(), measures latency, emits started + completed */
+    trackRetrieval(opts: TrackRetrievalOptions): Promise<unknown>;
+    /** Emit a raw retrieval completed event (when you have the stats already) */
+    recordRetrieval(opts: RawRetrievalOptions): void;
+    /** Track an embedding call — wraps fn(), measures latency */
+    trackEmbedding(opts: TrackEmbeddingOptions): Promise<unknown>;
+    /** Emit a raw embedding event */
+    recordEmbedding(opts: RawEmbeddingOptions): void;
+    /** Track a guardrail check — wraps fn(), measures latency, handles block */
+    trackGuardrail(opts: TrackGuardrailOptions): Promise<'pass' | 'fail' | 'modified'>;
+    /** Emit a raw guardrail result (check or block) */
+    recordGuardrail(opts: RawGuardrailOptions): void;
+    /** Record a prompt template render event */
+    recordPromptRender(opts: RecordPromptRenderOptions): void;
+    end(): Promise<void>;
+}
+export declare class RunHandle {
+    private readonly client;
+    readonly runId: string;
+    readonly sessionId: string;
+    readonly agentId: string;
+    readonly orgId: string;
+    readonly traceId: string;
+    readonly spanId: string;
+    private stepCount;
+    private ended;
+    constructor(client: SensuClient, opts: {
+        runId: string;
+        sessionId: string;
+        agentId: string;
+        orgId: string;
+        traceId: string;
+        spanId: string;
+    });
+    private base;
+    startStep(opts?: StartStepOptions): StepHandle;
+    /** Record user feedback for this run */
+    recordFeedback(opts: RecordFeedbackOptions): void;
+    /** Record an automated eval score for this run */
+    recordEvalScore(opts: RecordEvalScoreOptions): void;
+    /** Emit an agent.handoff event from this run to another agent */
+    handoff(opts: HandoffOptions): void;
+    end(status?: 'completed' | 'failed'): Promise<void>;
+}
+export declare class SensuClient {
+    private readonly apiKey;
+    private readonly baseUrl;
+    private readonly agentId;
+    private readonly orgId;
+    private readonly batchSize;
+    private readonly flushIntervalMs;
+    readonly disabled: boolean;
+    private readonly disableLivePricing;
+    private readonly onLoopDetected?;
+    private readonly loopThreshold;
+    private readonly runToolCallCounts;
+    private readonly pricingCache;
+    private buffer;
+    private flushTimer;
+    constructor(opts?: SensuClientOptions);
+    /** Enqueue an event for batched sending */
+    enqueue(event: TelemetryEvent): void;
+    /** Flush all buffered events to the Sensu API */
+    flush(): Promise<void>;
+    /** Track a tool call for loop detection; fires onLoopDetected when threshold is reached. */
+    notifyToolCall(runId: string, toolName: string): void;
+    /** Remove per-run loop counters when the run ends to avoid memory leaks. */
+    clearRunLoopState(runId: string): void;
+    /** Start a new agent run */
+    startRun(opts?: StartRunOptions): RunHandle;
+    /**
+     * Spawn a child agent run from within a parent run.
+     * Emits `agent.spawned` on the parent and returns a RunHandle for the child.
+     */
+    spawnRun(parentRun: RunHandle, opts: SpawnRunOptions): RunHandle;
+    /** Explicitly start a session (sets channel and end_user_id) */
+    startSession(opts?: StartSessionOptions): string;
+    /** Resume a previous session */
+    resumeSession(opts: ResumeSessionOptions): string;
+    /** Record a prompt version deployment (org-level event, not tied to a run) */
+    deployPromptVersion(opts: DeployPromptVersionOptions): void;
+    /**
+     * Resolve per-1M-token pricing for a model.
+     * Fetches from the Senzu API on first use and caches for the session lifetime.
+     * Falls back to the bundled MODEL_PRICING table if the API is unreachable or the
+     * model is unknown, and to a near-zero sentinel if it's missing from both.
+     */
+    resolvePricing(provider: string, model: string): Promise<[number, number]>;
+    destroy(): void;
+}

package/dist/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,KAAK,EACV,kBAAkB,EAClB,eAAe,EACf,gBAAgB,EAChB,eAAe,EACf,wBAAwB,EACxB,gBAAgB,EAChB,iBAAiB,EAEjB,qBAAqB,EACrB,mBAAmB,EACnB,qBAAqB,EACrB,mBAAmB,EACnB,qBAAqB,EACrB,sBAAsB,EACtB,cAAc,EACd,eAAe,EACf,qBAAqB,EACrB,mBAAmB,EACnB,yBAAyB,EACzB,0BAA0B,EAC1B,mBAAmB,EACnB,oBAAoB,EAErB,MAAM,YAAY,CAAC;AAMpB,qBAAa,UAAU;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAc;IACrC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,QAAQ,CAAC,CAAS;IAC1B,OAAO,CAAC,KAAK,CAAS;gBAGpB,MAAM,EAAE,WAAW,EACnB,IAAI,EAAE;QACJ,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;QACd,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,CAAC,EAAE,MAAM,CAAC;KACf;IAcH,OAAO,CAAC,IAAI;IAeZ,oEAAoE;IAC9D,QAAQ,CAAC,IAAI,EAAE,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC;IAmDvD;;;;OAIG;IACG,iBAAiB,CAAC,IAAI,EAAE,wBAAwB,GAAG,OAAO,CAAC,MAAM,CAAC;IAiExE,kEAAkE;IAClE,SAAS,CAAC,IAAI,EAAE,iBAAiB,GAAG,IAAI;IAUxC,uDAAuD;IACjD,SAAS,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC;IA6BzD,uFAAuF;IACjF,cAAc,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,OAAO,CAAC;IAuCnE,6EAA6E;IAC7E,eAAe,CAAC,IAAI,EAAE,mBAAmB,GAAG,IAAI;IAgBhD,6DAA6D;IACvD,cAAc,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,OAAO,CAAC;IA0BnE,iCAAiC;IACjC,eAAe,CAAC,IAAI,EAAE,mBAAmB,GAAG,IAAI;IAahD,4EAA4E;IACtE,cAAc,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,MAAM,GAAG,MAAM,GAAG,UAAU,CAAC;IAoCxF,mDAAmD;IACnD,eAAe,CAAC,IAAI,EAAE,mBAAmB,GAAG,IAAI;IAwBhD,4CAA4C;IAC5C,kBAAkB,CAAC,IAAI,EAAE,yBAAyB,GAAG,IAAI;IAYnD,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAS3B;AAMD,qBAAa,SAAS;IACpB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAc;IACrC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,KAAK,CAAS;gBAGpB,MAAM,EAAE,WAAW,EACnB,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM,CAAC;QACd,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,EAAE,MAAM,CAAC;KAChB;IAWH,OAAO,CAAC,IAAI;IAcZ,SAAS,CAAC,IAAI,GAAE,gBAAqB,GAAG,UAAU;IA0BlD,wCAAwC;IACxC,cAAc,CAAC,IAAI,EAAE,qBAAqB,GAAG,IAAI;IAWjD,kDAAkD;IAClD,eAAe,CAAC,IAAI,EAAE,sBAAsB,GAAG,IAAI;IAanD,iEAAiE;IACjE,OAAO,CAAC,IAAI,EAAE,cAAc,GAAG,IAAI;IAU7B,GAAG,CAAC,MAAM,GAAE,WAAW,GAAG,QAAsB,GAAG,OAAO,CAAC,IAAI,CAAC;CAQvE;AAMD,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IAE3B,OAAO,CAAC,MAAM,CAAwB;IACtC,OAAO,CAAC,UAAU,CAA+C;gBAErD,IAAI,GAAE,kBAAuB;IA8BzC,2CAA2C;IAC3C,OAAO,CAAC,KAAK,EAAE,cAAc,GAAG,IAAI;IAQpC,iDAAiD;IAC3C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAuB5B,4BAA4B;IAC5B,QAAQ,CAAC,IAAI,GAAE,eAAoB,GAAG,SAAS;IA8B/C;;;OAGG;IACH,QAAQ,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE,eAAe,GAAG,SAAS;IAgDhE,gEAAgE;IAChE,YAAY,CAAC,IAAI,GAAE,mBAAwB,GAAG,MAAM;IAsBpD,gCAAgC;IAChC,aAAa,CAAC,IAAI,EAAE,oBAAoB,GAAG,MAAM;IAuBjD,8EAA8E;IAC9E,mBAAmB,CAAC,IAAI,EAAE,0BAA0B,GAAG,IAAI;IAkB3D,OAAO,IAAI,IAAI;CAGhB"}