@cogitator-ai/core 0.3.0 → 0.4.0

package/dist/causal/graph/causal-graph.js

@@ -0,0 +1,321 @@
+export class CausalGraphImpl {
+    id;
+    name;
+    nodes = new Map();
+    edges = new Map();
+    parentMap = new Map();
+    childMap = new Map();
+    edgeIndex = new Map();
+    createdAt;
+    updatedAt;
+    version;
+    constructor(id, name) {
+        this.id = id;
+        this.name = name;
+        this.createdAt = Date.now();
+        this.updatedAt = this.createdAt;
+        this.version = 1;
+    }
+    static fromData(data) {
+        const graph = new CausalGraphImpl(data.id, data.name);
+        graph.createdAt = data.createdAt;
+        graph.updatedAt = data.updatedAt;
+        graph.version = data.version;
+        for (const node of data.nodes) {
+            graph.addNode(node);
+        }
+        for (const edge of data.edges) {
+            graph.addEdge(edge);
+        }
+        return graph;
+    }
+    addNode(node) {
+        if (this.nodes.has(node.id)) {
+            throw new Error(`Node ${node.id} already exists`);
+        }
+        this.nodes.set(node.id, { ...node });
+        this.parentMap.set(node.id, new Set());
+        this.childMap.set(node.id, new Set());
+        this.touch();
+    }
+    removeNode(nodeId) {
+        if (!this.nodes.has(nodeId))
+            return;
+        const edgesToRemove = [];
+        for (const [edgeId, edge] of this.edges) {
+            if (edge.source === nodeId || edge.target === nodeId) {
+                edgesToRemove.push(edgeId);
+            }
+        }
+        for (const edgeId of edgesToRemove) {
+            this.removeEdge(edgeId);
+        }
+        this.nodes.delete(nodeId);
+        this.parentMap.delete(nodeId);
+        this.childMap.delete(nodeId);
+        this.touch();
+    }
+    getNode(nodeId) {
+        const node = this.nodes.get(nodeId);
+        return node ? { ...node } : undefined;
+    }
+    getNodes() {
+        return Array.from(this.nodes.values()).map((n) => ({ ...n }));
+    }
+    hasNode(nodeId) {
+        return this.nodes.has(nodeId);
+    }
+    addEdge(edge) {
+        if (this.edges.has(edge.id)) {
+            throw new Error(`Edge ${edge.id} already exists`);
+        }
+        if (!this.nodes.has(edge.source)) {
+            throw new Error(`Source node ${edge.source} does not exist`);
+        }
+        if (!this.nodes.has(edge.target)) {
+            throw new Error(`Target node ${edge.target} does not exist`);
+        }
+        const indexKey = `${edge.source}:${edge.target}`;
+        if (this.edgeIndex.has(indexKey)) {
+            throw new Error(`Edge from ${edge.source} to ${edge.target} already exists`);
+        }
+        this.edges.set(edge.id, { ...edge });
+        this.edgeIndex.set(indexKey, edge.id);
+        this.childMap.get(edge.source).add(edge.target);
+        this.parentMap.get(edge.target).add(edge.source);
+        this.touch();
+    }
+    removeEdge(edgeId) {
+        const edge = this.edges.get(edgeId);
+        if (!edge)
+            return;
+        const indexKey = `${edge.source}:${edge.target}`;
+        this.edgeIndex.delete(indexKey);
+        this.childMap.get(edge.source)?.delete(edge.target);
+        this.parentMap.get(edge.target)?.delete(edge.source);
+        this.edges.delete(edgeId);
+        this.touch();
+    }
+    getEdge(edgeId) {
+        const edge = this.edges.get(edgeId);
+        return edge ? { ...edge } : undefined;
+    }
+    getEdges() {
+        return Array.from(this.edges.values()).map((e) => ({ ...e }));
+    }
+    getEdgeBetween(source, target) {
+        const edgeId = this.edgeIndex.get(`${source}:${target}`);
+        return edgeId ? this.getEdge(edgeId) : undefined;
+    }
+    getParents(nodeId) {
+        const parentIds = this.parentMap.get(nodeId);
+        if (!parentIds)
+            return [];
+        return Array.from(parentIds)
+            .map((id) => this.getNode(id))
+            .filter((n) => n !== undefined);
+    }
+    getChildren(nodeId) {
+        const childIds = this.childMap.get(nodeId);
+        if (!childIds)
+            return [];
+        return Array.from(childIds)
+            .map((id) => this.getNode(id))
+            .filter((n) => n !== undefined);
+    }
+    getAncestors(nodeId) {
+        const ancestors = new Set();
+        const queue = [...(this.parentMap.get(nodeId) || [])];
+        while (queue.length > 0) {
+            const current = queue.shift();
+            if (ancestors.has(current))
+                continue;
+            ancestors.add(current);
+            const parents = this.parentMap.get(current);
+            if (parents) {
+                for (const parent of parents) {
+                    if (!ancestors.has(parent)) {
+                        queue.push(parent);
+                    }
+                }
+            }
+        }
+        return Array.from(ancestors)
+            .map((id) => this.getNode(id))
+            .filter((n) => n !== undefined);
+    }
+    getDescendants(nodeId) {
+        const descendants = new Set();
+        const queue = [...(this.childMap.get(nodeId) || [])];
+        while (queue.length > 0) {
+            const current = queue.shift();
+            if (descendants.has(current))
+                continue;
+            descendants.add(current);
+            const children = this.childMap.get(current);
+            if (children) {
+                for (const child of children) {
+                    if (!descendants.has(child)) {
+                        queue.push(child);
+                    }
+                }
+            }
+        }
+        return Array.from(descendants)
+            .map((id) => this.getNode(id))
+            .filter((n) => n !== undefined);
+    }
+    findPaths(from, to, maxLength = 10) {
+        const paths = [];
+        const visited = new Set();
+        const dfs = (current, target, path, edges, depth) => {
+            if (depth > maxLength)
+                return;
+            if (current === target) {
+                paths.push({
+                    nodes: [...path],
+                    edges: [...edges],
+                    totalStrength: edges.reduce((acc, e) => acc * e.strength, 1),
+                    isBlocked: false,
+                    blockingNodes: [],
+                });
+                return;
+            }
+            visited.add(current);
+            const children = this.childMap.get(current);
+            if (children) {
+                for (const child of children) {
+                    if (!visited.has(child)) {
+                        const edge = this.getEdgeBetween(current, child);
+                        if (edge) {
+                            dfs(child, target, [...path, child], [...edges, edge], depth + 1);
+                        }
+                    }
+                }
+            }
+            visited.delete(current);
+        };
+        dfs(from, to, [from], [], 0);
+        return paths;
+    }
+    getMarkovBlanket(nodeId) {
+        const blanket = new Set();
+        const parents = this.parentMap.get(nodeId);
+        if (parents) {
+            for (const p of parents)
+                blanket.add(p);
+        }
+        const children = this.childMap.get(nodeId);
+        if (children) {
+            for (const c of children) {
+                blanket.add(c);
+                const coparents = this.parentMap.get(c);
+                if (coparents) {
+                    for (const cp of coparents) {
+                        if (cp !== nodeId)
+                            blanket.add(cp);
+                    }
+                }
+            }
+        }
+        return Array.from(blanket)
+            .map((id) => this.getNode(id))
+            .filter((n) => n !== undefined);
+    }
+    topologicalSort() {
+        const inDegree = new Map();
+        for (const nodeId of this.nodes.keys()) {
+            inDegree.set(nodeId, 0);
+        }
+        for (const edge of this.edges.values()) {
+            inDegree.set(edge.target, (inDegree.get(edge.target) || 0) + 1);
+        }
+        const queue = [];
+        for (const [nodeId, degree] of inDegree) {
+            if (degree === 0)
+                queue.push(nodeId);
+        }
+        const result = [];
+        while (queue.length > 0) {
+            const current = queue.shift();
+            result.push(current);
+            const children = this.childMap.get(current);
+            if (children) {
+                for (const child of children) {
+                    const newDegree = (inDegree.get(child) || 0) - 1;
+                    inDegree.set(child, newDegree);
+                    if (newDegree === 0)
+                        queue.push(child);
+                }
+            }
+        }
+        if (result.length !== this.nodes.size) {
+            throw new Error('Graph contains a cycle');
+        }
+        return result;
+    }
+    hasCycle() {
+        const visited = new Set();
+        const recStack = new Set();
+        const hasCycleDFS = (nodeId) => {
+            visited.add(nodeId);
+            recStack.add(nodeId);
+            const children = this.childMap.get(nodeId);
+            if (children) {
+                for (const child of children) {
+                    if (!visited.has(child)) {
+                        if (hasCycleDFS(child))
+                            return true;
+                    }
+                    else if (recStack.has(child)) {
+                        return true;
+                    }
+                }
+            }
+            recStack.delete(nodeId);
+            return false;
+        };
+        for (const nodeId of this.nodes.keys()) {
+            if (!visited.has(nodeId)) {
+                if (hasCycleDFS(nodeId))
+                    return true;
+            }
+        }
+        return false;
+    }
+    clone() {
+        return CausalGraphImpl.fromData(this.toData());
+    }
+    toData() {
+        return {
+            id: this.id,
+            name: this.name,
+            nodes: this.getNodes(),
+            edges: this.getEdges(),
+            createdAt: this.createdAt,
+            updatedAt: this.updatedAt,
+            version: this.version,
+        };
+    }
+    touch() {
+        this.updatedAt = Date.now();
+        this.version++;
+    }
+}
+export function getTripleType(graph, a, b, c) {
+    const abEdge = graph.getEdgeBetween(a, b);
+    const baEdge = graph.getEdgeBetween(b, a);
+    const bcEdge = graph.getEdgeBetween(b, c);
+    const cbEdge = graph.getEdgeBetween(c, b);
+    if (abEdge && bcEdge) {
+        return 'chain';
+    }
+    if (baEdge && bcEdge) {
+        return 'fork';
+    }
+    if (abEdge && cbEdge) {
+        return 'collider';
+    }
+    return null;
+}
+//# sourceMappingURL=causal-graph.js.map

