@traccia2/sdk 0.0.1 → 0.0.2

package/dist/integrations/auto-langgraph.d.ts

@@ -0,0 +1,104 @@
+/**
+ * Automatic LangGraph instrumentation - simple one-line setup.
+ *
+ * Wrap your LangGraph nodes and conditionals to get tracing without
+ * any boilerplate code.
+ */
+/**
+ * Wrap a StateGraph with automatic tracing.
+ *
+ * @example
+ * const graph = new StateGraph(GraphState);
+ * graph.addNode('step1', processStep1);
+ * graph.addNode('step2', processStep2);
+ * graph.addEdge('step1', 'step2');
+ *
+ * // Just wrap your graph!
+ * const traced = wrapGraphWithTracing(graph);
+ * const compiled = traced.compile();
+ *
+ * const result = await compiled.invoke({ input: 'data' });
+ * // Fully traced with no other changes!
+ */
+export declare function wrapGraphWithTracing(graph: any, options?: any): any;
+/**
+ * Create a traced node function with one line.
+ *
+ * @example
+ * // Wrap your existing node:
+ * const existingNode = (state) => ({ result: 'done' });
+ * const traced = tracedNode('process-step', existingNode);
+ * graph.addNode('process', traced);
+ */
+export declare function tracedNode(nodeName: string, nodeFunction: (state: any) => Promise<any> | any): (state: any) => Promise<any>;
+/**
+ * Create a traced conditional edge router with one line.
+ *
+ * @example
+ * // Wrap your existing router:
+ * const decideRoute = (state) => state.approved ? 'done' : 'review';
+ * const traced = tracedConditional('route-check', decideRoute);
+ * graph.addConditionalEdges('check', traced);
+ */
+export declare function tracedConditional(conditionalName: string, routeFunction: (state: Record<string, any>) => string): (state: Record<string, any>) => string;
+/**
+ * Factory function to create a traced agent graph from simple config.
+ *
+ * @example
+ * const graph = createSimpleTracedGraph({
+ *   nodes: {
+ *     'process': async (state) => ({ result: await process(state.input) }),
+ *     'review': async (state) => ({ reviewed: true }),
+ *   },
+ *   edges: [
+ *     { from: '__start__', to: 'process' },
+ *     { from: 'process', to: 'review' },
+ *     { from: 'review', to: '__end__' },
+ *   ],
+ * });
+ *
+ * const result = await graph.compile().invoke({ input: 'data' });
+ */
+export declare function createSimpleTracedGraph(options: {
+    nodes: Record<string, (state: any) => Promise<any>>;
+    edges: Array<{
+        from: string;
+        to: string;
+    }>;
+    conditionals?: Record<string, {
+        router: (state: any) => string;
+        routes: string[];
+    }>;
+    startState?: Record<string, any>;
+}): Promise<any>;
+/**
+ * Trace a single agent function or tool with minimal setup.
+ *
+ * @example
+ * const tool = traceableFunction('my-tool', async (input) => {
+ *   return processInput(input);
+ * });
+ *
+ * const result = await tool('data');
+ * // Automatically traced!
+ */
+export declare function traceableFunction<T extends (...args: any[]) => any>(spanName: string, fn: T): T;
+/**
+ * Quick setup for a common agent pattern: input → process → route → output.
+ *
+ * @example
+ * const graph = await createAgentWorkflow({
+ *   processInput: async (state) => ({ processed: await analyze(state.input) }),
+ *   routeDecision: (state) => state.processed.needsReview ? 'review' : 'done',
+ *   reviewStep: async (state) => ({ reviewed: true }),
+ * });
+ *
+ * const result = await graph.compile().invoke({ input: 'user input' });
+ */
+export declare function createAgentWorkflow(options: {
+    processInput: (state: Record<string, any>) => Promise<Record<string, any>>;
+    routeDecision: (state: Record<string, any>) => string;
+    reviewStep?: (state: Record<string, any>) => Promise<Record<string, any>>;
+    finalStep?: (state: Record<string, any>) => Promise<Record<string, any>>;
+}): Promise<any>;
+//# sourceMappingURL=auto-langgraph.d.ts.map

