@attest-ai/core 0.3.0

package/dist/adapters/anthropic.d.ts

@@ -0,0 +1,31 @@
+import type { Trace } from "../proto/types.js";
+interface AnthropicTextBlock {
+    type: "text";
+    text: string;
+}
+interface AnthropicToolUseBlock {
+    type: "tool_use";
+    name: string;
+    input: Record<string, unknown>;
+}
+type AnthropicContentBlock = AnthropicTextBlock | AnthropicToolUseBlock;
+interface AnthropicUsage {
+    input_tokens: number;
+    output_tokens: number;
+}
+interface AnthropicMessage {
+    model?: string;
+    usage?: AnthropicUsage;
+    content: AnthropicContentBlock[];
+}
+export declare class AnthropicAdapter {
+    private readonly agentId;
+    constructor(agentId?: string);
+    traceFromResponse(response: AnthropicMessage, options?: {
+        inputMessages?: Record<string, unknown>[];
+        costUsd?: number;
+        latencyMs?: number;
+    }): Trace;
+}
+export {};
+//# sourceMappingURL=anthropic.d.ts.map

package/dist/adapters/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/adapters/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAG/C,UAAU,kBAAkB;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,qBAAqB;IAC7B,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,KAAK,qBAAqB,GAAG,kBAAkB,GAAG,qBAAqB,CAAC;AAExE,UAAU,cAAc;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,UAAU,gBAAgB;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,cAAc,CAAC;IACvB,OAAO,EAAE,qBAAqB,EAAE,CAAC;CAClC;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAEjC,OAAO,CAAC,EAAE,MAAM;IAI5B,iBAAiB,CACf,QAAQ,EAAE,gBAAgB,EAC1B,OAAO,CAAC,EAAE;QACR,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;QAC1C,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,GACA,KAAK;CAwCT"}

package/dist/adapters/anthropic.js