package/dist/causal/graph/causal-graph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"causal-graph.js","sourceRoot":"","sources":["../../../src/causal/graph/causal-graph.ts"],"names":[],"mappings":"AASA,MAAM,OAAO,eAAe;IACjB,EAAE,CAAS;IACX,IAAI,CAAS;IAEd,KAAK,GAA4B,IAAI,GAAG,EAAE,CAAC;IAC3C,KAAK,GAA4B,IAAI,GAAG,EAAE,CAAC;IAC3C,SAAS,GAA6B,IAAI,GAAG,EAAE,CAAC;IAChD,QAAQ,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC/C,SAAS,GAAwB,IAAI,GAAG,EAAE,CAAC;IAC3C,SAAS,CAAS;IAClB,SAAS,CAAS;IAClB,OAAO,CAAS;IAExB,YAAY,EAAU,EAAE,IAAY;QAClC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,IAAqB;QACnC,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QACjC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QACjC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAE7B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,CAAC,IAAgB;QACtB,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,EAAE,iBAAiB,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QACvC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,GAAG,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,UAAU,CAAC,MAAc;QACvB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO;QAEpC,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACxC,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;gBACrD,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;YACnC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC1B,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC9B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,OAAO,CAAC,MAAc;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACxC,CAAC;IAED,QAAQ;QACN,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,OAAO,CAAC,MAAc;QACpB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,CAAC,IAAgB;QACtB,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,EAAE,iBAAiB,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,MAAM,iBAAiB,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,MAAM,iBAAiB,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,QAAQ,GAAG,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QACjD,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,CAAC,MAAM,OAAO,IAAI,CAAC,MAAM,iBAAiB,CAAC,CAAC;QAC/E,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClD,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,UAAU,CAAC,MAAc;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,MAAM,QAAQ,GAAG,GAAG,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QACjD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1B,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,OAAO,CAAC,MAAc;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACxC,CAAC;IAED,QAAQ;QACN,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,cAAc,CAAC,MAAc,EAAE,MAAc;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;QACzD,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACnD,CAAC;IAED,UAAU,CAAC,MAAc;QACvB,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,SAAS;YAAE,OAAO,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;aACzB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,WAAW,CAAC,MAAc;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,QAAQ;YAAE,OAAO,EAAE,CAAC;QACzB,OAAO,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC;aACxB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAEtD,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YAC/B,IAAI,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC;gBAAE,SAAS;YACrC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACvB,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,OAAO,EAAE,CAAC;gBACZ,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;oBAC7B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC3B,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBACrB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;aACzB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;QACtC,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAErD,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YAC/B,IAAI,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC;gBAAE,SAAS;YACvC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;oBAC7B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC5B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACpB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC;aAC3B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,EAAU,EAAE,SAAS,GAAG,EAAE;QAChD,MAAM,KAAK,GAAiB,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAElC,MAAM,GAAG,GAAG,CACV,OAAe,EACf,MAAc,EACd,IAAc,EACd,KAAmB,EACnB,KAAa,EACb,EAAE;YACF,IAAI,KAAK,GAAG,SAAS;gBAAE,OAAO;YAC9B,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;gBACvB,KAAK,CAAC,IAAI,CAAC;oBACT,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC;oBAChB,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC;oBACjB,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAC5D,SAAS,EAAE,KAAK;oBAChB,aAAa,EAAE,EAAE;iBAClB,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;oBAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBACxB,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;wBACjD,IAAI,IAAI,EAAE,CAAC;4BACT,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,IAAI,CAAC,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;wBACpE,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YACD,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC1B,CAAC,CAAC;QAEF,GAAG,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC7B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gBAAgB,CAAC,MAAc;QAC7B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAElC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,OAAO,EAAE,CAAC;YACZ,KAAK,MAAM,CAAC,IAAI,OAAO;gBAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,QAAQ,EAAE,CAAC;YACb,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACf,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBACxC,IAAI,SAAS,EAAE,CAAC;oBACd,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;wBAC3B,IAAI,EAAE,KAAK,MAAM;4BAAE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;aACvB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAmB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,eAAe;QACb,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC3C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YACvC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC1B,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YACvC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACxC,IAAI,MAAM,KAAK,CAAC;gBAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAErB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC5C,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;oBAC7B,MAAM,SAAS,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;oBACjD,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;oBAC/B,IAAI,SAAS,KAAK,CAAC;wBAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,QAAQ;QACN,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAClC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;QAEnC,MAAM,WAAW,GAAG,CAAC,MAAc,EAAW,EAAE;YAC9C,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpB,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAErB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;oBAC7B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBACxB,IAAI,WAAW,CAAC,KAAK,CAAC;4BAAE,OAAO,IAAI,CAAC;oBACtC,CAAC;yBAAM,IAAI,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC/B,OAAO,IAAI,CAAC;oBACd,CAAC;gBACH,CAAC;YACH,CAAC;YAED,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACxB,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;QAEF,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC;YACvC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBACzB,IAAI,WAAW,CAAC,MAAM,CAAC;oBAAE,OAAO,IAAI,CAAC;YACvC,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK;QACH,OAAO,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,MAAM;QACJ,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE;YACtB,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE;YACtB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK;QACX,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;CACF;AAED,MAAM,UAAU,aAAa,CAC3B,KAAkB,EAClB,CAAS,EACT,CAAS,EACT,CAAS;IAET,MAAM,MAAM,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,KAAK,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAE1C,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;QACrB,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;QACrB,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/causal/graph/graph-builder.d.ts