package/dist/integrations/auto-langgraph.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-langgraph.d.ts","sourceRoot":"","sources":["../../src/integrations/auto-langgraph.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAQH;;;;;;;;;;;;;;;GAeG;AACH,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,GAAE,GAAQ,GAAG,GAAG,CAOvE;AAED;;;;;;;;GAQG;AACH,wBAAgB,UAAU,CACxB,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,GAC/C,CAAC,KAAK,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAE9B;AAED;;;;;;;;GAQG;AACH,wBAAgB,iBAAiB,CAC/B,eAAe,EAAE,MAAM,EACvB,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,MAAM,GACpD,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,MAAM,CAExC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAsB,uBAAuB,CAAC,OAAO,EAAE;IACrD,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;IACpD,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC3C,YAAY,CAAC,EAAE,MAAM,CACnB,MAAM,EACN;QAAE,MAAM,EAAE,CAAC,KAAK,EAAE,GAAG,KAAK,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,EAAE,CAAA;KAAE,CACrD,CAAC;IACF,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAClC,GAAG,OAAO,CAAC,GAAG,CAAC,CA0Cf;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,EACjE,QAAQ,EAAE,MAAM,EAChB,EAAE,EAAE,CAAC,GACJ,CAAC,CA4BH;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,mBAAmB,CAAC,OAAO,EAAE;IACjD,YAAY,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;IAC3E,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,MAAM,CAAC;IACtD,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;IAC1E,SAAS,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;CAC1E,GAAG,OAAO,CAAC,GAAG,CAAC,CA8Cf"}

package/dist/integrations/auto-langgraph.js