@@ -0,0 +1,42 @@
+import { TraceBuilder } from "../trace.js";
+export class AnthropicAdapter {
+    agentId;
+    constructor(agentId) {
+        this.agentId = agentId;
+    }
+    traceFromResponse(response, options) {
+        const builder = new TraceBuilder(this.agentId);
+        if (options?.inputMessages) {
+            builder.setInput({ messages: options.inputMessages });
+        }
+        const completionParts = [];
+        for (const block of response.content) {
+            if (block.type === "text") {
+                completionParts.push(block.text);
+            }
+            else if (block.type === "tool_use") {
+                builder.addToolCall(block.name, {
+                    args: block.input,
+                });
+            }
+        }
+        const completionText = completionParts.join("\n");
+        const stepArgs = {};
+        if (response.model !== undefined)
+            stepArgs.model = response.model;
+        builder.addLlmCall("completion", { args: stepArgs, result: { completion: completionText } });
+        builder.setOutput({ message: completionText });
+        let totalTokens;
+        if (response.usage) {
+            totalTokens = response.usage.input_tokens + response.usage.output_tokens;
+        }
+        builder.setMetadata({
+            total_tokens: totalTokens,
+            cost_usd: options?.costUsd,
+            latency_ms: options?.latencyMs,
+            model: response.model,
+        });
+        return builder.build();
+    }
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/adapters/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/adapters/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AA0B3C,MAAM,OAAO,gBAAgB;IACV,OAAO,CAAqB;IAE7C,YAAY,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,iBAAiB,CACf,QAA0B,EAC1B,OAIC;QAED,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE/C,IAAI,OAAO,EAAE,aAAa,EAAE,CAAC;YAC3B,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,eAAe,GAAa,EAAE,CAAC;QACrC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACnC,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE;oBAC9B,IAAI,EAAE,KAAK,CAAC,KAAK;iBAClB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,cAAc,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAElD,MAAM,QAAQ,GAA4B,EAAE,CAAC;QAC7C,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS;YAAE,QAAQ,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAElE,OAAO,CAAC,UAAU,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,CAAC,CAAC;QAC7F,OAAO,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;QAE/C,IAAI,WAA+B,CAAC;QACpC,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC;QAC3E,CAAC;QAED,OAAO,CAAC,WAAW,CAAC;YAClB,YAAY,EAAE,WAAW;YACzB,QAAQ,EAAE,OAAO,EAAE,OAAO;YAC1B,UAAU,EAAE,OAAO,EAAE,SAAS;YAC9B,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;CACF"}

package/dist/adapters/gemini.d.ts

@@ -0,0 +1,30 @@
+import type { Trace } from "../proto/types.js";
+interface GeminiFunctionCall {
+    name: string;
+    args?: Record<string, unknown>;
+}
+interface GeminiPart {
+    text?: string;
+    functionCall?: GeminiFunctionCall;
+}
+interface GeminiCandidate {
+    content: {
+        parts: GeminiPart[];
+    };
+}
+interface GeminiResponse {
+    text?: string;
+    candidates?: GeminiCandidate[];
+}
+export declare class GeminiAdapter {
+    private readonly agentId;
+    constructor(agentId?: string);
+    traceFromResponse(response: GeminiResponse, options?: {
+        inputText?: string;
+        costUsd?: number;
+        latencyMs?: number;
+        model?: string;
+    }): Trace;
+}
+export {};
+//# sourceMappingURL=gemini.d.ts.map

package/dist/adapters/gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/adapters/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAG/C,UAAU,kBAAkB;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,UAAU,UAAU;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,kBAAkB,CAAC;CACnC;AAED,UAAU,eAAe;IACvB,OAAO,EAAE;QAAE,KAAK,EAAE,UAAU,EAAE,CAAA;KAAE,CAAC;CAClC;AAED,UAAU,cAAc;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,eAAe,EAAE,CAAC;CAChC;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAEjC,OAAO,CAAC,EAAE,MAAM;IAI5B,iBAAiB,CACf,QAAQ,EAAE,cAAc,EACxB,OAAO,CAAC,EAAE;QACR,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GACA,KAAK;CAsCT"}

package/dist/adapters/gemini.js

@@ -0,0 +1,41 @@
+import { TraceBuilder } from "../trace.js";
+export class GeminiAdapter {
+    agentId;
+    constructor(agentId) {
+        this.agentId = agentId;
+    }
+    traceFromResponse(response, options) {
+        const builder = new TraceBuilder(this.agentId);
+        if (options?.inputText) {
+            builder.setInput({ text: options.inputText });
+        }
+        let completionText = "";
+        if (response.text != null) {
+            completionText = response.text;
+        }
+        else if (response.candidates && response.candidates.length > 0) {
+            completionText = response.candidates[0].content.parts
+                .filter((p) => p.text != null)
+                .map((p) => p.text)
+                .join("");
+        }
+        if (response.candidates && response.candidates.length > 0) {
+            for (const part of response.candidates[0].content.parts) {
+                if (part.functionCall) {
+                    builder.addToolCall(part.functionCall.name, {
+                        args: part.functionCall.args ?? {},
+                    });
+                }
+            }
+        }
+        builder.addLlmCall("completion", { result: { completion: completionText } });
+        builder.setOutput({ message: completionText });
+        builder.setMetadata({
+            cost_usd: options?.costUsd,
+            latency_ms: options?.latencyMs,
+            model: options?.model,
+        });
+        return builder.build();
+    }
+}
+//# sourceMappingURL=gemini.js.map

package/dist/adapters/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/adapters/gemini.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAqB3C,MAAM,OAAO,aAAa;IACP,OAAO,CAAqB;IAE7C,YAAY,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,iBAAiB,CACf,QAAwB,EACxB,OAKC;QAED,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE/C,IAAI,OAAO,EAAE,SAAS,EAAE,CAAC;YACvB,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,cAAc,GAAG,EAAE,CAAC;QACxB,IAAI,QAAQ,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;YAC1B,cAAc,GAAG,QAAQ,CAAC,IAAI,CAAC;QACjC,CAAC;aAAM,IAAI,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjE,cAAc,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK;iBAClD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC;iBAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAK,CAAC;iBACnB,IAAI,CAAC,EAAE,CAAC,CAAC;QACd,CAAC;QAED,IAAI,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACxD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE;wBAC1C,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;qBACnC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,CAAC,UAAU,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,CAAC,CAAC;QAC7E,OAAO,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;QAE/C,OAAO,CAAC,WAAW,CAAC;YAClB,QAAQ,EAAE,OAAO,EAAE,OAAO;YAC1B,UAAU,EAAE,OAAO,EAAE,SAAS;YAC9B,KAAK,EAAE,OAAO,EAAE,KAAK;SACtB,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;CACF"}

package/dist/adapters/index.d.ts

@@ -0,0 +1,11 @@
+import type { Trace } from "../proto/types.js";
+export interface TraceAdapter {
+    capture(...args: unknown[]): Trace;
+}
+export { ManualAdapter } from "./manual.js";
+export { OpenAIAdapter } from "./openai.js";
+export { AnthropicAdapter } from "./anthropic.js";
+export { GeminiAdapter } from "./gemini.js";
+export { OllamaAdapter } from "./ollama.js";
+export { OTelAdapter } from "./otel.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/adapters/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAE/C,MAAM,WAAW,YAAY;IAC3B,OAAO,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,KAAK,CAAC;CACpC;AAED,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC"}