@@ -0,0 +1,68 @@
+import type { CausalGraph, CausalRelationType, VariableType, StructuralEquation } from '@cogitator-ai/types';
+export declare class CausalGraphBuilder {
+    private graphId;
+    private graphName;
+    private nodes;
+    private pendingEdges;
+    private currentNode;
+    private constructor();
+    static create(name: string, id?: string): CausalGraphBuilder;
+    variable(id: string, name: string, type?: VariableType): CausalGraphBuilder;
+    treatment(id: string, name: string): CausalGraphBuilder;
+    outcome(id: string, name: string): CausalGraphBuilder;
+    confounder(id: string, name: string): CausalGraphBuilder;
+    mediator(id: string, name: string): CausalGraphBuilder;
+    instrumental(id: string, name: string): CausalGraphBuilder;
+    collider(id: string, name: string): CausalGraphBuilder;
+    latent(id: string, name: string): CausalGraphBuilder;
+    withDomain(type: 'continuous' | 'discrete' | 'binary' | 'categorical', options?: {
+        values?: (string | number | boolean)[];
+        min?: number;
+        max?: number;
+    }): CausalGraphBuilder;
+    withEquation(equation: StructuralEquation): CausalGraphBuilder;
+    causes(target: string, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+    }): CausalGraphBuilder;
+    enables(target: string, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+    }): CausalGraphBuilder;
+    prevents(target: string, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+    }): CausalGraphBuilder;
+    mediates(target: string, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+    }): CausalGraphBuilder;
+    confounds(target: string, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+    }): CausalGraphBuilder;
+    moderates(target: string, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+    }): CausalGraphBuilder;
+    edge(target: string, relationType: CausalRelationType, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+        timelag?: number;
+    }): CausalGraphBuilder;
+    from(nodeId: string): CausalGraphBuilder;
+    connect(source: string, target: string, relationType: CausalRelationType, options?: {
+        strength?: number;
+        confidence?: number;
+        mechanism?: string;
+    }): CausalGraphBuilder;
+    build(): CausalGraph;
+}
+//# sourceMappingURL=graph-builder.d.ts.map