@@ -0,0 +1,240 @@
+"use strict";
+/**
+ * Automatic LangGraph instrumentation - simple one-line setup.
+ *
+ * Wrap your LangGraph nodes and conditionals to get tracing without
+ * any boilerplate code.
+ */
+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 __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.wrapGraphWithTracing = wrapGraphWithTracing;
+exports.tracedNode = tracedNode;
+exports.tracedConditional = tracedConditional;
+exports.createSimpleTracedGraph = createSimpleTracedGraph;
+exports.traceableFunction = traceableFunction;
+exports.createAgentWorkflow = createAgentWorkflow;
+const langgraph_instrumentation_1 = require("./langgraph-instrumentation");
+/**
+ * Wrap a StateGraph with automatic tracing.
+ *
+ * @example
+ * const graph = new StateGraph(GraphState);
+ * graph.addNode('step1', processStep1);
+ * graph.addNode('step2', processStep2);
+ * graph.addEdge('step1', 'step2');
+ *
+ * // Just wrap your graph!
+ * const traced = wrapGraphWithTracing(graph);
+ * const compiled = traced.compile();
+ *
+ * const result = await compiled.invoke({ input: 'data' });
+ * // Fully traced with no other changes!
+ */
+function wrapGraphWithTracing(graph, options = {}) {
+    return (0, langgraph_instrumentation_1.instrumentLangGraph)(graph, {
+        traceGraphExecution: true,
+        traceNodeExecution: true,
+        captureGraphState: options.captureState ?? false,
+        ...options,
+    });
+}
+/**
+ * Create a traced node function with one line.
+ *
+ * @example
+ * // Wrap your existing node:
+ * const existingNode = (state) => ({ result: 'done' });
+ * const traced = tracedNode('process-step', existingNode);
+ * graph.addNode('process', traced);
+ */
+function tracedNode(nodeName, nodeFunction) {
+    return (0, langgraph_instrumentation_1.createTracedNode)(nodeName, nodeFunction);
+}
+/**
+ * Create a traced conditional edge router with one line.
+ *
+ * @example
+ * // Wrap your existing router:
+ * const decideRoute = (state) => state.approved ? 'done' : 'review';
+ * const traced = tracedConditional('route-check', decideRoute);
+ * graph.addConditionalEdges('check', traced);
+ */
+function tracedConditional(conditionalName, routeFunction) {
+    return (0, langgraph_instrumentation_1.createTracedConditional)(conditionalName, routeFunction);
+}
+/**
+ * Factory function to create a traced agent graph from simple config.
+ *
+ * @example
+ * const graph = createSimpleTracedGraph({
+ *   nodes: {
+ *     'process': async (state) => ({ result: await process(state.input) }),
+ *     'review': async (state) => ({ reviewed: true }),
+ *   },
+ *   edges: [
+ *     { from: '__start__', to: 'process' },
+ *     { from: 'process', to: 'review' },
+ *     { from: 'review', to: '__end__' },
+ *   ],
+ * });
+ *
+ * const result = await graph.compile().invoke({ input: 'data' });
+ */
+async function createSimpleTracedGraph(options) {
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const langgraphModule = require('@langchain/langgraph');
+        const { nodes, edges, conditionals = {}, startState = {} } = options;
+        // Create state graph
+        const graph = new langgraphModule.StateGraph({
+            channels: startState,
+        });
+        // Add traced nodes
+        for (const [nodeName, nodeFunc] of Object.entries(nodes)) {
+            const tracedNodeFunc = tracedNode(nodeName, nodeFunc);
+            graph.addNode(nodeName, tracedNodeFunc);
+        }
+        // Add edges
+        for (const edge of edges) {
+            if (edge.from === '__start__') {
+                graph.setEntryPoint(edge.to);
+            }
+            else if (edge.to === '__end__') {
+                graph.setFinishPoint(edge.from);
+            }
+            else {
+                graph.addEdge(edge.from, edge.to);
+            }
+        }
+        // Add conditional edges
+        for (const [condName, condConfig] of Object.entries(conditionals)) {
+            const tracedRouter = tracedConditional(condName, condConfig.router);
+            graph.addConditionalEdges(condName, tracedRouter, condConfig.routes);
+        }
+        // Wrap entire graph with tracing
+        return wrapGraphWithTracing(graph);
+    }
+    catch (error) {
+        const err = error;
+        throw new Error(`Failed to create simple traced graph: ${err.message}. Make sure @langchain/langgraph is installed.`);
+    }
+}
+/**
+ * Trace a single agent function or tool with minimal setup.
+ *
+ * @example
+ * const tool = traceableFunction('my-tool', async (input) => {
+ *   return processInput(input);
+ * });
+ *
+ * const result = await tool('data');
+ * // Automatically traced!
+ */
+function traceableFunction(spanName, fn) {
+    return (async (...args) => {
+        const { getTracer } = await Promise.resolve().then(() => __importStar(require('../auto')));
+        const tracer = getTracer('traceable-function');
+        return tracer.startActiveSpan(spanName, async (span) => {
+            try {
+                span.setAttribute('args_count', args.length);
+                if (args[0] && typeof args[0] === 'object') {
+                    span.setAttribute('input_keys', Object.keys(args[0]).join(','));
+                }
+                const result = await fn(...args);
+                if (result && typeof result === 'object') {
+                    span.setAttribute('output_keys', Object.keys(result).join(','));
+                }
+                span.setAttribute('success', true);
+                return result;
+            }
+            catch (error) {
+                if (error instanceof Error) {
+                    span.recordException(error);
+                }
+                throw error;
+            }
+        });
+    });
+}
+/**
+ * Quick setup for a common agent pattern: input → process → route → output.
+ *
+ * @example
+ * const graph = await createAgentWorkflow({
+ *   processInput: async (state) => ({ processed: await analyze(state.input) }),
+ *   routeDecision: (state) => state.processed.needsReview ? 'review' : 'done',
+ *   reviewStep: async (state) => ({ reviewed: true }),
+ * });
+ *
+ * const result = await graph.compile().invoke({ input: 'user input' });
+ */
+async function createAgentWorkflow(options) {
+    try {
+        const { processInput, routeDecision, reviewStep, finalStep, } = options;
+        const nodes = {
+            process: processInput,
+        };
+        if (reviewStep)
+            nodes.review = reviewStep;
+        if (finalStep)
+            nodes.final = finalStep;
+        const edges = [
+            { from: '__start__', to: 'process' },
+        ];
+        const conditionals = {
+            route: {
+                router: routeDecision,
+                routes: ['done', ...(reviewStep ? ['review'] : [])],
+            },
+        };
+        edges.push({ from: 'process', to: 'route' });
+        if (reviewStep) {
+            edges.push({ from: 'review', to: finalStep ? 'final' : '__end__' });
+        }
+        if (finalStep) {
+            edges.push({ from: 'final', to: '__end__' });
+        }
+        return createSimpleTracedGraph({
+            nodes,
+            edges,
+            conditionals,
+        });
+    }
+    catch (error) {
+        const err = error;
+        throw new Error(`Failed to create agent workflow: ${err.message}`);
+    }
+}
+//# sourceMappingURL=auto-langgraph.js.map