package/dist/adapters/index.js

@@ -0,0 +1,7 @@
+export { ManualAdapter } from "./manual.js";
+export { OpenAIAdapter } from "./openai.js";
+export { AnthropicAdapter } from "./anthropic.js";
+export { GeminiAdapter } from "./gemini.js";
+export { OllamaAdapter } from "./ollama.js";
+export { OTelAdapter } from "./otel.js";
+//# sourceMappingURL=index.js.map

package/dist/adapters/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/adapters/index.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC"}

package/dist/adapters/manual.d.ts

@@ -0,0 +1,9 @@
+import type { Trace } from "../proto/types.js";
+import { TraceBuilder } from "../trace.js";
+export declare class ManualAdapter {
+    private readonly agentId;
+    constructor(agentId?: string);
+    capture(builderFn: (builder: TraceBuilder) => void): Trace;
+    createBuilder(): TraceBuilder;
+}
+//# sourceMappingURL=manual.d.ts.map

package/dist/adapters/manual.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"manual.d.ts","sourceRoot":"","sources":["../../src/adapters/manual.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAEjC,OAAO,CAAC,EAAE,MAAM;IAI5B,OAAO,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,YAAY,KAAK,IAAI,GAAG,KAAK;IAM1D,aAAa,IAAI,YAAY;CAG9B"}

package/dist/adapters/manual.js

@@ -0,0 +1,16 @@
+import { TraceBuilder } from "../trace.js";
+export class ManualAdapter {
+    agentId;
+    constructor(agentId) {
+        this.agentId = agentId;
+    }
+    capture(builderFn) {
+        const builder = new TraceBuilder(this.agentId);
+        builderFn(builder);
+        return builder.build();
+    }
+    createBuilder() {
+        return new TraceBuilder(this.agentId);
+    }
+}
+//# sourceMappingURL=manual.js.map

package/dist/adapters/manual.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manual.js","sourceRoot":"","sources":["../../src/adapters/manual.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,MAAM,OAAO,aAAa;IACP,OAAO,CAAqB;IAE7C,YAAY,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,OAAO,CAAC,SAA0C;QAChD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,SAAS,CAAC,OAAO,CAAC,CAAC;QACnB,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;CACF"}

package/dist/adapters/ollama.d.ts

@@ -0,0 +1,19 @@
+import type { Trace } from "../proto/types.js";
+interface OllamaResponse {
+    model?: string;
+    message?: {
+        content?: string;
+    };
+    eval_count?: number;
+    prompt_eval_count?: number;
+}
+export declare class OllamaAdapter {
+    private readonly agentId;
+    constructor(agentId?: string);
+    traceFromResponse(response: OllamaResponse, options?: {
+        inputMessages?: Record<string, unknown>[];
+        latencyMs?: number;
+    }): Trace;
+}
+export {};
+//# sourceMappingURL=ollama.d.ts.map

package/dist/adapters/ollama.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/adapters/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAG/C,UAAU,cAAc;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAEjC,OAAO,CAAC,EAAE,MAAM;IAI5B,iBAAiB,CACf,QAAQ,EAAE,cAAc,EACxB,OAAO,CAAC,EAAE;QACR,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;QAC1C,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,GACA,KAAK;CA6BT"}

package/dist/adapters/ollama.js

@@ -0,0 +1,30 @@
+import { TraceBuilder } from "../trace.js";
+export class OllamaAdapter {
+    agentId;
+    constructor(agentId) {
+        this.agentId = agentId;
+    }
+    traceFromResponse(response, options) {
+        const builder = new TraceBuilder(this.agentId);
+        if (options?.inputMessages) {
+            builder.setInput({ messages: options.inputMessages });
+        }
+        const completionText = response.message?.content ?? "";
+        builder.addLlmCall("completion", {
+            args: { model: response.model ?? "" },
+            result: { completion: completionText },
+        });
+        builder.setOutput({ message: completionText });
+        let totalTokens;
+        if (response.eval_count !== undefined && response.prompt_eval_count !== undefined) {
+            totalTokens = response.eval_count + response.prompt_eval_count;
+        }
+        builder.setMetadata({
+            total_tokens: totalTokens,
+            latency_ms: options?.latencyMs,
+            model: response.model,
+        });
+        return builder.build();
+    }
+}
+//# sourceMappingURL=ollama.js.map