package/dist/causal/graph/graph-builder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph-builder.d.ts","sourceRoot":"","sources":["../../../src/causal/graph/graph-builder.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAGV,WAAW,EACX,kBAAkB,EAClB,YAAY,EACZ,kBAAkB,EACnB,MAAM,qBAAqB,CAAC;AAK7B,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,KAAK,CAAsC;IACnD,OAAO,CAAC,YAAY,CAAqC;IACzD,OAAO,CAAC,WAAW,CAAuB;IAE1C,OAAO;IAKP,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,kBAAkB;IAO5D,QAAQ,CACN,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,MAAM,EACZ,IAAI,GAAE,YAAyB,GAC9B,kBAAkB;IAUrB,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,kBAAkB;IAIvD,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,kBAAkB;IAIrD,UAAU,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,kBAAkB;IAIxD,QAAQ,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,kBAAkB;IAItD,YAAY,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,kBAAkB;IAI1D,QAAQ,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,kBAAkB;IAItD,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,kBAAkB;IAIpD,UAAU,CACR,IAAI,EAAE,YAAY,GAAG,UAAU,GAAG,QAAQ,GAAG,aAAa,EAC1D,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,CAAC,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;QAAC,GAAG,CAAC,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,MAAM,CAAA;KAAE,GAC/E,kBAAkB;IASrB,YAAY,CAAC,QAAQ,EAAE,kBAAkB,GAAG,kBAAkB;IAS9D,MAAM,CACJ,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,kBAAkB;IAIrB,OAAO,CACL,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,kBAAkB;IAIrB,QAAQ,CACN,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,kBAAkB;IAIrB,QAAQ,CACN,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,kBAAkB;IAIrB,SAAS,CACP,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,kBAAkB;IAIrB,SAAS,CACP,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,kBAAkB;IAIrB,IAAI,CACF,MAAM,EAAE,MAAM,EACd,YAAY,EAAE,kBAAkB,EAChC,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,GACzF,kBAAkB;IAgBrB,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,kBAAkB;IAQxC,OAAO,CACL,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,MAAM,EACd,YAAY,EAAE,kBAAkB,EAChC,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,kBAAkB;IAYrB,KAAK,IAAI,WAAW;CAiBrB"}