package/dist/integrations/auto-langgraph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-langgraph.js","sourceRoot":"","sources":["../../src/integrations/auto-langgraph.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwBH,oDAOC;AAWD,gCAKC;AAWD,8CAKC;AAoBD,0DAkDC;AAaD,8CA+BC;AAcD,kDAmDC;AAhPD,2EAIqC;AAErC;;;;;;;;;;;;;;;GAeG;AACH,SAAgB,oBAAoB,CAAC,KAAU,EAAE,UAAe,EAAE;IAChE,OAAO,IAAA,+CAAmB,EAAC,KAAK,EAAE;QAChC,mBAAmB,EAAE,IAAI;QACzB,kBAAkB,EAAE,IAAI;QACxB,iBAAiB,EAAE,OAAO,CAAC,YAAY,IAAI,KAAK;QAChD,GAAG,OAAO;KACX,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;;GAQG;AACH,SAAgB,UAAU,CACxB,QAAgB,EAChB,YAAgD;IAEhD,OAAO,IAAA,4CAAgB,EAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;AAClD,CAAC;AAED;;;;;;;;GAQG;AACH,SAAgB,iBAAiB,CAC/B,eAAuB,EACvB,aAAqD;IAErD,OAAO,IAAA,mDAAuB,EAAC,eAAe,EAAE,aAAa,CAAC,CAAC;AACjE,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACI,KAAK,UAAU,uBAAuB,CAAC,OAQ7C;IACC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,eAAe,GAAG,OAAO,CAAC,sBAAsB,CAAC,CAAC;QACxD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,GAAG,EAAE,EAAE,UAAU,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;QAErE,qBAAqB;QACrB,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,UAAU,CAAC;YAC3C,QAAQ,EAAE,UAAU;SACrB,CAAC,CAAC;QAEH,mBAAmB;QACnB,KAAK,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzD,MAAM,cAAc,GAAG,UAAU,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACtD,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;QAC1C,CAAC;QAED,YAAY;QACZ,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC9B,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,IAAI,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;gBACjC,KAAK,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,KAAK,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YAClE,MAAM,YAAY,GAAG,iBAAiB,CAAC,QAAQ,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;YACpE,KAAK,CAAC,mBAAmB,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;QACvE,CAAC;QAED,iCAAiC;QACjC,OAAO,oBAAoB,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,GAAG,KAAc,CAAC;QAC3B,MAAM,IAAI,KAAK,CACb,yCAAyC,GAAG,CAAC,OAAO,gDAAgD,CACrG,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAgB,iBAAiB,CAC/B,QAAgB,EAChB,EAAK;IAEL,OAAO,CAAC,KAAK,EAAE,GAAG,IAAW,EAAE,EAAE;QAC/B,MAAM,EAAE,SAAS,EAAE,GAAG,wDAAa,SAAS,GAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAE/C,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACrD,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC7C,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;oBAC3C,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;gBAClE,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;gBAEjC,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;oBACzC,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;gBAClE,CAAC;gBACD,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBAEnC,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;oBAC3B,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC9B,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAM,CAAC;AACV,CAAC;AAED;;;;;;;;;;;GAWG;AACI,KAAK,UAAU,mBAAmB,CAAC,OAKzC;IACC,IAAI,CAAC;QACH,MAAM,EACJ,YAAY,EACZ,aAAa,EACb,UAAU,EACV,SAAS,GACV,GAAG,OAAO,CAAC;QAEZ,MAAM,KAAK,GAAiD;YAC1D,OAAO,EAAE,YAAY;SACtB,CAAC;QAEF,IAAI,UAAU;YAAE,KAAK,CAAC,MAAM,GAAG,UAAU,CAAC;QAC1C,IAAI,SAAS;YAAE,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;QAEvC,MAAM,KAAK,GAAwC;YACjD,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE;SACrC,CAAC;QAEF,MAAM,YAAY,GAAwB;YACxC,KAAK,EAAE;gBACL,MAAM,EAAE,aAAa;gBACrB,MAAM,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;aACpD;SACF,CAAC;QAEF,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAE7C,IAAI,UAAU,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,OAAO,uBAAuB,CAAC;YAC7B,KAAK;YACL,KAAK;YACL,YAAY;SACb,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,GAAG,KAAc,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,oCAAoC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IACrE,CAAC;AACH,CAAC"}