package/dist/adapters/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/adapters/ollama.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAS3C,MAAM,OAAO,aAAa;IACP,OAAO,CAAqB;IAE7C,YAAY,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,iBAAiB,CACf,QAAwB,EACxB,OAGC;QAED,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE/C,IAAI,OAAO,EAAE,aAAa,EAAE,CAAC;YAC3B,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAEvD,OAAO,CAAC,UAAU,CAAC,YAAY,EAAE;YAC/B,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,IAAI,EAAE,EAAE;YACrC,MAAM,EAAE,EAAE,UAAU,EAAE,cAAc,EAAE;SACvC,CAAC,CAAC;QAEH,OAAO,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;QAE/C,IAAI,WAA+B,CAAC;QACpC,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,IAAI,QAAQ,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;YAClF,WAAW,GAAG,QAAQ,CAAC,UAAU,GAAG,QAAQ,CAAC,iBAAiB,CAAC;QACjE,CAAC;QAED,OAAO,CAAC,WAAW,CAAC;YAClB,YAAY,EAAE,WAAW;YACzB,UAAU,EAAE,OAAO,EAAE,SAAS;YAC9B,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;CACF"}

package/dist/adapters/openai.d.ts

@@ -0,0 +1,34 @@
+import type { Trace } from "../proto/types.js";
+interface OpenAIUsage {
+    total_tokens?: number;
+}
+interface OpenAIToolCall {
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+interface OpenAIMessage {
+    content: string | null;
+    tool_calls?: OpenAIToolCall[];
+}
+interface OpenAIChoice {
+    message: OpenAIMessage;
+}
+interface OpenAIChatCompletion {
+    model?: string;
+    usage?: OpenAIUsage;
+    choices: OpenAIChoice[];
+}
+export declare class OpenAIAdapter {
+    private readonly agentId;
+    constructor(agentId?: string);
+    traceFromResponse(response: OpenAIChatCompletion, options?: {
+        inputMessages?: Record<string, unknown>[];
+        costUsd?: number;
+        latencyMs?: number;
+        structuredOutput?: Record<string, unknown>;
+    }): Trace;
+}
+export {};
+//# sourceMappingURL=openai.d.ts.map

package/dist/adapters/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/adapters/openai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAG/C,UAAU,WAAW;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,UAAU,cAAc;IACtB,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;CAC/C;AAED,UAAU,aAAa;IACrB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;CAC/B;AAED,UAAU,YAAY;IACpB,OAAO,EAAE,aAAa,CAAC;CACxB;AAED,UAAU,oBAAoB;IAC5B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,OAAO,EAAE,YAAY,EAAE,CAAC;CACzB;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAEjC,OAAO,CAAC,EAAE,MAAM;IAI5B,iBAAiB,CACf,QAAQ,EAAE,oBAAoB,EAC9B,OAAO,CAAC,EAAE;QACR,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;QAC1C,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KAC5C,GACA,KAAK;CA0CT"}

package/dist/adapters/openai.js

@@ -0,0 +1,42 @@
+import { TraceBuilder } from "../trace.js";
+export class OpenAIAdapter {
+    agentId;
+    constructor(agentId) {
+        this.agentId = agentId;
+    }
+    traceFromResponse(response, options) {
+        const builder = new TraceBuilder(this.agentId);
+        if (options?.inputMessages) {
+            builder.setInput({ messages: options.inputMessages });
+        }
+        const message = response.choices[0].message;
+        const completionText = message.content ?? "";
+        const stepArgs = {};
+        if (response.model !== undefined)
+            stepArgs.model = response.model;
+        const stepResult = { completion: completionText };
+        if (response.usage?.total_tokens !== undefined) {
+            stepResult.tokens = response.usage.total_tokens;
+        }
+        builder.addLlmCall("completion", { args: stepArgs, result: stepResult });
+        if (message.tool_calls) {
+            for (const tc of message.tool_calls) {
+                builder.addToolCall(tc.function.name, {
+                    args: { arguments: tc.function.arguments },
+                });
+            }
+        }
+        builder.setOutput({
+            message: completionText,
+            structured: options?.structuredOutput ?? {},
+        });
+        builder.setMetadata({
+            total_tokens: response.usage?.total_tokens,
+            cost_usd: options?.costUsd,
+            latency_ms: options?.latencyMs,
+            model: response.model,
+        });
+        return builder.build();
+    }
+}
+//# sourceMappingURL=openai.js.map

package/dist/adapters/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/adapters/openai.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAyB3C,MAAM,OAAO,aAAa;IACP,OAAO,CAAqB;IAE7C,YAAY,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,iBAAiB,CACf,QAA8B,EAC9B,OAKC;QAED,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAE/C,IAAI,OAAO,EAAE,aAAa,EAAE,CAAC;YAC3B,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,aAAa,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QAC5C,MAAM,cAAc,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;QAE7C,MAAM,QAAQ,GAA4B,EAAE,CAAC;QAC7C,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS;YAAE,QAAQ,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAElE,MAAM,UAAU,GAA4B,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;QAC3E,IAAI,QAAQ,CAAC,KAAK,EAAE,YAAY,KAAK,SAAS,EAAE,CAAC;YAC/C,UAAU,CAAC,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QAClD,CAAC;QAED,OAAO,CAAC,UAAU,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;QAEzE,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBACpC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE;oBACpC,IAAI,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE;iBAC3C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,CAAC,SAAS,CAAC;YAChB,OAAO,EAAE,cAAc;YACvB,UAAU,EAAE,OAAO,EAAE,gBAAgB,IAAI,EAAE;SAC5C,CAAC,CAAC;QAEH,OAAO,CAAC,WAAW,CAAC;YAClB,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,YAAY;YAC1C,QAAQ,EAAE,OAAO,EAAE,OAAO;YAC1B,UAAU,EAAE,OAAO,EAAE,SAAS;YAC9B,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;CACF"}

package/dist/adapters/otel.d.ts

@@ -0,0 +1,25 @@
+import type { Trace } from "../proto/types.js";
+interface OTelSpanContext {
+    traceId: string;
+}
+interface OTelSpan {
+    name: string;
+    startTime: [number, number];
+    endTime: [number, number];
+    attributes?: Record<string, unknown>;
+    parentSpanId?: string;
+    spanContext(): OTelSpanContext;
+}
+export declare class OTelAdapter {
+    private readonly agentId;
+    constructor(agentId?: string);
+    static fromSpans(spans: OTelSpan[], agentId?: string): Trace;
+    private buildTrace;
+    private findRootSpan;
+    private classifySpan;
+    private extractLlmStep;
+    private extractToolStep;
+    private spanMetadata;
+}
+export {};
+//# sourceMappingURL=otel.d.ts.map

package/dist/adapters/otel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"otel.d.ts","sourceRoot":"","sources":["../../src/adapters/otel.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAG/C,UAAU,eAAe;IACvB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,UAAU,QAAQ;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5B,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,IAAI,eAAe,CAAC;CAChC;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAEjC,OAAO,CAAC,EAAE,MAAM;IAI5B,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,KAAK;IAK5D,OAAO,CAAC,UAAU;IAuElB,OAAO,CAAC,YAAY;IAOpB,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,cAAc;IAgBtB,OAAO,CAAC,eAAe;IAWvB,OAAO,CAAC,YAAY;CAKrB"}

package/dist/adapters/otel.js

@@ -0,0 +1,133 @@
+import { TraceBuilder } from "../trace.js";
+export class OTelAdapter {
+    agentId;
+    constructor(agentId) {
+        this.agentId = agentId;
+    }
+    static fromSpans(spans, agentId) {
+        const adapter = new OTelAdapter(agentId);
+        return adapter.buildTrace(spans);
+    }
+    buildTrace(spans) {
+        const sorted = [...spans].sort((a, b) => {
+            const aTime = a.startTime[0] * 1e9 + a.startTime[1];
+            const bTime = b.startTime[0] * 1e9 + b.startTime[1];
+            return aTime - bTime;
+        });
+        const builder = new TraceBuilder(this.agentId);
+        const rootSpan = this.findRootSpan(sorted);
+        if (rootSpan) {
+            const traceId = rootSpan.spanContext().traceId;
+            if (traceId) {
+                builder.setTraceId(`otel_${traceId.slice(0, 16)}`);
+            }
+        }
+        let outputMessage = "";
+        let totalTokens;
+        let latencyMs;
+        let model;
+        for (const span of sorted) {
+            const attrs = span.attributes ?? {};
+            const stepType = this.classifySpan(attrs, span.name);
+            if (stepType === "llm_call") {
+                const [stepArgs, stepResult] = this.extractLlmStep(attrs);
+                builder.addLlmCall(span.name, {
+                    args: stepArgs,
+                    result: stepResult,
+                    metadata: this.spanMetadata(span),
+                });
+                const completion = attrs["gen_ai.completion"];
+                if (completion)
+                    outputMessage = String(completion);
+                const inputTokens = Number(attrs["gen_ai.usage.input_tokens"] ?? 0);
+                const outputTokens = Number(attrs["gen_ai.usage.output_tokens"] ?? 0);
+                const spanTokens = inputTokens + outputTokens;
+                if (spanTokens > 0)
+                    totalTokens = (totalTokens ?? 0) + spanTokens;
+                if (model === undefined) {
+                    const responseModel = attrs["gen_ai.response.model"];
+                    const requestModel = attrs["gen_ai.request.model"];
+                    if (responseModel)
+                        model = String(responseModel);
+                    else if (requestModel)
+                        model = String(requestModel);
+                }
+            }
+            else if (stepType === "tool_call") {
+                const [stepArgs, stepResult] = this.extractToolStep(attrs);
+                const toolName = String(attrs["gen_ai.tool.name"] ?? span.name);
+                builder.addToolCall(toolName, {
+                    args: stepArgs,
+                    result: stepResult,
+                    metadata: this.spanMetadata(span),
+                });
+            }
+        }
+        if (rootSpan) {
+            const startNs = rootSpan.startTime[0] * 1e9 + rootSpan.startTime[1];
+            const endNs = rootSpan.endTime[0] * 1e9 + rootSpan.endTime[1];
+            latencyMs = Math.floor((endNs - startNs) / 1_000_000);
+        }
+        builder.setOutput({ message: outputMessage });
+        builder.setMetadata({ total_tokens: totalTokens, latency_ms: latencyMs, model });
+        return builder.build();
+    }
+    findRootSpan(spans) {
+        for (const span of spans) {
+            if (!span.parentSpanId)
+                return span;
+        }
+        return spans[0];
+    }
+    classifySpan(attrs, name) {
+        const op = String(attrs["gen_ai.operation.name"] ?? "");
+        if (op === "chat" || op === "completion" || op === "generate_content" || "gen_ai.completion" in attrs) {
+            return "llm_call";
+        }
+        if (op === "tool" || "gen_ai.tool.name" in attrs) {
+            return "tool_call";
+        }
+        const lower = name.toLowerCase();
+        if (lower.includes("completion") || lower.includes("chat"))
+            return "llm_call";
+        if (lower.includes("tool"))
+            return "tool_call";
+        return undefined;
+    }
+    extractLlmStep(attrs) {
+        const args = {};
+        const result = {};
+        if ("gen_ai.request.model" in attrs)
+            args.model = String(attrs["gen_ai.request.model"]);
+        if ("gen_ai.system" in attrs)
+            args.system = String(attrs["gen_ai.system"]);
+        if ("gen_ai.prompt" in attrs)
+            args.prompt = String(attrs["gen_ai.prompt"]);
+        if ("gen_ai.completion" in attrs)
+            result.completion = String(attrs["gen_ai.completion"]);
+        if ("gen_ai.usage.input_tokens" in attrs)
+            result.input_tokens = Number(attrs["gen_ai.usage.input_tokens"]);
+        if ("gen_ai.usage.output_tokens" in attrs)
+            result.output_tokens = Number(attrs["gen_ai.usage.output_tokens"]);
+        if ("gen_ai.response.model" in attrs)
+            result.model = String(attrs["gen_ai.response.model"]);
+        return [args, result];
+    }
+    extractToolStep(attrs) {
+        const args = {};
+        const result = {};
+        if ("gen_ai.tool.call.id" in attrs)
+            args.call_id = String(attrs["gen_ai.tool.call.id"]);
+        if ("gen_ai.tool.parameters" in attrs)
+            args.parameters = attrs["gen_ai.tool.parameters"];
+        if ("gen_ai.tool.output" in attrs)
+            result.output = attrs["gen_ai.tool.output"];
+        return [args, result];
+    }
+    spanMetadata(span) {
+        const startNs = span.startTime[0] * 1e9 + span.startTime[1];
+        const endNs = span.endTime[0] * 1e9 + span.endTime[1];
+        return { duration_ms: Math.floor((endNs - startNs) / 1_000_000) };
+    }
+}
+//# sourceMappingURL=otel.js.map

package/dist/adapters/otel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"otel.js","sourceRoot":"","sources":["../../src/adapters/otel.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAe3C,MAAM,OAAO,WAAW;IACL,OAAO,CAAqB;IAE7C,YAAY,OAAgB;QAC1B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,SAAS,CAAC,KAAiB,EAAE,OAAgB;QAClD,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC;QACzC,OAAO,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IAEO,UAAU,CAAC,KAAiB;QAClC,MAAM,MAAM,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACpD,MAAM,KAAK,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACpD,OAAO,KAAK,GAAG,KAAK,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAE3C,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC;YAC/C,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,UAAU,CAAC,QAAQ,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,IAAI,aAAa,GAAG,EAAE,CAAC;QACvB,IAAI,WAA+B,CAAC;QACpC,IAAI,SAA6B,CAAC;QAClC,IAAI,KAAyB,CAAC;QAE9B,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAErD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;gBAC5B,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;gBAC1D,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE;oBAC5B,IAAI,EAAE,QAAQ;oBACd,MAAM,EAAE,UAAU;oBAClB,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;iBAClC,CAAC,CAAC;gBAEH,MAAM,UAAU,GAAG,KAAK,CAAC,mBAAmB,CAAC,CAAC;gBAC9C,IAAI,UAAU;oBAAE,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;gBAEnD,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpE,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,IAAI,CAAC,CAAC,CAAC;gBACtE,MAAM,UAAU,GAAG,WAAW,GAAG,YAAY,CAAC;gBAC9C,IAAI,UAAU,GAAG,CAAC;oBAAE,WAAW,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;gBAElE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,MAAM,aAAa,GAAG,KAAK,CAAC,uBAAuB,CAAC,CAAC;oBACrD,MAAM,YAAY,GAAG,KAAK,CAAC,sBAAsB,CAAC,CAAC;oBACnD,IAAI,aAAa;wBAAE,KAAK,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;yBAC5C,IAAI,YAAY;wBAAE,KAAK,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;iBAAM,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;gBACpC,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC3D,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC;gBAChE,OAAO,CAAC,WAAW,CAAC,QAAQ,EAAE;oBAC5B,IAAI,EAAE,QAAQ;oBACd,MAAM,EAAE,UAAU;oBAClB,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;iBAClC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC9D,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,GAAG,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC;QACxD,CAAC;QAED,OAAO,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;QAC9C,OAAO,CAAC,WAAW,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QAEjF,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAEO,YAAY,CAAC,KAAiB;QACpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,YAAY;gBAAE,OAAO,IAAI,CAAC;QACtC,CAAC;QACD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAEO,YAAY,CAAC,KAA8B,EAAE,IAAY;QAC/D,MAAM,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,EAAE,CAAC,CAAC;QACxD,IAAI,EAAE,KAAK,MAAM,IAAI,EAAE,KAAK,YAAY,IAAI,EAAE,KAAK,kBAAkB,IAAI,mBAAmB,IAAI,KAAK,EAAE,CAAC;YACtG,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,IAAI,EAAE,KAAK,MAAM,IAAI,kBAAkB,IAAI,KAAK,EAAE,CAAC;YACjD,OAAO,WAAW,CAAC;QACrB,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACjC,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,UAAU,CAAC;QAC9E,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,WAAW,CAAC;QAC/C,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,cAAc,CAAC,KAA8B;QACnD,MAAM,IAAI,GAA4B,EAAE,CAAC;QACzC,MAAM,MAAM,GAA4B,EAAE,CAAC;QAE3C,IAAI,sBAAsB,IAAI,KAAK;YAAE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC,CAAC;QACxF,IAAI,eAAe,IAAI,KAAK;YAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;QAC3E,IAAI,eAAe,IAAI,KAAK;YAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;QAE3E,IAAI,mBAAmB,IAAI,KAAK;YAAE,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;QACzF,IAAI,2BAA2B,IAAI,KAAK;YAAE,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,CAAC;QAC3G,IAAI,4BAA4B,IAAI,KAAK;YAAE,MAAM,CAAC,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;QAC9G,IAAI,uBAAuB,IAAI,KAAK;YAAE,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;QAE5F,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACxB,CAAC;IAEO,eAAe,CAAC,KAA8B;QACpD,MAAM,IAAI,GAA4B,EAAE,CAAC;QACzC,MAAM,MAAM,GAA4B,EAAE,CAAC;QAE3C,IAAI,qBAAqB,IAAI,KAAK;YAAE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;QACxF,IAAI,wBAAwB,IAAI,KAAK;YAAE,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,wBAAwB,CAAC,CAAC;QACzF,IAAI,oBAAoB,IAAI,KAAK;YAAE,MAAM,CAAC,MAAM,GAAG,KAAK,CAAC,oBAAoB,CAAC,CAAC;QAE/E,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACxB,CAAC;IAEO,YAAY,CAAC,IAAc;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACtD,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,GAAG,OAAO,CAAC,GAAG,SAAS,CAAC,EAAE,CAAC;IACpE,CAAC;CACF"}

package/dist/agent.d.ts

@@ -0,0 +1,14 @@
+import type { Trace } from "./proto/types.js";
+import { AgentResult } from "./result.js";
+import { TraceBuilder } from "./trace.js";
+export declare class Agent {
+    readonly name: string;
+    private readonly fn;
+    constructor(name: string, fn?: (builder: TraceBuilder, args: Record<string, unknown>) => unknown);
+    run(args?: Record<string, unknown>): AgentResult;
+    arun(args?: Record<string, unknown>): Promise<AgentResult>;
+    private setOutput;
+    withTrace(trace: Trace): AgentResult;
+}
+export declare function agent(name: string, fn: (builder: TraceBuilder, args: Record<string, unknown>) => unknown): (args?: Record<string, unknown>) => AgentResult;
+//# sourceMappingURL=agent.d.ts.map

package/dist/agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAG1C,qBAAa,KAAK;IAChB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAkF;gBAGnG,IAAI,EAAE,MAAM,EACZ,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO;IAMxE,GAAG,CAAC,IAAI,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,GAAG,WAAW;IAc9C,IAAI,CAAC,IAAI,GAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,GAAG,OAAO,CAAC,WAAW,CAAC;IAcpE,OAAO,CAAC,SAAS;IAUjB,SAAS,CAAC,KAAK,EAAE,KAAK,GAAG,WAAW;CAGrC;AAED,wBAAgB,KAAK,CACnB,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,GACpE,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,WAAW,CASjD"}

package/dist/agent.js

@@ -0,0 +1,54 @@
+import { AgentResult } from "./result.js";
+import { TraceBuilder } from "./trace.js";
+import { activeBuilder } from "./context.js";
+export class Agent {
+    name;
+    fn;
+    constructor(name, fn) {
+        this.name = name;
+        this.fn = fn;
+    }
+    run(args = {}) {
+        if (this.fn === undefined) {
+            throw new Error("No agent function provided. Pass fn to Agent().");
+        }
+        const builder = new TraceBuilder(this.name);
+        builder.setInput(args);
+        const output = activeBuilder.run(builder, () => this.fn(builder, args));
+        this.setOutput(builder, output);
+        return new AgentResult(builder.build());
+    }
+    async arun(args = {}) {
+        if (this.fn === undefined) {
+            throw new Error("No agent function provided. Pass fn to Agent().");
+        }
+        const builder = new TraceBuilder(this.name);
+        builder.setInput(args);
+        const output = await activeBuilder.run(builder, async () => this.fn(builder, args));
+        this.setOutput(builder, output);
+        return new AgentResult(builder.build());
+    }
+    setOutput(builder, output) {
+        if (output !== null && output !== undefined && typeof output === "object" && !Array.isArray(output)) {
+            builder.setOutput(output);
+        }
+        else if (output !== undefined) {
+            builder.setOutput({ result: output });
+        }
+        else {
+            builder.setOutput({ result: null });
+        }
+    }
+    withTrace(trace) {
+        return new AgentResult(trace);
+    }
+}
+export function agent(name, fn) {
+    const wrapped = new Agent(name, fn);
+    const wrapper = (args) => {
+        return wrapped.run(args ?? {});
+    };
+    wrapper.agent = wrapped;
+    return wrapper;
+}
+//# sourceMappingURL=agent.js.map

package/dist/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,MAAM,OAAO,KAAK;IACP,IAAI,CAAS;IACL,EAAE,CAAkF;IAErG,YACE,IAAY,EACZ,EAAsE;QAEtE,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAED,GAAG,CAAC,OAAgC,EAAE;QACpC,IAAI,IAAI,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAEvB,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,EAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACzE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAEhC,OAAO,IAAI,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAgC,EAAE;QAC3C,IAAI,IAAI,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAEvB,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,EAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;QACrF,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAEhC,OAAO,IAAI,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAC1C,CAAC;IAEO,SAAS,CAAC,OAAqB,EAAE,MAAe;QACtD,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,SAAS,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACpG,OAAO,CAAC,SAAS,CAAC,MAAiC,CAAC,CAAC;QACvD,CAAC;aAAM,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAChC,OAAO,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAY;QACpB,OAAO,IAAI,WAAW,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;CACF;AAED,MAAM,UAAU,KAAK,CACnB,IAAY,EACZ,EAAqE;IAErE,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAEpC,MAAM,OAAO,GAAG,CAAC,IAA8B,EAAe,EAAE;QAC9D,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IACjC,CAAC,CAAC;IAED,OAA8C,CAAC,KAAK,GAAG,OAAO,CAAC;IAChE,OAAO,OAAO,CAAC;AACjB,CAAC"}