package/dist/causal/graph/graph-builder.js

@@ -0,0 +1,128 @@
+import { CausalGraphImpl } from './causal-graph';
+let edgeCounter = 0;
+export class CausalGraphBuilder {
+    graphId;
+    graphName;
+    nodes = new Map();
+    pendingEdges = [];
+    currentNode = null;
+    constructor(id, name) {
+        this.graphId = id;
+        this.graphName = name;
+    }
+    static create(name, id) {
+        return new CausalGraphBuilder(id || `graph-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`, name);
+    }
+    variable(id, name, type = 'observed') {
+        this.nodes.set(id, {
+            id,
+            name,
+            variableType: type,
+        });
+        this.currentNode = id;
+        return this;
+    }
+    treatment(id, name) {
+        return this.variable(id, name, 'treatment');
+    }
+    outcome(id, name) {
+        return this.variable(id, name, 'outcome');
+    }
+    confounder(id, name) {
+        return this.variable(id, name, 'confounder');
+    }
+    mediator(id, name) {
+        return this.variable(id, name, 'mediator');
+    }
+    instrumental(id, name) {
+        return this.variable(id, name, 'instrumental');
+    }
+    collider(id, name) {
+        return this.variable(id, name, 'collider');
+    }
+    latent(id, name) {
+        return this.variable(id, name, 'latent');
+    }
+    withDomain(type, options) {
+        if (!this.currentNode) {
+            throw new Error('No current node to set domain on');
+        }
+        const node = this.nodes.get(this.currentNode);
+        node.domain = { type, ...options };
+        return this;
+    }
+    withEquation(equation) {
+        if (!this.currentNode) {
+            throw new Error('No current node to set equation on');
+        }
+        const node = this.nodes.get(this.currentNode);
+        node.equation = equation;
+        return this;
+    }
+    causes(target, options) {
+        return this.edge(target, 'causes', options);
+    }
+    enables(target, options) {
+        return this.edge(target, 'enables', options);
+    }
+    prevents(target, options) {
+        return this.edge(target, 'prevents', options);
+    }
+    mediates(target, options) {
+        return this.edge(target, 'mediates', options);
+    }
+    confounds(target, options) {
+        return this.edge(target, 'confounds', options);
+    }
+    moderates(target, options) {
+        return this.edge(target, 'moderates', options);
+    }
+    edge(target, relationType, options) {
+        if (!this.currentNode) {
+            throw new Error('No current node to create edge from');
+        }
+        this.pendingEdges.push({
+            source: this.currentNode,
+            target,
+            relationType,
+            strength: options?.strength ?? 1.0,
+            confidence: options?.confidence ?? 1.0,
+            mechanism: options?.mechanism,
+            timelag: options?.timelag,
+        });
+        return this;
+    }
+    from(nodeId) {
+        if (!this.nodes.has(nodeId)) {
+            throw new Error(`Node ${nodeId} does not exist`);
+        }
+        this.currentNode = nodeId;
+        return this;
+    }
+    connect(source, target, relationType, options) {
+        this.pendingEdges.push({
+            source,
+            target,
+            relationType,
+            strength: options?.strength ?? 1.0,
+            confidence: options?.confidence ?? 1.0,
+            mechanism: options?.mechanism,
+        });
+        return this;
+    }
+    build() {
+        const graph = new CausalGraphImpl(this.graphId, this.graphName);
+        for (const node of this.nodes.values()) {
+            graph.addNode(node);
+        }
+        for (const edgeData of this.pendingEdges) {
+            const edge = {
+                id: `edge-${++edgeCounter}`,
+                ...edgeData,
+            };
+            graph.addEdge(edge);
+        }
+        return graph;
+    }
+}
+//# sourceMappingURL=graph-builder.js.map