package/dist/integrations/index.d.ts

@@ -5,5 +5,8 @@
  * @module integrations
  */
 export { TraciaCallbackHandler } from './langchain-callback';
+export { getTraciaHandler, withTracing, createTracedOpenAI, createTracedAgentExecutor, createTracedLLMChain, traced, setupLangChainWithTracing, } from './auto-langchain';
 export { instrumentLangGraph, createTracedNode, createTracedConditional, } from './langgraph-instrumentation';
+export { wrapGraphWithTracing, tracedNode, tracedConditional, createSimpleTracedGraph, traceableFunction, createAgentWorkflow, } from './auto-langgraph';
+export { createOllamaWithTracing, setupOllamaWithTracing, createOllamaChatbot, createOllamaStreamingChatbot, getOllamaSetupInstructions, POPULAR_OLLAMA_MODELS, } from './ollama-integration';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAChB,uBAAuB,GACxB,MAAM,6BAA6B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EACL,gBAAgB,EAChB,WAAW,EACX,kBAAkB,EAClB,yBAAyB,EACzB,oBAAoB,EACpB,MAAM,EACN,yBAAyB,GAC1B,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,mBAAmB,EACnB,gBAAgB,EAChB,uBAAuB,GACxB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EACL,oBAAoB,EACpB,UAAU,EACV,iBAAiB,EACjB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,uBAAuB,EACvB,sBAAsB,EACtB,mBAAmB,EACnB,4BAA4B,EAC5B,0BAA0B,EAC1B,qBAAqB,GACtB,MAAM,sBAAsB,CAAC"}

package/dist/integrations/index.js

