cost-katana 2.0.0

package/dist/providers/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":";;;AAAA,iCAAsC;AAGtC,iDAAuC;AA8BvC,MAAa,cAAe,SAAQ,mBAAY;IAI9C,YAAY,MAAsB;QAChC,KAAK,CAAC,MAAM,CAAC,CAAC;QAEd,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,QAAQ,IAAI,2BAA2B,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,MAAc;QAC5C,IAAI,CAAC;YAGH,MAAM,OAAO,GAAG,IAAA,sBAAM,EAAC,IAAI,CAAC,CAAC;YAC7B,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,OAAO,CAAC,IAAI,CAAC,yCAAyC,EAAE,KAAK,CAAC,CAAC;YAC/D,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAwB;QACxC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,cAAc,CAAC;QACzE,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,GAAG,QAAQ,EAAE,CAAC;QAExC,MAAM,IAAI,GAAwB;YAChC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,UAAU,EAAE,OAAO,CAAC,SAAS;YAC7B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,KAAK,EAAE,OAAO,CAAC,IAAI;YACnB,IAAI,EAAE,OAAO,CAAC,aAAa;YAC3B,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,OAAO,CAAC,MAAM;SACrB,CAAC;QAEF,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC/B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACvC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,IAAI,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,KAAK,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;wBACpE,MAAM,MAAM,GAAG,KAA4B,CAAC;wBAC5C,QAAQ,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,QAAQ,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,OAAO,IAAI,EAAE,CAAC;oBAEd,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;gBACjD,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,IAAI,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtD,CAAC;YACD,MAAM,OAAO,GAAG,IAAyB,CAAC;YAE1C,OAAO;gBACL,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;oBACrC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAoB,EAAE,EAAE,CAAC,CAAC;wBAC7C,IAAI,EAAE,MAAM,CAAC,IAAI;wBACjB,OAAO,EACL,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ;4BAClC,MAAM,CAAC,OAAO,KAAK,IAAI;4BACvB,MAAM,IAAI,MAAM,CAAC,OAAO;4BACxB,SAAS,IAAI,MAAM,CAAC,OAAO;4BACzB,CAAC,CAAE,MAAM,CAAC,OAAgD;4BAC1D,CAAC,CAAC,SAAS;wBACf,KAAK,EAAE,MAAM,CAAC,KAAK;wBACnB,YAAY,EAAE,MAAM,CAAC,aAAa;qBACnC,CAAC,CAAC;oBACL,CAAC,CAAC,EAAE;gBACN,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,QAAQ,EAAE;oBACR,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;iBAC/C;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED,UAAU,CAAC,KAAc;QACvB,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAChD,MAAM,CAAC,GAAG,KAA6B,CAAC;YACxC,OAAO;gBACL,aAAa,EAAE,CAAC,CAAC,aAAa,IAAI,CAAC;gBACnC,iBAAiB,EAAE,CAAC,CAAC,iBAAiB,IAAI,CAAC;gBAC3C,YAAY,EAAE,CAAC,CAAC,YAAY,IAAI,CAAC;aAClC,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;IACrE,CAAC;IAES,eAAe,CAAC,KAAe;QACvC,OAAQ,KAAqB,CAAC,aAAa,CAAC;IAC9C,CAAC;IAES,mBAAmB,CAAC,KAAe;QAC3C,OAAQ,KAAqB,CAAC,iBAAiB,CAAC;IAClD,CAAC;IAGD,KAAK,CAAC,eAAe,CAAC,KAAwB,EAAE,QAAgB,wBAAwB;QACtF,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,aAAa,CAAC;QAExC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACvC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK;oBACL,KAAK;iBACN,CAAC;aACH,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,IAAI,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,KAAK,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;wBACpE,MAAM,MAAM,GAAG,KAA4B,CAAC;wBAC5C,QAAQ,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,QAAQ,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,OAAO,IAAI,EAAE,CAAC;oBAEd,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;gBACjD,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;YACnD,CAAC;YAED,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,SAAS,CAAC;QAEpC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACvC;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,IAAI,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,KAAK,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;wBACpE,MAAM,MAAM,GAAG,KAA4B,CAAC;wBAC5C,QAAQ,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,QAAQ,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,OAAO,IAAI,EAAE,CAAC;oBAEd,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;gBACjD,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;YACnD,CAAC;YAED,MAAM,IAAI,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE,CAAC;gBAC3D,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtD,CAAC;YACD,OAAQ,IAAgC,CAAC,IAAiB,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;CACF;AAlND,wCAkNC"}