package/dist/causal/graph/graph-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph-builder.js","sourceRoot":"","sources":["../../../src/causal/graph/graph-builder.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,IAAI,WAAW,GAAG,CAAC,CAAC;AAEpB,MAAM,OAAO,kBAAkB;IACrB,OAAO,CAAS;IAChB,SAAS,CAAS;IAClB,KAAK,GAA4B,IAAI,GAAG,EAAE,CAAC;IAC3C,YAAY,GAAkC,EAAE,CAAC;IACjD,WAAW,GAAkB,IAAI,CAAC;IAE1C,YAAoB,EAAU,EAAE,IAAY;QAC1C,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;IACxB,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,IAAY,EAAE,EAAW;QACrC,OAAO,IAAI,kBAAkB,CAC3B,EAAE,IAAI,SAAS,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EACrE,IAAI,CACL,CAAC;IACJ,CAAC;IAED,QAAQ,CACN,EAAU,EACV,IAAY,EACZ,OAAqB,UAAU;QAE/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE;YACjB,EAAE;YACF,IAAI;YACJ,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,SAAS,CAAC,EAAU,EAAE,IAAY;QAChC,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO,CAAC,EAAU,EAAE,IAAY;QAC9B,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IAC5C,CAAC;IAED,UAAU,CAAC,EAAU,EAAE,IAAY;QACjC,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,YAAY,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ,CAAC,EAAU,EAAE,IAAY;QAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IAC7C,CAAC;IAED,YAAY,CAAC,EAAU,EAAE,IAAY;QACnC,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;IACjD,CAAC;IAED,QAAQ,CAAC,EAAU,EAAE,IAAY;QAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;IAC7C,CAAC;IAED,MAAM,CAAC,EAAU,EAAE,IAAY;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC3C,CAAC;IAED,UAAU,CACR,IAA0D,EAC1D,OAAgF;QAEhF,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAE,CAAC;QAC/C,IAAI,CAAC,MAAM,GAAG,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,YAAY,CAAC,QAA4B;QACvC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAE,CAAC;QAC/C,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CACJ,MAAc,EACd,OAAwE;QAExE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED,OAAO,CACL,MAAc,EACd,OAAwE;QAExE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ,CACN,MAAc,EACd,OAAwE;QAExE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED,QAAQ,CACN,MAAc,EACd,OAAwE;QAExE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED,SAAS,CACP,MAAc,EACd,OAAwE;QAExE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,SAAS,CACP,MAAc,EACd,OAAwE;QAExE,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,IAAI,CACF,MAAc,EACd,YAAgC,EAChC,OAA0F;QAE1F,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;YACrB,MAAM,EAAE,IAAI,CAAC,WAAW;YACxB,MAAM;YACN,YAAY;YACZ,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,GAAG;YAClC,UAAU,EAAE,OAAO,EAAE,UAAU,IAAI,GAAG;YACtC,SAAS,EAAE,OAAO,EAAE,SAAS;YAC7B,OAAO,EAAE,OAAO,EAAE,OAAO;SAC1B,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC,MAAc;QACjB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,QAAQ,MAAM,iBAAiB,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CACL,MAAc,EACd,MAAc,EACd,YAAgC,EAChC,OAAwE;QAExE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;YACrB,MAAM;YACN,MAAM;YACN,YAAY;YACZ,QAAQ,EAAE,OAAO,EAAE,QAAQ,IAAI,GAAG;YAClC,UAAU,EAAE,OAAO,EAAE,UAAU,IAAI,GAAG;YACtC,SAAS,EAAE,OAAO,EAAE,SAAS;SAC9B,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK;QACH,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAEhE,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YACvC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACzC,MAAM,IAAI,GAAe;gBACvB,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE;gBAC3B,GAAG,QAAQ;aACZ,CAAC;YACF,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/dist/causal/index.d.ts