@@ -6,11 +6,36 @@
  * @module integrations
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createTracedConditional = exports.createTracedNode = exports.instrumentLangGraph = exports.TraciaCallbackHandler = void 0;
+exports.POPULAR_OLLAMA_MODELS = exports.getOllamaSetupInstructions = exports.createOllamaStreamingChatbot = exports.createOllamaChatbot = exports.setupOllamaWithTracing = exports.createOllamaWithTracing = exports.createAgentWorkflow = exports.traceableFunction = exports.createSimpleTracedGraph = exports.tracedConditional = exports.tracedNode = exports.wrapGraphWithTracing = exports.createTracedConditional = exports.createTracedNode = exports.instrumentLangGraph = exports.setupLangChainWithTracing = exports.traced = exports.createTracedLLMChain = exports.createTracedAgentExecutor = exports.createTracedOpenAI = exports.withTracing = exports.getTraciaHandler = exports.TraciaCallbackHandler = void 0;
+// LangChain integrations
 var langchain_callback_1 = require("./langchain-callback");
 Object.defineProperty(exports, "TraciaCallbackHandler", { enumerable: true, get: function () { return langchain_callback_1.TraciaCallbackHandler; } });
+var auto_langchain_1 = require("./auto-langchain");
+Object.defineProperty(exports, "getTraciaHandler", { enumerable: true, get: function () { return auto_langchain_1.getTraciaHandler; } });
+Object.defineProperty(exports, "withTracing", { enumerable: true, get: function () { return auto_langchain_1.withTracing; } });
+Object.defineProperty(exports, "createTracedOpenAI", { enumerable: true, get: function () { return auto_langchain_1.createTracedOpenAI; } });
+Object.defineProperty(exports, "createTracedAgentExecutor", { enumerable: true, get: function () { return auto_langchain_1.createTracedAgentExecutor; } });
+Object.defineProperty(exports, "createTracedLLMChain", { enumerable: true, get: function () { return auto_langchain_1.createTracedLLMChain; } });
+Object.defineProperty(exports, "traced", { enumerable: true, get: function () { return auto_langchain_1.traced; } });
+Object.defineProperty(exports, "setupLangChainWithTracing", { enumerable: true, get: function () { return auto_langchain_1.setupLangChainWithTracing; } });
+// LangGraph integrations
 var langgraph_instrumentation_1 = require("./langgraph-instrumentation");
 Object.defineProperty(exports, "instrumentLangGraph", { enumerable: true, get: function () { return langgraph_instrumentation_1.instrumentLangGraph; } });
 Object.defineProperty(exports, "createTracedNode", { enumerable: true, get: function () { return langgraph_instrumentation_1.createTracedNode; } });
 Object.defineProperty(exports, "createTracedConditional", { enumerable: true, get: function () { return langgraph_instrumentation_1.createTracedConditional; } });
+var auto_langgraph_1 = require("./auto-langgraph");
+Object.defineProperty(exports, "wrapGraphWithTracing", { enumerable: true, get: function () { return auto_langgraph_1.wrapGraphWithTracing; } });
+Object.defineProperty(exports, "tracedNode", { enumerable: true, get: function () { return auto_langgraph_1.tracedNode; } });
+Object.defineProperty(exports, "tracedConditional", { enumerable: true, get: function () { return auto_langgraph_1.tracedConditional; } });
+Object.defineProperty(exports, "createSimpleTracedGraph", { enumerable: true, get: function () { return auto_langgraph_1.createSimpleTracedGraph; } });
+Object.defineProperty(exports, "traceableFunction", { enumerable: true, get: function () { return auto_langgraph_1.traceableFunction; } });
+Object.defineProperty(exports, "createAgentWorkflow", { enumerable: true, get: function () { return auto_langgraph_1.createAgentWorkflow; } });
+// Ollama integrations
+var ollama_integration_1 = require("./ollama-integration");
+Object.defineProperty(exports, "createOllamaWithTracing", { enumerable: true, get: function () { return ollama_integration_1.createOllamaWithTracing; } });
+Object.defineProperty(exports, "setupOllamaWithTracing", { enumerable: true, get: function () { return ollama_integration_1.setupOllamaWithTracing; } });
+Object.defineProperty(exports, "createOllamaChatbot", { enumerable: true, get: function () { return ollama_integration_1.createOllamaChatbot; } });
+Object.defineProperty(exports, "createOllamaStreamingChatbot", { enumerable: true, get: function () { return ollama_integration_1.createOllamaStreamingChatbot; } });
+Object.defineProperty(exports, "getOllamaSetupInstructions", { enumerable: true, get: function () { return ollama_integration_1.getOllamaSetupInstructions; } });
+Object.defineProperty(exports, "POPULAR_OLLAMA_MODELS", { enumerable: true, get: function () { return ollama_integration_1.POPULAR_OLLAMA_MODELS; } });
 //# sourceMappingURL=index.js.map