package/dist/trace/client.d.ts

@@ -0,0 +1,48 @@
+import { StartSpanInput, EndSpanInput, RecordMessageInput, SessionGraph, SessionDetails, TraceService } from './types';
+export interface TraceClientConfig {
+    apiUrl?: string;
+    apiKey: string;
+    projectId: string;
+    headers?: Record<string, string>;
+    timeout?: number;
+}
+export declare class TraceClient implements TraceService {
+    private config;
+    constructor(config: TraceClientConfig);
+    private request;
+    startSpan(input: StartSpanInput): Promise<{
+        traceId: string;
+        sessionId: string;
+    }>;
+    endSpan(traceId: string, input: EndSpanInput): Promise<void>;
+    recordMessage(input: RecordMessageInput): Promise<void>;
+    getSessionGraph(sessionId: string): Promise<SessionGraph>;
+    getSessionDetails(sessionId: string): Promise<SessionDetails>;
+    listSessions(params?: {
+        userId?: string;
+        label?: string;
+        from?: Date;
+        to?: Date;
+        page?: number;
+        limit?: number;
+    }): Promise<{
+        sessions: SessionDetails[];
+        total: number;
+        page: number;
+        pages: number;
+    }>;
+    getSessionsSummary(params?: {
+        from?: Date;
+        to?: Date;
+    }): Promise<{
+        totalSessions: number;
+        activeSessions: number;
+        totalCost: number;
+        avgDuration: number;
+        errorRate: number;
+    }>;
+    ingestBatch(spans: Array<StartSpanInput & EndSpanInput & {
+        traceId: string;
+    }>): Promise<void>;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/trace/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/trace/client.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,cAAc,EACd,YAAY,EACZ,kBAAkB,EAClB,YAAY,EACZ,cAAc,EACd,YAAY,EACb,MAAM,SAAS,CAAC;AAEjB,MAAM,WAAW,iBAAiB;IAIhC,MAAM,CAAC,EAAE,MAAM,CAAC;IAKhB,MAAM,EAAE,MAAM,CAAC;IAKf,SAAS,EAAE,MAAM,CAAC;IAKlB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAKjC,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AA2BD,qBAAa,WAAY,YAAW,YAAY;IAC9C,OAAO,CAAC,MAAM,CAA8B;gBAEhC,MAAM,EAAE,iBAAiB;YAkBvB,OAAO;IAoCf,SAAS,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;IAiBjF,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAS5D,aAAa,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IASvD,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAKzD,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC;IAQ7D,YAAY,CAAC,MAAM,CAAC,EAAE;QAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,IAAI,CAAC,EAAE,IAAI,CAAC;QACZ,EAAE,CAAC,EAAE,IAAI,CAAC;QACV,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC;QACV,QAAQ,EAAE,cAAc,EAAE,CAAC;QAC3B,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;IAmBI,kBAAkB,CAAC,MAAM,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,IAAI,CAAC;QAAC,EAAE,CAAC,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC;QACrE,aAAa,EAAE,MAAM,CAAC;QACtB,cAAc,EAAE,MAAM,CAAC;QACvB,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IAeI,WAAW,CACf,KAAK,EAAE,KAAK,CAAC,cAAc,GAAG,YAAY,GAAG;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAChE,OAAO,CAAC,IAAI,CAAC;CAOjB"}

package/dist/trace/client.js

@@ -0,0 +1,130 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TraceClient = void 0;
+class TraceClient {
+    constructor(config) {
+        this.config = {
+            apiUrl: config.apiUrl || 'https://cost-katana-backend.store',
+            apiKey: config.apiKey,
+            projectId: config.projectId,
+            headers: config.headers || {},
+            timeout: config.timeout || 30000
+        };
+        if (!this.config.apiKey) {
+            throw new Error('API key is required for TraceClient');
+        }
+        if (!this.config.projectId) {
+            throw new Error('Project ID is required for TraceClient');
+        }
+    }
+    async request(method, path, body) {
+        const controller = new AbortController();
+        const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
+        try {
+            const response = await fetch(`${this.config.apiUrl}${path}`, {
+                method,
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Bearer ${this.config.apiKey}`,
+                    'X-Project-ID': this.config.projectId,
+                    ...this.config.headers
+                },
+                body: body ? JSON.stringify(body) : undefined,
+                signal: controller.signal
+            });
+            clearTimeout(timeoutId);
+            if (!response.ok) {
+                const error = await response.json().catch(() => ({ message: response.statusText }));
+                throw new Error(`Trace API error: ${error.message || response.statusText}`);
+            }
+            return response.json();
+        }
+        catch (error) {
+            clearTimeout(timeoutId);
+            if (error.name === 'AbortError') {
+                throw new Error('Trace API request timeout');
+            }
+            throw error;
+        }
+    }
+    async startSpan(input) {
+        try {
+            const response = await this.request('POST', '/v1/traces/start', input);
+            return {
+                traceId: response.traceId,
+                sessionId: response.sessionId
+            };
+        }
+        catch (error) {
+            console.error('Failed to start trace span:', error);
+            return {
+                traceId: `mock-${Date.now()}-${Math.random()}`,
+                sessionId: input.sessionId || `mock-session-${Date.now()}`
+            };
+        }
+    }
+    async endSpan(traceId, input) {
+        try {
+            await this.request('POST', `/v1/traces/${traceId}/end`, input);
+        }
+        catch (error) {
+            console.error('Failed to end trace span:', error);
+        }
+    }
+    async recordMessage(input) {
+        try {
+            await this.request('POST', '/v1/traces/messages', input);
+        }
+        catch (error) {
+            console.error('Failed to record message:', error);
+        }
+    }
+    async getSessionGraph(sessionId) {
+        const response = await this.request('GET', `/v1/sessions/${sessionId}/graph`);
+        return response;
+    }
+    async getSessionDetails(sessionId) {
+        const response = await this.request('GET', `/v1/sessions/${sessionId}/details`);
+        return response;
+    }
+    async listSessions(params) {
+        const queryParams = new URLSearchParams();
+        if (params) {
+            if (params.userId)
+                queryParams.append('userId', params.userId);
+            if (params.label)
+                queryParams.append('label', params.label);
+            if (params.from)
+                queryParams.append('from', params.from.toISOString());
+            if (params.to)
+                queryParams.append('to', params.to.toISOString());
+            if (params.page)
+                queryParams.append('page', params.page.toString());
+            if (params.limit)
+                queryParams.append('limit', params.limit.toString());
+        }
+        const response = await this.request('GET', `/v1/sessions?${queryParams}`);
+        return response;
+    }
+    async getSessionsSummary(params) {
+        const queryParams = new URLSearchParams();
+        if (params) {
+            if (params.from)
+                queryParams.append('from', params.from.toISOString());
+            if (params.to)
+                queryParams.append('to', params.to.toISOString());
+        }
+        const response = await this.request('GET', `/v1/sessions/summary?${queryParams}`);
+        return response;
+    }
+    async ingestBatch(spans) {
+        try {
+            await this.request('POST', '/v1/traces/ingest', { spans });
+        }
+        catch (error) {
+            console.error('Failed to ingest trace batch:', error);
+        }
+    }
+}
+exports.TraceClient = TraceClient;
+//# sourceMappingURL=client.js.map

package/dist/trace/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/trace/client.ts"],"names":[],"mappings":";;;AAiEA,MAAa,WAAW;IAGtB,YAAY,MAAyB;QACnC,IAAI,CAAC,MAAM,GAAG;YACZ,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,mCAAmC;YAC5D,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,KAAK;SACjC,CAAC;QAEF,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,IAAY,EAAE,IAAU;QAC5D,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE5E,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,EAAE,EAAE;gBAC3D,MAAM;gBACN,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;oBAC7C,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;oBACrC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO;iBACvB;gBACD,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;gBAC7C,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,YAAY,CAAC,SAAS,CAAC,CAAC;YAExB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBACpF,MAAM,IAAI,KAAK,CAAC,oBAAoB,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YAC9E,CAAC;YAED,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,YAAY,CAAC,SAAS,CAAC,CAAC;YAExB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,KAAqB;QACnC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAAC;YACvE,OAAO;gBACL,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,SAAS,EAAE,QAAQ,CAAC,SAAS;aAC9B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YAEpD,OAAO;gBACL,OAAO,EAAE,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE;gBAC9C,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,gBAAgB,IAAI,CAAC,GAAG,EAAE,EAAE;aAC3D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAe,EAAE,KAAmB;QAChD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,cAAc,OAAO,MAAM,EAAE,KAAK,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;QAEpD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,KAAyB;QAC3C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,qBAAqB,EAAE,KAAK,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,KAAK,CAAC,CAAC;QAEpD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAiB;QACrC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,SAAS,QAAQ,CAAC,CAAC;QAC9E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QACvC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,SAAS,UAAU,CAAC,CAAC;QAChF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKD,KAAK,CAAC,YAAY,CAAC,MAOlB;QAMC,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC;QAE1C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,MAAM;gBAAE,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YAC/D,IAAI,MAAM,CAAC,KAAK;gBAAE,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5D,IAAI,MAAM,CAAC,IAAI;gBAAE,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YACvE,IAAI,MAAM,CAAC,EAAE;gBAAE,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;YACjE,IAAI,MAAM,CAAC,IAAI;gBAAE,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YACpE,IAAI,MAAM,CAAC,KAAK;gBAAE,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;QACzE,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,gBAAgB,WAAW,EAAE,CAAC,CAAC;QAC1E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKD,KAAK,CAAC,kBAAkB,CAAC,MAAmC;QAO1D,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC;QAE1C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,IAAI;gBAAE,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YACvE,IAAI,MAAM,CAAC,EAAE;gBAAE,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,wBAAwB,WAAW,EAAE,CAAC,CAAC;QAClF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKD,KAAK,CAAC,WAAW,CACf,KAAiE;QAEjE,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,mBAAmB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;CACF;AAtKD,kCAsKC"}

package/dist/trace/index.d.ts

@@ -0,0 +1,7 @@
+export * from './types';
+export { createTraceMiddleware, createChildSpan, type Request, type Response, type NextFunction } from './middleware';
+export { TrackedOpenAI, TrackedAnthropic, createTrackedProvider } from './providers';
+export { LocalTraceService, type LocalTraceServiceConfig } from './service';
+export { TraceClient, type TraceClientConfig } from './client';
+export type { TraceContext, StartSpanInput, EndSpanInput, RecordMessageInput, TraceNode, SessionGraph, SessionDetails, TraceService } from './types';
+//# sourceMappingURL=index.d.ts.map

package/dist/trace/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/trace/index.ts"],"names":[],"mappings":"AA2CA,cAAc,SAAS,CAAC;AAGxB,OAAO,EACL,qBAAqB,EACrB,eAAe,EACf,KAAK,OAAO,EACZ,KAAK,QAAQ,EACb,KAAK,YAAY,EAClB,MAAM,cAAc,CAAC;AAGtB,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAGrF,OAAO,EAAE,iBAAiB,EAAE,KAAK,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,KAAK,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAG/D,YAAY,EACV,YAAY,EACZ,cAAc,EACd,YAAY,EACZ,kBAAkB,EAClB,SAAS,EACT,YAAY,EACZ,cAAc,EACd,YAAY,EACb,MAAM,SAAS,CAAC"}

package/dist/trace/index.js

@@ -0,0 +1,30 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TraceClient = exports.LocalTraceService = exports.createTrackedProvider = exports.TrackedAnthropic = exports.TrackedOpenAI = exports.createChildSpan = exports.createTraceMiddleware = void 0;
+__exportStar(require("./types"), exports);
+var middleware_1 = require("./middleware");
+Object.defineProperty(exports, "createTraceMiddleware", { enumerable: true, get: function () { return middleware_1.createTraceMiddleware; } });
+Object.defineProperty(exports, "createChildSpan", { enumerable: true, get: function () { return middleware_1.createChildSpan; } });
+var providers_1 = require("./providers");
+Object.defineProperty(exports, "TrackedOpenAI", { enumerable: true, get: function () { return providers_1.TrackedOpenAI; } });
+Object.defineProperty(exports, "TrackedAnthropic", { enumerable: true, get: function () { return providers_1.TrackedAnthropic; } });
+Object.defineProperty(exports, "createTrackedProvider", { enumerable: true, get: function () { return providers_1.createTrackedProvider; } });
+var service_1 = require("./service");
+Object.defineProperty(exports, "LocalTraceService", { enumerable: true, get: function () { return service_1.LocalTraceService; } });
+var client_1 = require("./client");
+Object.defineProperty(exports, "TraceClient", { enumerable: true, get: function () { return client_1.TraceClient; } });
+//# sourceMappingURL=index.js.map

package/dist/trace/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/trace/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AA2CA,0CAAwB;AAGxB,2CAMsB;AALpB,mHAAA,qBAAqB,OAAA;AACrB,6GAAA,eAAe,OAAA;AAOjB,yCAAqF;AAA5E,0GAAA,aAAa,OAAA;AAAE,6GAAA,gBAAgB,OAAA;AAAE,kHAAA,qBAAqB,OAAA;AAG/D,qCAA4E;AAAnE,4GAAA,iBAAiB,OAAA;AAC1B,mCAA+D;AAAtD,qGAAA,WAAW,OAAA"}

package/dist/trace/middleware.d.ts

@@ -0,0 +1,70 @@
+import { TraceContext } from './types';
+export interface Request {
+    headers: {
+        [key: string]: string | string[] | undefined;
+    };
+    method: string;
+    path: string;
+    query: any;
+    ip?: string;
+    traceContext?: TraceContext;
+    get(name: string): string | undefined;
+}
+export interface Response {
+    locals: {
+        sessionId?: string;
+        traceId?: string;
+        traceContext?: TraceContext;
+    };
+    statusCode: number;
+    send: (body?: any) => Response;
+    json: (body: any) => Response;
+    end: (chunk?: any) => Response;
+    on(event: string, listener: (chunk?: any) => void): this;
+    getHeaders(): {
+        [key: string]: string | string[] | number | undefined;
+    };
+}
+export interface NextFunction {
+    (err?: Error | string): void;
+}
+interface TraceMiddlewareOptions {
+    sessionHeader?: string;
+    startSpan: (input: {
+        sessionId: string;
+        name: string;
+        type: string;
+        metadata?: Record<string, any>;
+    }) => Promise<{
+        traceId: string;
+    }>;
+    endSpan: (traceId: string, input: {
+        status: 'ok' | 'error';
+        error?: {
+            message: string;
+            stack?: string;
+        };
+        metadata?: Record<string, any>;
+    }) => Promise<void>;
+    includeDetails?: boolean;
+}
+export declare function createTraceMiddleware(options: TraceMiddlewareOptions): (req: Request, res: Response, next: NextFunction) => Promise<void>;
+export declare function createChildSpan(req: Request, input: {
+    name: string;
+    type: string;
+    metadata?: Record<string, any>;
+}, startSpanFn: (input: any) => Promise<{
+    traceId: string;
+}>): Promise<{
+    traceId: string;
+    end: (_endInput: {
+        status: "ok" | "error";
+        error?: {
+            message: string;
+            stack?: string;
+        };
+        metadata?: Record<string, any>;
+    }) => Promise<void>;
+}>;
+export {};
+//# sourceMappingURL=middleware.d.ts.map

package/dist/trace/middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/trace/middleware.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAGvC,MAAM,WAAW,OAAO;IACtB,OAAO,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAA;KAAE,CAAC;IAC1D,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,GAAG,CAAC;IACX,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;CACvC;AAED,MAAM,WAAW,QAAQ;IACvB,MAAM,EAAE;QACN,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,YAAY,CAAC,EAAE,YAAY,CAAC;KAC7B,CAAC;IACF,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,QAAQ,CAAC;IAC/B,IAAI,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,QAAQ,CAAC;IAC9B,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,GAAG,KAAK,QAAQ,CAAC;IAC/B,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,EAAE,GAAG,KAAK,IAAI,GAAG,IAAI,CAAC;IACzD,UAAU,IAAI;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,MAAM,GAAG,SAAS,CAAA;KAAE,CAAC;CACzE;AAED,MAAM,WAAW,YAAY;IAC3B,CAAC,GAAG,CAAC,EAAE,KAAK,GAAG,MAAM,GAAG,IAAI,CAAC;CAC9B;AAED,UAAU,sBAAsB;IAI9B,aAAa,CAAC,EAAE,MAAM,CAAC;IAKvB,SAAS,EAAE,CAAC,KAAK,EAAE;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,KAAK,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAKnC,OAAO,EAAE,CACP,OAAO,EAAE,MAAM,EACf,KAAK,EAAE;QACL,MAAM,EAAE,IAAI,GAAG,OAAO,CAAC;QACvB,KAAK,CAAC,EAAE;YAAE,OAAO,EAAE,MAAM,CAAC;YAAC,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC;QAC5C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,KACE,OAAO,CAAC,IAAI,CAAC,CAAC;IAKnB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAiBD,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,sBAAsB,IAGrD,KAAK,OAAO,EAAE,KAAK,QAAQ,EAAE,MAAM,YAAY,mBA4F9D;AAyBD,wBAAsB,eAAe,CACnC,GAAG,EAAE,OAAO,EACZ,KAAK,EAAE;IACL,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC,EACD,WAAW,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,OAAO,CAAC;IAAE,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC;;qBAiBhC;QACrB,MAAM,EAAE,IAAI,GAAG,OAAO,CAAC;QACvB,KAAK,CAAC,EAAE;YAAE,OAAO,EAAE,MAAM,CAAC;YAAC,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC;QAC5C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC;GAKJ"}

package/dist/trace/middleware.js

@@ -0,0 +1,104 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTraceMiddleware = createTraceMiddleware;
+exports.createChildSpan = createChildSpan;
+const uuid_1 = require("uuid");
+function createTraceMiddleware(options) {
+    const { sessionHeader = 'x-session-id', startSpan, endSpan, includeDetails = false } = options;
+    return async (req, res, next) => {
+        const sessionId = req.headers[sessionHeader] || (0, uuid_1.v4)();
+        const metadata = {
+            method: req.method,
+            path: req.path,
+            query: req.query
+        };
+        if (includeDetails) {
+            metadata.headers = req.headers;
+            metadata.ip = req.ip;
+            metadata.userAgent = req.get('user-agent');
+        }
+        try {
+            const { traceId } = await startSpan({
+                sessionId,
+                name: `${req.method} ${req.path}`,
+                type: 'http',
+                metadata
+            });
+            const traceContext = {
+                sessionId,
+                traceId,
+                parentId: undefined
+            };
+            req.traceContext = traceContext;
+            res.locals.sessionId = sessionId;
+            res.locals.traceId = traceId;
+            res.locals.traceContext = traceContext;
+            const originalSend = res.send;
+            const originalJson = res.json;
+            const originalEnd = res.end;
+            const endTrace = async (error) => {
+                try {
+                    await endSpan(traceId, {
+                        status: error || res.statusCode >= 400 ? 'error' : 'ok',
+                        error: error
+                            ? {
+                                message: error.message,
+                                stack: error.stack
+                            }
+                            : undefined,
+                        metadata: {
+                            statusCode: res.statusCode,
+                            ...(includeDetails
+                                ? {
+                                    responseHeaders: res.getHeaders()
+                                }
+                                : {})
+                        }
+                    });
+                }
+                catch (err) {
+                    console.error('Failed to end trace span:', err);
+                }
+            };
+            res.send = function (data) {
+                endTrace();
+                return originalSend.call(this, data);
+            };
+            res.json = function (data) {
+                endTrace();
+                return originalJson.call(this, data);
+            };
+            res.end = function (chunk) {
+                endTrace();
+                return originalEnd.call(this, chunk);
+            };
+            res.on('error', (error) => {
+                endTrace(error);
+            });
+            next();
+        }
+        catch (error) {
+            console.error('Failed to start trace span:', error);
+            next();
+        }
+    };
+}
+async function createChildSpan(req, input, startSpanFn) {
+    const parentContext = req.traceContext;
+    if (!parentContext) {
+        throw new Error('No trace context found on request');
+    }
+    const { traceId } = await startSpanFn({
+        sessionId: parentContext.sessionId,
+        parentId: parentContext.traceId,
+        name: input.name,
+        type: input.type,
+        metadata: input.metadata
+    });
+    return {
+        traceId,
+        end: async (_endInput) => {
+        }
+    };
+}
+//# sourceMappingURL=middleware.js.map

package/dist/trace/middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/trace/middleware.ts"],"names":[],"mappings":";;AAsFA,sDA+FC;AAyBD,0CAiCC;AA1OD,+BAAoC;AAiFpC,SAAgB,qBAAqB,CAAC,OAA+B;IACnE,MAAM,EAAE,aAAa,GAAG,cAAc,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;IAE/F,OAAO,KAAK,EAAE,GAAY,EAAE,GAAa,EAAE,IAAkB,EAAE,EAAE;QAE/D,MAAM,SAAS,GAAI,GAAG,CAAC,OAAO,CAAC,aAAa,CAAY,IAAI,IAAA,SAAM,GAAE,CAAC;QAGrE,MAAM,QAAQ,GAAwB;YACpC,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,KAAK,EAAE,GAAG,CAAC,KAAK;SACjB,CAAC;QAEF,IAAI,cAAc,EAAE,CAAC;YACnB,QAAQ,CAAC,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;YAC/B,QAAQ,CAAC,EAAE,GAAG,GAAG,CAAC,EAAE,CAAC;YACrB,QAAQ,CAAC,SAAS,GAAG,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC;YACH,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,SAAS,CAAC;gBAClC,SAAS;gBACT,IAAI,EAAE,GAAG,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,IAAI,EAAE;gBACjC,IAAI,EAAE,MAAM;gBACZ,QAAQ;aACT,CAAC,CAAC;YAGH,MAAM,YAAY,GAAiB;gBACjC,SAAS;gBACT,OAAO;gBACP,QAAQ,EAAE,SAAS;aACpB,CAAC;YAEF,GAAG,CAAC,YAAY,GAAG,YAAY,CAAC;YAChC,GAAG,CAAC,MAAM,CAAC,SAAS,GAAG,SAAS,CAAC;YACjC,GAAG,CAAC,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC;YAC7B,GAAG,CAAC,MAAM,CAAC,YAAY,GAAG,YAAY,CAAC;YAGvC,MAAM,YAAY,GAAG,GAAG,CAAC,IAAI,CAAC;YAC9B,MAAM,YAAY,GAAG,GAAG,CAAC,IAAI,CAAC;YAC9B,MAAM,WAAW,GAAG,GAAG,CAAC,GAAG,CAAC;YAE5B,MAAM,QAAQ,GAAG,KAAK,EAAE,KAAa,EAAE,EAAE;gBACvC,IAAI,CAAC;oBACH,MAAM,OAAO,CAAC,OAAO,EAAE;wBACrB,MAAM,EAAE,KAAK,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;wBACvD,KAAK,EAAE,KAAK;4BACV,CAAC,CAAC;gCACE,OAAO,EAAE,KAAK,CAAC,OAAO;gCACtB,KAAK,EAAE,KAAK,CAAC,KAAK;6BACnB;4BACH,CAAC,CAAC,SAAS;wBACb,QAAQ,EAAE;4BACR,UAAU,EAAE,GAAG,CAAC,UAAU;4BAC1B,GAAG,CAAC,cAAc;gCAChB,CAAC,CAAC;oCACE,eAAe,EAAE,GAAG,CAAC,UAAU,EAAE;iCAClC;gCACH,CAAC,CAAC,EAAE,CAAC;yBACR;qBACF,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,2BAA2B,EAAE,GAAG,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC,CAAC;YAEF,GAAG,CAAC,IAAI,GAAG,UAAU,IAAS;gBAC5B,QAAQ,EAAE,CAAC;gBACX,OAAO,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACvC,CAAC,CAAC;YAEF,GAAG,CAAC,IAAI,GAAG,UAAU,IAAS;gBAC5B,QAAQ,EAAE,CAAC;gBACX,OAAO,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACvC,CAAC,CAAC;YAEF,GAAG,CAAC,GAAG,GAAG,UAAU,KAAW;gBAC7B,QAAQ,EAAE,CAAC;gBACX,OAAO,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACvC,CAAC,CAAC;YAGF,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAU,EAAE,EAAE;gBAC7B,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;YAEH,IAAI,EAAE,CAAC;QACT,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YACpD,IAAI,EAAE,CAAC;QACT,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAyBM,KAAK,UAAU,eAAe,CACnC,GAAY,EACZ,KAIC,EACD,WAAyD;IAEzD,MAAM,aAAa,GAAG,GAAG,CAAC,YAAY,CAAC;IACvC,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;IAED,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,WAAW,CAAC;QACpC,SAAS,EAAE,aAAa,CAAC,SAAS;QAClC,QAAQ,EAAE,aAAa,CAAC,OAAO;QAC/B,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ;KACzB,CAAC,CAAC;IAEH,OAAO;QACL,OAAO;QACP,GAAG,EAAE,KAAK,EAAE,SAIX,EAAE,EAAE;QAGL,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/trace/providers.d.ts

@@ -0,0 +1,83 @@
+import { TraceContext } from './types';
+import { OpenAIProvider } from '../providers/openai';
+import { AnthropicProvider } from '../providers/anthropic';
+import { ProviderRequest, ProviderResponse } from '../types/providers';
+interface TracedProviderOptions {
+    traceContext?: TraceContext;
+    startSpan: (input: {
+        sessionId: string;
+        parentId?: string;
+        name: string;
+        type: string;
+        metadata?: Record<string, any>;
+    }) => Promise<{
+        traceId: string;
+    }>;
+    endSpan: (traceId: string, input: {
+        status: 'ok' | 'error';
+        error?: {
+            message: string;
+            stack?: string;
+        };
+        model?: string;
+        tokens?: {
+            input: number;
+            output: number;
+        };
+        costUSD?: number;
+        metadata?: Record<string, any>;
+    }) => Promise<void>;
+    recordMessage?: (input: {
+        sessionId: string;
+        traceId: string;
+        role: 'user' | 'assistant' | 'system';
+        content: string;
+    }) => Promise<void>;
+    calculateCost?: (input: {
+        model: string;
+        inputTokens: number;
+        outputTokens: number;
+        provider: string;
+    }) => number;
+}
+export declare class TrackedOpenAI {
+    private provider;
+    private options;
+    constructor(config: any & TracedProviderOptions);
+    makeRequest(request: ProviderRequest): Promise<ProviderResponse>;
+    chatCompletion(params: {
+        model: string;
+        messages: Array<{
+            role: string;
+            content: string;
+        }>;
+        maxTokens?: number;
+        temperature?: number;
+        topP?: number;
+    }): Promise<ProviderResponse>;
+    getProvider(): OpenAIProvider;
+}
+export declare class TrackedAnthropic {
+    private provider;
+    private options;
+    constructor(config: any & TracedProviderOptions);
+    makeRequest(request: ProviderRequest): Promise<ProviderResponse>;
+    messages(params: {
+        model: string;
+        messages: Array<{
+            role: string;
+            content: string;
+        }>;
+        system?: string;
+        maxTokens?: number;
+        temperature?: number;
+    }): Promise<ProviderResponse>;
+    getProvider(): AnthropicProvider;
+}
+export declare function createTrackedProvider<T extends object>(provider: T, methodName: keyof T, traceContext: TraceContext | undefined, traceService: {
+    startSpan: TracedProviderOptions['startSpan'];
+    endSpan: TracedProviderOptions['endSpan'];
+    recordMessage?: TracedProviderOptions['recordMessage'];
+}): T;
+export {};
+//# sourceMappingURL=providers.d.ts.map

package/dist/trace/providers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers.d.ts","sourceRoot":"","sources":["../../src/trace/providers.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAIvE,UAAU,qBAAqB;IAI7B,YAAY,CAAC,EAAE,YAAY,CAAC;IAK5B,SAAS,EAAE,CAAC,KAAK,EAAE;QACjB,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,KAAK,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAKnC,OAAO,EAAE,CACP,OAAO,EAAE,MAAM,EACf,KAAK,EAAE;QACL,MAAM,EAAE,IAAI,GAAG,OAAO,CAAC;QACvB,KAAK,CAAC,EAAE;YAAE,OAAO,EAAE,MAAM,CAAC;YAAC,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC;QAC5C,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC;QAC3C,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,KACE,OAAO,CAAC,IAAI,CAAC,CAAC;IAKnB,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE;QACtB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;QACtC,OAAO,EAAE,MAAM,CAAC;KACjB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAKpB,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE;QACtB,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,QAAQ,EAAE,MAAM,CAAC;KAClB,KAAK,MAAM,CAAC;CACd;AAuBD,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAiB;IACjC,OAAO,CAAC,OAAO,CAAwB;gBAE3B,MAAM,EAAE,GAAG,GAAG,qBAAqB;IAiBzC,WAAW,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA8HhE,cAAc,CAAC,MAAM,EAAE;QAC3B,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACnD,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAgB7B,WAAW,IAAI,cAAc;CAG9B;AAuBD,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,OAAO,CAAwB;gBAE3B,MAAM,EAAE,GAAG,GAAG,qBAAqB;IAiBzC,WAAW,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA+HhE,QAAQ,CAAC,MAAM,EAAE;QACrB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACnD,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAsB7B,WAAW,IAAI,iBAAiB;CAGjC;AAeD,wBAAgB,qBAAqB,CAAC,CAAC,SAAS,MAAM,EACpD,QAAQ,EAAE,CAAC,EACX,UAAU,EAAE,MAAM,CAAC,EACnB,YAAY,EAAE,YAAY,GAAG,SAAS,EACtC,YAAY,EAAE;IACZ,SAAS,EAAE,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAC9C,OAAO,EAAE,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAC1C,aAAa,CAAC,EAAE,qBAAqB,CAAC,eAAe,CAAC,CAAC;CACxD,GACA,CAAC,CA+DH"}