@@ -0,0 +1,28 @@
+export { CausalGraphImpl, getTripleType } from './graph/causal-graph';
+export { CausalGraphBuilder } from './graph/graph-builder';
+export { dSeparation, findMinimalSeparatingSet } from './inference/d-separation';
+export { findBackdoorAdjustment, findFrontdoorAdjustment, findAllAdjustmentSets, } from './inference/adjustment';
+export { CounterfactualReasoner, evaluateCounterfactual } from './inference/counterfactual';
+export type { CounterfactualReasonerOptions } from './inference/counterfactual';
+export { CausalInferenceEngine } from './inference/inference-engine';
+export type { CausalInferenceEngineOptions } from './inference/inference-engine';
+export { CausalExtractor } from './discovery/causal-extractor';
+export type { CausalExtractorOptions } from './discovery/causal-extractor';
+export { CausalHypothesisGenerator } from './discovery/hypothesis-generator';
+export type { HypothesisGeneratorOptions } from './discovery/hypothesis-generator';
+export { CausalValidator } from './discovery/causal-validator';
+export type { CausalValidatorOptions, ValidationContext, ForkRequest } from './discovery/causal-validator';
+export { buildCausalExtractionPrompt, buildHypothesisGenerationPrompt, buildCausalValidationPrompt, buildErrorCausalAnalysisPrompt, parseCausalExtractionResponse, parseHypothesisResponse, parseValidationResponse, parseRootCauseResponse, } from './discovery/prompts';
+export type { ExtractedRelationship, CausalExtractionResult, GeneratedHypothesis, HypothesisGenerationResult, ValidationResult, RootCauseAnalysisResult, } from './discovery/prompts';
+export { CausalEffectPredictor } from './capabilities/effect-predictor';
+export type { EffectPredictorOptions } from './capabilities/effect-predictor';
+export { CausalExplainer } from './capabilities/causal-explainer';
+export type { CausalExplainerOptions } from './capabilities/causal-explainer';
+export { CausalPlanner } from './capabilities/causal-planner';
+export type { CausalPlannerOptions } from './capabilities/causal-planner';
+export { InMemoryCausalGraphStore } from './stores/causal-graph-store';
+export { InMemoryCausalPatternStore } from './stores/causal-pattern-store';
+export { InMemoryInterventionLog } from './stores/intervention-log';
+export { CausalReasoner } from './causal-reasoner';
+export type { CausalReasonerOptions } from './causal-reasoner';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/causal/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE3D,OAAO,EAAE,WAAW,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AACjF,OAAO,EACL,sBAAsB,EACtB,uBAAuB,EACvB,qBAAqB,GACtB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAC5F,YAAY,EAAE,6BAA6B,EAAE,MAAM,4BAA4B,CAAC;AAChF,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACrE,YAAY,EAAE,4BAA4B,EAAE,MAAM,8BAA8B,CAAC;AAEjF,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,YAAY,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAC7E,YAAY,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AACnF,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,YAAY,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAC3G,OAAO,EACL,2BAA2B,EAC3B,+BAA+B,EAC/B,2BAA2B,EAC3B,8BAA8B,EAC9B,6BAA6B,EAC7B,uBAAuB,EACvB,uBAAuB,EACvB,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EACV,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,0BAA0B,EAC1B,gBAAgB,EAChB,uBAAuB,GACxB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AACxE,YAAY,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAC9E,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAClE,YAAY,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAC9D,YAAY,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AAE1E,OAAO,EAAE,wBAAwB,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAC3E,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAEpE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,YAAY,EAAE,qBAAqB,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/causal/index.js