package/dist/integrations/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,2DAA6D;AAApD,2HAAA,qBAAqB,OAAA;AAC9B,yEAIqC;AAHnC,gIAAA,mBAAmB,OAAA;AACnB,6HAAA,gBAAgB,OAAA;AAChB,oIAAA,uBAAuB,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/integrations/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,yBAAyB;AACzB,2DAA6D;AAApD,2HAAA,qBAAqB,OAAA;AAC9B,mDAQ0B;AAPxB,kHAAA,gBAAgB,OAAA;AAChB,6GAAA,WAAW,OAAA;AACX,oHAAA,kBAAkB,OAAA;AAClB,2HAAA,yBAAyB,OAAA;AACzB,sHAAA,oBAAoB,OAAA;AACpB,wGAAA,MAAM,OAAA;AACN,2HAAA,yBAAyB,OAAA;AAG3B,yBAAyB;AACzB,yEAIqC;AAHnC,gIAAA,mBAAmB,OAAA;AACnB,6HAAA,gBAAgB,OAAA;AAChB,oIAAA,uBAAuB,OAAA;AAEzB,mDAO0B;AANxB,sHAAA,oBAAoB,OAAA;AACpB,4GAAA,UAAU,OAAA;AACV,mHAAA,iBAAiB,OAAA;AACjB,yHAAA,uBAAuB,OAAA;AACvB,mHAAA,iBAAiB,OAAA;AACjB,qHAAA,mBAAmB,OAAA;AAGrB,sBAAsB;AACtB,2DAO8B;AAN5B,6HAAA,uBAAuB,OAAA;AACvB,4HAAA,sBAAsB,OAAA;AACtB,yHAAA,mBAAmB,OAAA;AACnB,kIAAA,4BAA4B,OAAA;AAC5B,gIAAA,0BAA0B,OAAA;AAC1B,2HAAA,qBAAqB,OAAA"}

package/dist/integrations/ollama-integration.d.ts