@@ -0,0 +1,18 @@
+export { CausalGraphImpl, getTripleType } from './graph/causal-graph';
+export { CausalGraphBuilder } from './graph/graph-builder';
+export { dSeparation, findMinimalSeparatingSet } from './inference/d-separation';
+export { findBackdoorAdjustment, findFrontdoorAdjustment, findAllAdjustmentSets, } from './inference/adjustment';
+export { CounterfactualReasoner, evaluateCounterfactual } from './inference/counterfactual';
+export { CausalInferenceEngine } from './inference/inference-engine';
+export { CausalExtractor } from './discovery/causal-extractor';
+export { CausalHypothesisGenerator } from './discovery/hypothesis-generator';
+export { CausalValidator } from './discovery/causal-validator';
+export { buildCausalExtractionPrompt, buildHypothesisGenerationPrompt, buildCausalValidationPrompt, buildErrorCausalAnalysisPrompt, parseCausalExtractionResponse, parseHypothesisResponse, parseValidationResponse, parseRootCauseResponse, } from './discovery/prompts';
+export { CausalEffectPredictor } from './capabilities/effect-predictor';
+export { CausalExplainer } from './capabilities/causal-explainer';
+export { CausalPlanner } from './capabilities/causal-planner';
+export { InMemoryCausalGraphStore } from './stores/causal-graph-store';
+export { InMemoryCausalPatternStore } from './stores/causal-pattern-store';
+export { InMemoryInterventionLog } from './stores/intervention-log';
+export { CausalReasoner } from './causal-reasoner';
+//# sourceMappingURL=index.js.map

package/dist/causal/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/causal/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE3D,OAAO,EAAE,WAAW,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AACjF,OAAO,EACL,sBAAsB,EACtB,uBAAuB,EACvB,qBAAqB,GACtB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AAE5F,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAGrE,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EAAE,yBAAyB,EAAE,MAAM,kCAAkC,CAAC;AAE7E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAE/D,OAAO,EACL,2BAA2B,EAC3B,+BAA+B,EAC/B,2BAA2B,EAC3B,8BAA8B,EAC9B,6BAA6B,EAC7B,uBAAuB,EACvB,uBAAuB,EACvB,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAU7B,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAExE,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAElE,OAAO,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAG9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,6BAA6B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAC3E,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAEpE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/causal/inference/adjustment.d.ts

@@ -0,0 +1,5 @@
+import type { CausalGraph, AdjustmentSet } from '@cogitator-ai/types';
+export declare function findBackdoorAdjustment(graph: CausalGraph, treatment: string, outcome: string): AdjustmentSet | null;
+export declare function findFrontdoorAdjustment(graph: CausalGraph, treatment: string, outcome: string): AdjustmentSet | null;
+export declare function findAllAdjustmentSets(graph: CausalGraph, treatment: string, outcome: string, maxSize?: number): AdjustmentSet[];
+//# sourceMappingURL=adjustment.d.ts.map

package/dist/causal/inference/adjustment.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"adjustment.d.ts","sourceRoot":"","sources":["../../../src/causal/inference/adjustment.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAGtE,wBAAgB,sBAAsB,CACpC,KAAK,EAAE,WAAW,EAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,GACd,aAAa,GAAG,IAAI,CA2CtB;AAwCD,wBAAgB,uBAAuB,CACrC,KAAK,EAAE,WAAW,EAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,GACd,aAAa,GAAG,IAAI,CA8BtB;AAiGD,wBAAgB,qBAAqB,CACnC,KAAK,EAAE,WAAW,EAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,MAAM,EACf,OAAO,SAAI,GACV,aAAa,EAAE,CA6CjB"}