@@ -0,0 +1,122 @@
+/**
+ * Ollama integration for LangChain and LangGraph with automatic tracing.
+ *
+ * Run Ollama models locally and trace them automatically with Traccia.
+ *
+ * @example
+ * // First, start Ollama:
+ * // ollama pull llama2
+ * // ollama serve
+ *
+ * // Then use in your code:
+ * const model = await createOllamaWithTracing({
+ *   model: 'llama2',
+ *   baseUrl: 'http://localhost:11434',
+ * });
+ *
+ * const response = await model.invoke({ input: 'Hello!' });
+ * // Automatically traced!
+ */
+/**
+ * Create a traced Ollama model for LangChain.
+ *
+ * @example
+ * const model = await createOllamaWithTracing({
+ *   model: 'mistral',
+ *   baseUrl: 'http://localhost:11434',
+ *   temperature: 0.7,
+ * });
+ *
+ * const response = await model.invoke({ input: 'Write a poem' });
+ * // Automatically traced!
+ */
+export declare function createOllamaWithTracing(config: {
+    model: string;
+    baseUrl?: string;
+    temperature?: number;
+    topK?: number;
+    topP?: number;
+    [key: string]: any;
+}): Promise<any>;
+/**
+ * Set up LangChain with Ollama and automatic tracing.
+ *
+ * @example
+ * const { model, executor } = await setupOllamaWithTracing({
+ *   model: 'mistral',
+ *   tools: [weatherTool, calculatorTool],
+ *   systemPrompt: 'You are a helpful assistant.',
+ * });
+ *
+ * const result = await executor.invoke({ input: 'What is the weather?' });
+ * // Automatically traced!
+ */
+export declare function setupOllamaWithTracing(options: {
+    model: string;
+    modelConfig?: Record<string, any>;
+    baseUrl?: string;
+    tools?: any[];
+    systemPrompt?: string;
+}): Promise<{
+    model: any;
+    executor: any;
+    handler: any;
+}>;
+/**
+ * Create a simple chatbot using Ollama with automatic tracing.
+ *
+ * @example
+ * const chatbot = await createOllamaChatbot({
+ *   model: 'neural-chat',
+ *   systemPrompt: 'You are a helpful assistant.',
+ * });
+ *
+ * const response = await chatbot('What is machine learning?');
+ * // Automatically traced!
+ */
+export declare function createOllamaChatbot(options: {
+    model: string;
+    baseUrl?: string;
+    systemPrompt?: string;
+    temperature?: number;
+}): Promise<(input: string) => Promise<string>>;
+/**
+ * Available Ollama models you can pull and use.
+ *
+ * @example
+ * const models = POPULAR_OLLAMA_MODELS;
+ * // Use any of: mistral, neural-chat, llama2, orca-mini, etc.
+ */
+export declare const POPULAR_OLLAMA_MODELS: {
+    name: string;
+    description: string;
+    size: string;
+    command: string;
+}[];
+/**
+ * Helper to get setup instructions for Ollama.
+ *
+ * @example
+ * console.log(getOllamaSetupInstructions());
+ */
+export declare function getOllamaSetupInstructions(): string;
+/**
+ * Create a streaming chatbot with Ollama.
+ *
+ * @example
+ * const chatbot = await createOllamaStreamingChatbot({
+ *   model: 'mistral',
+ *   onChunk: (chunk) => process.stdout.write(chunk),
+ * });
+ *
+ * await chatbot('Tell me a story');
+ * // Streams response as it's generated!
+ */
+export declare function createOllamaStreamingChatbot(options: {
+    model: string;
+    baseUrl?: string;
+    systemPrompt?: string;
+    temperature?: number;
+    onChunk?: (chunk: string) => void;
+}): Promise<(input: string) => Promise<void>>;
+//# sourceMappingURL=ollama-integration.d.ts.map

package/dist/integrations/ollama-integration.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama-integration.d.ts","sourceRoot":"","sources":["../../src/integrations/ollama-integration.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AAIH;;;;;;;;;;;;GAYG;AACH,wBAAsB,uBAAuB,CAAC,MAAM,EAAE;IACpD,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB,GAAG,OAAO,CAAC,GAAG,CAAC,CA0Bf;AAED;;;;;;;;;;;;GAYG;AACH,wBAAsB,sBAAsB,CAAC,OAAO,EAAE;IACpD,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAClC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GAAG,OAAO,CAAC;IACV,KAAK,EAAE,GAAG,CAAC;IACX,QAAQ,EAAE,GAAG,CAAC;IACd,OAAO,EAAE,GAAG,CAAC;CACd,CAAC,CAgCD;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,mBAAmB,CAAC,OAAO,EAAE;IACjD,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,GAAG,OAAO,CAAC,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC,CA6C9C;AAED;;;;;;GAMG;AACH,eAAO,MAAM,qBAAqB;;;;;GAqCjC,CAAC;AAEF;;;;;GAKG;AACH,wBAAgB,0BAA0B,IAAI,MAAM,CAkCnD;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,4BAA4B,CAAC,OAAO,EAAE;IAC1D,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CACnC,GAAG,OAAO,CAAC,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,CA4D5C"}