societyai 0.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (114) hide show
  1. package/CHANGELOG.md +111 -0
  2. package/LICENSE +21 -0
  3. package/README.md +879 -0
  4. package/dist/builder.d.ts +181 -0
  5. package/dist/builder.d.ts.map +1 -0
  6. package/dist/builder.js +667 -0
  7. package/dist/builder.js.map +1 -0
  8. package/dist/config.d.ts +43 -0
  9. package/dist/config.d.ts.map +1 -0
  10. package/dist/config.js +11 -0
  11. package/dist/config.js.map +1 -0
  12. package/dist/context.d.ts +107 -0
  13. package/dist/context.d.ts.map +1 -0
  14. package/dist/context.js +319 -0
  15. package/dist/context.js.map +1 -0
  16. package/dist/errors.d.ts +31 -0
  17. package/dist/errors.d.ts.map +1 -0
  18. package/dist/errors.js +85 -0
  19. package/dist/errors.js.map +1 -0
  20. package/dist/events.d.ts +219 -0
  21. package/dist/events.d.ts.map +1 -0
  22. package/dist/events.js +395 -0
  23. package/dist/events.js.map +1 -0
  24. package/dist/graph.d.ts +104 -0
  25. package/dist/graph.d.ts.map +1 -0
  26. package/dist/graph.js +366 -0
  27. package/dist/graph.js.map +1 -0
  28. package/dist/index.d.ts +28 -0
  29. package/dist/index.d.ts.map +1 -0
  30. package/dist/index.js +113 -0
  31. package/dist/index.js.map +1 -0
  32. package/dist/logger.d.ts +13 -0
  33. package/dist/logger.d.ts.map +1 -0
  34. package/dist/logger.js +78 -0
  35. package/dist/logger.js.map +1 -0
  36. package/dist/memory.d.ts +146 -0
  37. package/dist/memory.d.ts.map +1 -0
  38. package/dist/memory.js +353 -0
  39. package/dist/memory.js.map +1 -0
  40. package/dist/metrics.d.ts +143 -0
  41. package/dist/metrics.d.ts.map +1 -0
  42. package/dist/metrics.js +271 -0
  43. package/dist/metrics.js.map +1 -0
  44. package/dist/middleware.d.ts +147 -0
  45. package/dist/middleware.d.ts.map +1 -0
  46. package/dist/middleware.js +484 -0
  47. package/dist/middleware.js.map +1 -0
  48. package/dist/models.d.ts +32 -0
  49. package/dist/models.d.ts.map +1 -0
  50. package/dist/models.js +211 -0
  51. package/dist/models.js.map +1 -0
  52. package/dist/patterns.d.ts +6 -0
  53. package/dist/patterns.d.ts.map +1 -0
  54. package/dist/patterns.js +68 -0
  55. package/dist/patterns.js.map +1 -0
  56. package/dist/pipeline.d.ts +84 -0
  57. package/dist/pipeline.d.ts.map +1 -0
  58. package/dist/pipeline.js +569 -0
  59. package/dist/pipeline.js.map +1 -0
  60. package/dist/retry.d.ts +5 -0
  61. package/dist/retry.d.ts.map +1 -0
  62. package/dist/retry.js +70 -0
  63. package/dist/retry.js.map +1 -0
  64. package/dist/society.d.ts +94 -0
  65. package/dist/society.d.ts.map +1 -0
  66. package/dist/society.js +721 -0
  67. package/dist/society.js.map +1 -0
  68. package/dist/strategies.d.ts +55 -0
  69. package/dist/strategies.d.ts.map +1 -0
  70. package/dist/strategies.js +678 -0
  71. package/dist/strategies.js.map +1 -0
  72. package/dist/tools.d.ts +88 -0
  73. package/dist/tools.d.ts.map +1 -0
  74. package/dist/tools.js +366 -0
  75. package/dist/tools.js.map +1 -0
  76. package/dist/types.d.ts +213 -0
  77. package/dist/types.d.ts.map +1 -0
  78. package/dist/types.js +19 -0
  79. package/dist/types.js.map +1 -0
  80. package/dist/validation.d.ts +64 -0
  81. package/dist/validation.d.ts.map +1 -0
  82. package/dist/validation.js +334 -0
  83. package/dist/validation.js.map +1 -0
  84. package/dist/worker-pool.d.ts +17 -0
  85. package/dist/worker-pool.d.ts.map +1 -0
  86. package/dist/worker-pool.js +80 -0
  87. package/dist/worker-pool.js.map +1 -0
  88. package/docs/README.md +468 -0
  89. package/docs/advanced.md +616 -0
  90. package/docs/aggregation-strategies.md +926 -0
  91. package/docs/api-reference.md +771 -0
  92. package/docs/architecture.md +648 -0
  93. package/docs/context-system.md +642 -0
  94. package/docs/event-system.md +1047 -0
  95. package/docs/examples.md +576 -0
  96. package/docs/getting-started.md +564 -0
  97. package/docs/graph-execution.md +389 -0
  98. package/docs/memory-system.md +497 -0
  99. package/docs/metrics-observability.md +560 -0
  100. package/docs/middleware-system.md +1038 -0
  101. package/docs/migration.md +296 -0
  102. package/docs/pipeline-patterns.md +761 -0
  103. package/docs/structured-output.md +612 -0
  104. package/docs/tool-calling.md +491 -0
  105. package/docs/workflows.md +740 -0
  106. package/examples/README.md +234 -0
  107. package/examples/advanced-patterns.ts +115 -0
  108. package/examples/complete-integration.ts +327 -0
  109. package/examples/graph-workflow.ts +161 -0
  110. package/examples/memory-system.ts +155 -0
  111. package/examples/metrics-tracking.ts +243 -0
  112. package/examples/structured-output.ts +231 -0
  113. package/examples/tool-calling.ts +163 -0
  114. package/package.json +94 -0
@@ -0,0 +1,104 @@
1
+ import { AgentConfig, AIModel } from './types';
2
+ export interface GraphStepResult {
3
+ agentId: string;
4
+ output: string;
5
+ success: boolean;
6
+ duration: number;
7
+ metadata?: Record<string, unknown>;
8
+ }
9
+ export declare enum NodeType {
10
+ START = "start",
11
+ END = "end",
12
+ AGENT = "agent",
13
+ PARALLEL = "parallel",
14
+ AGGREGATE = "aggregate",
15
+ CONDITION = "condition",
16
+ TRANSFORM = "transform",
17
+ LOOP = "loop"
18
+ }
19
+ export interface GraphNode {
20
+ id: string;
21
+ type: NodeType;
22
+ agentId?: string;
23
+ agentIds?: string[];
24
+ aggregator?: (results: GraphStepResult[]) => string;
25
+ condition?: (result: string, context: GraphContext) => boolean;
26
+ transformer?: (result: string, context: GraphContext) => string;
27
+ loopCondition?: (iteration: number, result: string, context: GraphContext) => boolean;
28
+ maxIterations?: number;
29
+ metadata?: Record<string, unknown>;
30
+ }
31
+ export interface GraphEdge {
32
+ from: string;
33
+ to: string;
34
+ condition?: (result: string, context: GraphContext) => boolean;
35
+ label?: string;
36
+ metadata?: Record<string, unknown>;
37
+ }
38
+ export interface ConditionalEdge {
39
+ from: string;
40
+ condition: (result: string, context: GraphContext) => boolean;
41
+ truePath: string;
42
+ falsePath: string;
43
+ }
44
+ export interface GraphContext {
45
+ input: string;
46
+ currentResult: string;
47
+ nodeResults: Map<string, GraphStepResult>;
48
+ sharedData: Map<string, unknown>;
49
+ iterationCounts: Map<string, number>;
50
+ executionPath: string[];
51
+ startTime: number;
52
+ signal?: AbortSignal;
53
+ }
54
+ export interface GraphResult {
55
+ output: string;
56
+ success: boolean;
57
+ nodeResults: Map<string, GraphStepResult>;
58
+ executionPath: string[];
59
+ duration: number;
60
+ errors?: Error[];
61
+ metadata: Record<string, unknown>;
62
+ }
63
+ export declare class SocietyAsModel implements AIModel {
64
+ private graph;
65
+ private agents;
66
+ private options;
67
+ private logger;
68
+ constructor(graph: SocietyGraph, agents: AgentConfig[], options?: {
69
+ name?: string;
70
+ description?: string;
71
+ });
72
+ name(): string;
73
+ process(prompt: unknown, signal?: AbortSignal): Promise<string>;
74
+ supportsPromptType(promptType: string): boolean;
75
+ }
76
+ export declare class SocietyGraph {
77
+ private nodes;
78
+ private edges;
79
+ private logger;
80
+ constructor(nodes: GraphNode[], edges: GraphEdge[]);
81
+ private validateGraph;
82
+ execute(input: string, agents: AgentConfig[], signal?: AbortSignal): Promise<GraphResult>;
83
+ private executeNode;
84
+ private executeAgentNode;
85
+ private executeParallelNode;
86
+ private executeAggregateNode;
87
+ private executeConditionNode;
88
+ private executeTransformNode;
89
+ private executeLoopNode;
90
+ private executeNextNodes;
91
+ private buildPrompt;
92
+ visualize(): string;
93
+ toMermaid(): string;
94
+ }
95
+ export declare class GraphBuilder {
96
+ private nodes;
97
+ private edges;
98
+ static create(): GraphBuilder;
99
+ addNode(id: string, type: NodeType, config?: Partial<Omit<GraphNode, 'id' | 'type'>>): this;
100
+ addEdge(from: string, to: string, config?: Partial<Omit<GraphEdge, 'from' | 'to'>>): this;
101
+ addConditionalEdge(config: ConditionalEdge): this;
102
+ build(): SocietyGraph;
103
+ }
104
+ //# sourceMappingURL=graph.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../src/graph.ts"],"names":[],"mappings":"AAiCA,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAY/C,MAAM,WAAW,eAAe;IAE9B,OAAO,EAAE,MAAM,CAAC;IAEhB,MAAM,EAAE,MAAM,CAAC;IAEf,OAAO,EAAE,OAAO,CAAC;IAEjB,QAAQ,EAAE,MAAM,CAAC;IAEjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAKD,oBAAY,QAAQ;IAElB,KAAK,UAAU;IAEf,GAAG,QAAQ;IAEX,KAAK,UAAU;IAEf,QAAQ,aAAa;IAErB,SAAS,cAAc;IAEvB,SAAS,cAAc;IAEvB,SAAS,cAAc;IAEvB,IAAI,SAAS;CACd;AAKD,MAAM,WAAW,SAAS;IAExB,EAAE,EAAE,MAAM,CAAC;IAEX,IAAI,EAAE,QAAQ,CAAC;IAEf,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IAEpB,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,eAAe,EAAE,KAAK,MAAM,CAAC;IAEpD,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,KAAK,OAAO,CAAC;IAE/D,WAAW,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,KAAK,MAAM,CAAC;IAEhE,aAAa,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,KAAK,OAAO,CAAC;IAEtF,aAAa,CAAC,EAAE,MAAM,CAAC;IAEvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAKD,MAAM,WAAW,SAAS;IAExB,IAAI,EAAE,MAAM,CAAC;IAEb,EAAE,EAAE,MAAM,CAAC;IAEX,SAAS,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,KAAK,OAAO,CAAC;IAE/D,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAKD,MAAM,WAAW,eAAe;IAE9B,IAAI,EAAE,MAAM,CAAC;IAEb,SAAS,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,KAAK,OAAO,CAAC;IAE9D,QAAQ,EAAE,MAAM,CAAC;IAEjB,SAAS,EAAE,MAAM,CAAC;CACnB;AAKD,MAAM,WAAW,YAAY;IAE3B,KAAK,EAAE,MAAM,CAAC;IAEd,aAAa,EAAE,MAAM,CAAC;IAEtB,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IAE1C,UAAU,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAEjC,eAAe,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAErC,aAAa,EAAE,MAAM,EAAE,CAAC;IAExB,SAAS,EAAE,MAAM,CAAC;IAElB,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAKD,MAAM,WAAW,WAAW;IAE1B,MAAM,EAAE,MAAM,CAAC;IAEf,OAAO,EAAE,OAAO,CAAC;IAEjB,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;IAE1C,aAAa,EAAE,MAAM,EAAE,CAAC;IAExB,QAAQ,EAAE,MAAM,CAAC;IAEjB,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC;IAEjB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAUD,qBAAa,cAAe,YAAW,OAAO;IAI1C,OAAO,CAAC,KAAK;IACb,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,OAAO;IALjB,OAAO,CAAC,MAAM,CAAe;gBAGnB,KAAK,EAAE,YAAY,EACnB,MAAM,EAAE,WAAW,EAAE,EACrB,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAO;IAM/D,IAAI,IAAI,MAAM;IAOR,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;IAuBrE,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;CAGhD;AASD,qBAAa,YAAY;IACvB,OAAO,CAAC,KAAK,CAAqC;IAClD,OAAO,CAAC,KAAK,CAAuC;IACpD,OAAO,CAAC,MAAM,CAAe;gBAG3B,KAAK,EAAE,SAAS,EAAE,EAClB,KAAK,EAAE,SAAS,EAAE;IA4BpB,OAAO,CAAC,aAAa;IA8Df,OAAO,CACX,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,WAAW,EAAE,EACrB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,WAAW,CAAC;YAoDT,WAAW;YAmEX,gBAAgB;YAmBhB,mBAAmB;YA0CnB,oBAAoB;YAYpB,oBAAoB;IA6BlC,OAAO,CAAC,oBAAoB;YAUd,eAAe;YA2Bf,gBAAgB;IAuB9B,OAAO,CAAC,WAAW;IAqBnB,SAAS,IAAI,MAAM;IAkBnB,SAAS,IAAI,MAAM;CA+CpB;AASD,qBAAa,YAAY;IACvB,OAAO,CAAC,KAAK,CAAmB;IAChC,OAAO,CAAC,KAAK,CAAmB;IAEhC,MAAM,CAAC,MAAM,IAAI,YAAY;IAO7B,OAAO,CACL,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,QAAQ,EACd,MAAM,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,MAAM,CAAC,CAAC,GAC/C,IAAI;IAQP,OAAO,CACL,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,MAAM,EACV,MAAM,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC,CAAC,GAC/C,IAAI;IAQP,kBAAkB,CAAC,MAAM,EAAE,eAAe,GAAG,IAAI;IAejD,KAAK,IAAI,YAAY;CAGtB"}
package/dist/graph.js ADDED
@@ -0,0 +1,366 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.GraphBuilder = exports.SocietyGraph = exports.SocietyAsModel = exports.NodeType = void 0;
4
+ const logger_1 = require("./logger");
5
+ const worker_pool_1 = require("./worker-pool");
6
+ const errors_1 = require("./errors");
7
+ var NodeType;
8
+ (function (NodeType) {
9
+ NodeType["START"] = "start";
10
+ NodeType["END"] = "end";
11
+ NodeType["AGENT"] = "agent";
12
+ NodeType["PARALLEL"] = "parallel";
13
+ NodeType["AGGREGATE"] = "aggregate";
14
+ NodeType["CONDITION"] = "condition";
15
+ NodeType["TRANSFORM"] = "transform";
16
+ NodeType["LOOP"] = "loop";
17
+ })(NodeType || (exports.NodeType = NodeType = {}));
18
+ class SocietyAsModel {
19
+ constructor(graph, agents, options = {}) {
20
+ this.graph = graph;
21
+ this.agents = agents;
22
+ this.options = options;
23
+ this.logger = (0, logger_1.getLogger)();
24
+ }
25
+ name() {
26
+ return this.options.name || 'SocietyModel';
27
+ }
28
+ async process(prompt, signal) {
29
+ const input = String(prompt);
30
+ this.logger.info(`Recursive Society '${this.name()}' started execution`);
31
+ try {
32
+ const result = await this.graph.execute(input, this.agents, signal);
33
+ if (!result.success) {
34
+ const error = result.errors?.[0] || new Error('Unknown error in recursive society');
35
+ throw error;
36
+ }
37
+ this.logger.info(`Recursive Society '${this.name()}' completed execution`);
38
+ return result.output;
39
+ }
40
+ catch (error) {
41
+ this.logger.error(`Recursive Society '${this.name()}' failed: ${error.message}`);
42
+ throw error;
43
+ }
44
+ }
45
+ supportsPromptType(promptType) {
46
+ return promptType === 'text';
47
+ }
48
+ }
49
+ exports.SocietyAsModel = SocietyAsModel;
50
+ class SocietyGraph {
51
+ constructor(nodes, edges) {
52
+ this.nodes = new Map();
53
+ this.edges = new Map();
54
+ this.logger = (0, logger_1.getLogger)();
55
+ for (const node of nodes) {
56
+ this.nodes.set(node.id, node);
57
+ }
58
+ for (const edge of edges) {
59
+ if (!this.nodes.has(edge.from)) {
60
+ throw new errors_1.ProcessingFailedError(`Source node not found: ${edge.from}`);
61
+ }
62
+ if (!this.nodes.has(edge.to)) {
63
+ throw new errors_1.ProcessingFailedError(`Target node not found: ${edge.to}`);
64
+ }
65
+ if (!this.edges.has(edge.from)) {
66
+ this.edges.set(edge.from, []);
67
+ }
68
+ this.edges.get(edge.from).push(edge);
69
+ }
70
+ this.validateGraph();
71
+ }
72
+ validateGraph() {
73
+ const startNodes = Array.from(this.nodes.values()).filter((n) => n.type === NodeType.START);
74
+ if (startNodes.length === 0) {
75
+ throw new errors_1.ProcessingFailedError('Graph must have at least one START node');
76
+ }
77
+ const endNodes = Array.from(this.nodes.values()).filter((n) => n.type === NodeType.END);
78
+ if (endNodes.length === 0) {
79
+ throw new errors_1.ProcessingFailedError('Graph must have at least one END node');
80
+ }
81
+ for (const node of this.nodes.values()) {
82
+ switch (node.type) {
83
+ case NodeType.AGENT:
84
+ if (!node.agentId) {
85
+ throw new errors_1.ProcessingFailedError(`AGENT node ${node.id} must have agentId`);
86
+ }
87
+ break;
88
+ case NodeType.PARALLEL:
89
+ if (!node.agentIds || node.agentIds.length === 0) {
90
+ throw new errors_1.ProcessingFailedError(`PARALLEL node ${node.id} must have agentIds`);
91
+ }
92
+ break;
93
+ case NodeType.AGGREGATE:
94
+ if (!node.aggregator) {
95
+ throw new errors_1.ProcessingFailedError(`AGGREGATE node ${node.id} must have aggregator function`);
96
+ }
97
+ break;
98
+ case NodeType.CONDITION:
99
+ if (!node.condition) {
100
+ throw new errors_1.ProcessingFailedError(`CONDITION node ${node.id} must have condition function`);
101
+ }
102
+ break;
103
+ case NodeType.TRANSFORM:
104
+ if (!node.transformer) {
105
+ throw new errors_1.ProcessingFailedError(`TRANSFORM node ${node.id} must have transformer function`);
106
+ }
107
+ break;
108
+ }
109
+ }
110
+ }
111
+ async execute(input, agents, signal) {
112
+ const startTime = Date.now();
113
+ const context = {
114
+ input,
115
+ currentResult: input,
116
+ nodeResults: new Map(),
117
+ sharedData: new Map(),
118
+ iterationCounts: new Map(),
119
+ executionPath: [],
120
+ startTime,
121
+ signal,
122
+ };
123
+ try {
124
+ const startNodes = Array.from(this.nodes.values()).filter((n) => n.type === NodeType.START);
125
+ for (const startNode of startNodes) {
126
+ await this.executeNode(startNode, context, agents);
127
+ }
128
+ const duration = Date.now() - startTime;
129
+ return {
130
+ output: context.currentResult,
131
+ success: true,
132
+ nodeResults: context.nodeResults,
133
+ executionPath: context.executionPath,
134
+ duration,
135
+ metadata: { totalNodes: this.nodes.size },
136
+ };
137
+ }
138
+ catch (error) {
139
+ const duration = Date.now() - startTime;
140
+ return {
141
+ output: context.currentResult,
142
+ success: false,
143
+ nodeResults: context.nodeResults,
144
+ executionPath: context.executionPath,
145
+ duration,
146
+ errors: [error],
147
+ metadata: { totalNodes: this.nodes.size },
148
+ };
149
+ }
150
+ }
151
+ async executeNode(node, context, agents) {
152
+ if (context.signal?.aborted) {
153
+ throw new Error('Execution cancelled');
154
+ }
155
+ context.executionPath.push(node.id);
156
+ this.logger.debug(`Executing node: ${node.id} (${node.type})`);
157
+ let result = context.currentResult;
158
+ switch (node.type) {
159
+ case NodeType.START:
160
+ result = context.input;
161
+ break;
162
+ case NodeType.END:
163
+ return;
164
+ case NodeType.AGENT:
165
+ result = await this.executeAgentNode(node, context, agents);
166
+ break;
167
+ case NodeType.PARALLEL:
168
+ result = await this.executeParallelNode(node, context, agents);
169
+ break;
170
+ case NodeType.AGGREGATE:
171
+ result = await this.executeAggregateNode(node, context);
172
+ break;
173
+ case NodeType.CONDITION:
174
+ result = await this.executeConditionNode(node, context, agents);
175
+ return;
176
+ case NodeType.TRANSFORM:
177
+ result = this.executeTransformNode(node, context);
178
+ break;
179
+ case NodeType.LOOP:
180
+ result = await this.executeLoopNode(node, context, agents);
181
+ break;
182
+ }
183
+ context.currentResult = result;
184
+ context.nodeResults.set(node.id, {
185
+ agentId: node.agentId || node.id,
186
+ output: result,
187
+ success: true,
188
+ duration: 0,
189
+ });
190
+ await this.executeNextNodes(node, context, agents);
191
+ }
192
+ async executeAgentNode(node, context, agents) {
193
+ const agent = agents.find((a) => a.id === node.agentId);
194
+ if (!agent) {
195
+ throw new errors_1.ProcessingFailedError(`Agent not found: ${node.agentId}`);
196
+ }
197
+ const prompt = this.buildPrompt(agent, context.currentResult, context);
198
+ const result = await agent.model.process(prompt, context.signal);
199
+ return result;
200
+ }
201
+ async executeParallelNode(node, context, agents) {
202
+ const nodeAgents = node.agentIds.map((id) => {
203
+ const agent = agents.find((a) => a.id === id);
204
+ if (!agent) {
205
+ throw new errors_1.ProcessingFailedError(`Agent not found: ${id}`);
206
+ }
207
+ return agent;
208
+ });
209
+ const pool = new worker_pool_1.WorkerPool(nodeAgents.length);
210
+ const results = [];
211
+ await Promise.all(nodeAgents.map((agent) => pool.submit(async () => {
212
+ const prompt = this.buildPrompt(agent, context.currentResult, context);
213
+ const result = await agent.model.process(prompt, context.signal);
214
+ const stepResult = {
215
+ agentId: agent.id,
216
+ output: result,
217
+ success: true,
218
+ duration: 0,
219
+ };
220
+ results.push(stepResult);
221
+ return stepResult;
222
+ })));
223
+ return results.map((r) => r.output).join('\n\n');
224
+ }
225
+ async executeAggregateNode(node, context) {
226
+ const results = Array.from(context.nodeResults.values());
227
+ return node.aggregator(results);
228
+ }
229
+ async executeConditionNode(node, context, agents) {
230
+ const conditionResult = node.condition(context.currentResult, context);
231
+ const edges = this.edges.get(node.id) || [];
232
+ const trueEdge = edges.find((e) => e.label === 'true');
233
+ const falseEdge = edges.find((e) => e.label === 'false');
234
+ const nextNodeId = conditionResult
235
+ ? trueEdge?.to
236
+ : falseEdge?.to;
237
+ if (nextNodeId) {
238
+ const nextNode = this.nodes.get(nextNodeId);
239
+ if (nextNode) {
240
+ await this.executeNode(nextNode, context, agents);
241
+ }
242
+ }
243
+ return context.currentResult;
244
+ }
245
+ executeTransformNode(node, context) {
246
+ return node.transformer(context.currentResult, context);
247
+ }
248
+ async executeLoopNode(node, context, agents) {
249
+ const maxIterations = node.maxIterations || 10;
250
+ let iteration = context.iterationCounts.get(node.id) || 0;
251
+ while (iteration < maxIterations) {
252
+ iteration++;
253
+ context.iterationCounts.set(node.id, iteration);
254
+ await this.executeNextNodes(node, context, agents);
255
+ if (node.loopCondition?.(iteration, context.currentResult, context)) {
256
+ break;
257
+ }
258
+ }
259
+ return context.currentResult;
260
+ }
261
+ async executeNextNodes(node, context, agents) {
262
+ const edges = this.edges.get(node.id) || [];
263
+ for (const edge of edges) {
264
+ if (edge.condition && !edge.condition(context.currentResult, context)) {
265
+ continue;
266
+ }
267
+ const nextNode = this.nodes.get(edge.to);
268
+ if (nextNode) {
269
+ await this.executeNode(nextNode, context, agents);
270
+ }
271
+ }
272
+ }
273
+ buildPrompt(agent, input, context) {
274
+ const template = agent.role.promptTemplate || '{input}';
275
+ return template
276
+ .replace('{input}', input)
277
+ .replace('{context}', JSON.stringify(Object.fromEntries(context.sharedData)))
278
+ .replace('{history}', Array.from(context.nodeResults.entries())
279
+ .map(([id, result]) => `[${id}]: ${result.output}`)
280
+ .join('\n'));
281
+ }
282
+ visualize() {
283
+ const lines = ['Graph Structure:', ''];
284
+ for (const node of this.nodes.values()) {
285
+ lines.push(`Node: ${node.id} (${node.type})`);
286
+ const edges = this.edges.get(node.id) || [];
287
+ for (const edge of edges) {
288
+ const label = edge.label ? ` [${edge.label}]` : '';
289
+ lines.push(` → ${edge.to}${label}`);
290
+ }
291
+ }
292
+ return lines.join('\n');
293
+ }
294
+ toMermaid() {
295
+ const lines = ['graph TD'];
296
+ lines.push(' classDef start fill:#f9f,stroke:#333,stroke-width:2px;');
297
+ lines.push(' classDef end fill:#9f9,stroke:#333,stroke-width:2px;');
298
+ lines.push(' classDef agent fill:#bbf,stroke:#333,stroke-width:1px;');
299
+ lines.push(' classDef condition fill:#f90,stroke:#333,stroke-width:1px;');
300
+ for (const node of this.nodes.values()) {
301
+ let shape = '[ ' + node.id + ' ]';
302
+ let style = '';
303
+ switch (node.type) {
304
+ case NodeType.START:
305
+ shape = '((' + node.id + '))';
306
+ style = ':::start';
307
+ break;
308
+ case NodeType.END:
309
+ shape = '((' + node.id + '))';
310
+ style = ':::end';
311
+ break;
312
+ case NodeType.CONDITION:
313
+ shape = '{' + node.id + '}';
314
+ style = ':::condition';
315
+ break;
316
+ case NodeType.AGENT:
317
+ shape = '[' + (node.agentId || node.id) + ']';
318
+ style = ':::agent';
319
+ break;
320
+ }
321
+ lines.push(` ${node.id}${shape}${style}`);
322
+ }
323
+ for (const [from, edges] of this.edges.entries()) {
324
+ for (const edge of edges) {
325
+ const label = edge.label ? `|${edge.label}|` : '-->';
326
+ const arrow = edge.label ? '-->' : '-->';
327
+ lines.push(` ${from} ${arrow} ${label} ${edge.to}`);
328
+ }
329
+ }
330
+ return lines.join('\n');
331
+ }
332
+ }
333
+ exports.SocietyGraph = SocietyGraph;
334
+ class GraphBuilder {
335
+ constructor() {
336
+ this.nodes = [];
337
+ this.edges = [];
338
+ }
339
+ static create() {
340
+ return new GraphBuilder();
341
+ }
342
+ addNode(id, type, config) {
343
+ this.nodes.push({ id, type, ...config });
344
+ return this;
345
+ }
346
+ addEdge(from, to, config) {
347
+ this.edges.push({ from, to, ...config });
348
+ return this;
349
+ }
350
+ addConditionalEdge(config) {
351
+ this.addEdge(config.from, config.truePath, {
352
+ condition: config.condition,
353
+ label: 'true',
354
+ });
355
+ this.addEdge(config.from, config.falsePath, {
356
+ condition: (result, ctx) => !config.condition(result, ctx),
357
+ label: 'false',
358
+ });
359
+ return this;
360
+ }
361
+ build() {
362
+ return new SocietyGraph(this.nodes, this.edges);
363
+ }
364
+ }
365
+ exports.GraphBuilder = GraphBuilder;
366
+ //# sourceMappingURL=graph.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"graph.js","sourceRoot":"","sources":["../src/graph.ts"],"names":[],"mappings":";;;AAkCA,qCAAqC;AACrC,+CAA2C;AAC3C,qCAAiD;AAyBjD,IAAY,QAiBX;AAjBD,WAAY,QAAQ;IAElB,2BAAe,CAAA;IAEf,uBAAW,CAAA;IAEX,2BAAe,CAAA;IAEf,iCAAqB,CAAA;IAErB,mCAAuB,CAAA;IAEvB,mCAAuB,CAAA;IAEvB,mCAAuB,CAAA;IAEvB,yBAAa,CAAA;AACf,CAAC,EAjBW,QAAQ,wBAAR,QAAQ,QAiBnB;AA4GD,MAAa,cAAc;IAGzB,YACU,KAAmB,EACnB,MAAqB,EACrB,UAAmD,EAAE;QAFrD,UAAK,GAAL,KAAK,CAAc;QACnB,WAAM,GAAN,MAAM,CAAe;QACrB,YAAO,GAAP,OAAO,CAA8C;QALvD,WAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;IAM1B,CAAC;IAKJ,IAAI;QACF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,cAAc,CAAC;IAC7C,CAAC;IAKD,KAAK,CAAC,OAAO,CAAC,MAAe,EAAE,MAAoB;QACjD,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,IAAI,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;QAEzE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAEpE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;gBACpF,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,IAAI,CAAC,IAAI,EAAE,uBAAuB,CAAC,CAAC;YAC3E,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,IAAI,CAAC,IAAI,EAAE,aAAc,KAAe,CAAC,OAAO,EAAE,CAAC,CAAC;YAC5F,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAKD,kBAAkB,CAAC,UAAkB;QACnC,OAAO,UAAU,KAAK,MAAM,CAAC;IAC/B,CAAC;CACF;AA7CD,wCA6CC;AASD,MAAa,YAAY;IAKvB,YACE,KAAkB,EAClB,KAAkB;QANZ,UAAK,GAA2B,IAAI,GAAG,EAAE,CAAC;QAC1C,UAAK,GAA6B,IAAI,GAAG,EAAE,CAAC;QAC5C,WAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QAO3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAChC,CAAC;QAGD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,MAAM,IAAI,8BAAqB,CAAC,0BAA0B,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACzE,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC7B,MAAM,IAAI,8BAAqB,CAAC,0BAA0B,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YACvE,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAChC,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAKO,aAAa;QAEnB,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACvD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,KAAK,CACjC,CAAC;QACF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,8BAAqB,CAAC,yCAAyC,CAAC,CAAC;QAC7E,CAAC;QAGD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACrD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,GAAG,CAC/B,CAAC;QACF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,8BAAqB,CAAC,uCAAuC,CAAC,CAAC;QAC3E,CAAC;QAGD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YACvC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,QAAQ,CAAC,KAAK;oBACjB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;wBAClB,MAAM,IAAI,8BAAqB,CAC7B,cAAc,IAAI,CAAC,EAAE,oBAAoB,CAC1C,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,KAAK,QAAQ,CAAC,QAAQ;oBACpB,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBACjD,MAAM,IAAI,8BAAqB,CAC7B,iBAAiB,IAAI,CAAC,EAAE,qBAAqB,CAC9C,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,KAAK,QAAQ,CAAC,SAAS;oBACrB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;wBACrB,MAAM,IAAI,8BAAqB,CAC7B,kBAAkB,IAAI,CAAC,EAAE,gCAAgC,CAC1D,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,KAAK,QAAQ,CAAC,SAAS;oBACrB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;wBACpB,MAAM,IAAI,8BAAqB,CAC7B,kBAAkB,IAAI,CAAC,EAAE,+BAA+B,CACzD,CAAC;oBACJ,CAAC;oBACD,MAAM;gBACR,KAAK,QAAQ,CAAC,SAAS;oBACrB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;wBACtB,MAAM,IAAI,8BAAqB,CAC7B,kBAAkB,IAAI,CAAC,EAAE,iCAAiC,CAC3D,CAAC;oBACJ,CAAC;oBACD,MAAM;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,OAAO,CACX,KAAa,EACb,MAAqB,EACrB,MAAoB;QAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAiB;YAC5B,KAAK;YACL,aAAa,EAAE,KAAK;YACpB,WAAW,EAAE,IAAI,GAAG,EAAE;YACtB,UAAU,EAAE,IAAI,GAAG,EAAE;YACrB,eAAe,EAAE,IAAI,GAAG,EAAE;YAC1B,aAAa,EAAE,EAAE;YACjB,SAAS;YACT,MAAM;SACP,CAAC;QAEF,IAAI,CAAC;YAEH,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACvD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,KAAK,CACjC,CAAC;YAGF,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACrD,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,OAAO;gBACL,MAAM,EAAE,OAAO,CAAC,aAAa;gBAC7B,OAAO,EAAE,IAAI;gBACb,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,aAAa,EAAE,OAAO,CAAC,aAAa;gBACpC,QAAQ;gBACR,QAAQ,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;aAC1C,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,OAAO;gBACL,MAAM,EAAE,OAAO,CAAC,aAAa;gBAC7B,OAAO,EAAE,KAAK;gBACd,WAAW,EAAE,OAAO,CAAC,WAAW;gBAChC,aAAa,EAAE,OAAO,CAAC,aAAa;gBACpC,QAAQ;gBACR,MAAM,EAAE,CAAC,KAAc,CAAC;gBACxB,QAAQ,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;aAC1C,CAAC;QACJ,CAAC;IACH,CAAC;IAKO,KAAK,CAAC,WAAW,CACvB,IAAe,EACf,OAAqB,EACrB,MAAqB;QAGrB,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QAGD,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;QAE/D,IAAI,MAAM,GAAW,OAAO,CAAC,aAAa,CAAC;QAE3C,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,QAAQ,CAAC,KAAK;gBAEjB,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;gBACvB,MAAM;YAER,KAAK,QAAQ,CAAC,GAAG;gBAEf,OAAO;YAET,KAAK,QAAQ,CAAC,KAAK;gBACjB,MAAM,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;gBAC5D,MAAM;YAER,KAAK,QAAQ,CAAC,QAAQ;gBACpB,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;gBAC/D,MAAM;YAER,KAAK,QAAQ,CAAC,SAAS;gBACrB,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBACxD,MAAM;YAER,KAAK,QAAQ,CAAC,SAAS;gBACrB,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;gBAChE,OAAO;YAET,KAAK,QAAQ,CAAC,SAAS;gBACrB,MAAM,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAClD,MAAM;YAER,KAAK,QAAQ,CAAC,IAAI;gBAChB,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;gBAC3D,MAAM;QACV,CAAC;QAGD,OAAO,CAAC,aAAa,GAAG,MAAM,CAAC;QAC/B,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;YAC/B,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,CAAC;SACZ,CAAC,CAAC;QAGH,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACrD,CAAC;IAKO,KAAK,CAAC,gBAAgB,CAC5B,IAAe,EACf,OAAqB,EACrB,MAAqB;QAErB,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,8BAAqB,CAAC,oBAAoB,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QAEjE,OAAO,MAAM,CAAC;IAChB,CAAC;IAKO,KAAK,CAAC,mBAAmB,CAC/B,IAAe,EACf,OAAqB,EACrB,MAAqB;QAErB,MAAM,UAAU,GAAG,IAAI,CAAC,QAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;YAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,MAAM,IAAI,8BAAqB,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC;YAC5D,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,wBAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,OAAO,GAAsB,EAAE,CAAC;QAGtC,MAAM,OAAO,CAAC,GAAG,CACf,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE;YAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;YACvE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;YAEjE,MAAM,UAAU,GAAoB;gBAClC,OAAO,EAAE,KAAK,CAAC,EAAE;gBACjB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,IAAI;gBACb,QAAQ,EAAE,CAAC;aACZ,CAAC;YAEF,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEzB,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,CAAC,CACJ,CAAC;QAGF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC;IAKO,KAAK,CAAC,oBAAoB,CAChC,IAAe,EACf,OAAqB;QAGrB,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC,UAAW,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC;IAKO,KAAK,CAAC,oBAAoB,CAChC,IAAe,EACf,OAAqB,EACrB,MAAqB;QAErB,MAAM,eAAe,GAAG,IAAI,CAAC,SAAU,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAGxE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QAC5C,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC;QACvD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,OAAO,CAAC,CAAC;QAEzD,MAAM,UAAU,GAAG,eAAe;YAChC,CAAC,CAAC,QAAQ,EAAE,EAAE;YACd,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC;QAElB,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC5C,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC,aAAa,CAAC;IAC/B,CAAC;IAKO,oBAAoB,CAC1B,IAAe,EACf,OAAqB;QAErB,OAAO,IAAI,CAAC,WAAY,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;IAKO,KAAK,CAAC,eAAe,CAC3B,IAAe,EACf,OAAqB,EACrB,MAAqB;QAErB,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC;QAC/C,IAAI,SAAS,GAAG,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QAE1D,OAAO,SAAS,GAAG,aAAa,EAAE,CAAC;YACjC,SAAS,EAAE,CAAC;YACZ,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;YAGhD,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YAGnD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE,CAAC;gBACpE,MAAM;YACR,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC,aAAa,CAAC;IAC/B,CAAC;IAKO,KAAK,CAAC,gBAAgB,CAC5B,IAAe,EACf,OAAqB,EACrB,MAAqB;QAErB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QAE5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAEzB,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,EAAE,CAAC;gBACtE,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzC,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;IAKO,WAAW,CACjB,KAAkB,EAClB,KAAa,EACb,OAAqB;QAErB,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,cAAc,IAAI,SAAS,CAAC;QAExD,OAAO,QAAQ;aACZ,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC;aACzB,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;aAC5E,OAAO,CACN,WAAW,EACX,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;aACtC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;aAClD,IAAI,CAAC,IAAI,CAAC,CACd,CAAC;IACN,CAAC;IAKD,SAAS;QACP,MAAM,KAAK,GAAa,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;QAEjD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YACvC,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;YAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAC5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnD,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAKD,SAAS;QACP,MAAM,KAAK,GAAa,CAAC,UAAU,CAAC,CAAC;QAGrC,KAAK,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;QACvE,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;QACrE,KAAK,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;QACvE,KAAK,CAAC,IAAI,CAAC,8DAA8D,CAAC,CAAC;QAG3E,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YACvC,IAAI,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;YAClC,IAAI,KAAK,GAAG,EAAE,CAAC;YAEf,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,QAAQ,CAAC,KAAK;oBACjB,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;oBAC9B,KAAK,GAAG,UAAU,CAAC;oBACnB,MAAM;gBACR,KAAK,QAAQ,CAAC,GAAG;oBACf,KAAK,GAAG,IAAI,GAAG,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;oBAC9B,KAAK,GAAG,QAAQ,CAAC;oBACjB,MAAM;gBACR,KAAK,QAAQ,CAAC,SAAS;oBACrB,KAAK,GAAG,GAAG,GAAG,IAAI,CAAC,EAAE,GAAG,GAAG,CAAC;oBAC5B,KAAK,GAAG,cAAc,CAAC;oBACvB,MAAM;gBACR,KAAK,QAAQ,CAAC,KAAK;oBACjB,KAAK,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC;oBAC9C,KAAK,GAAG,UAAU,CAAC;oBACnB,MAAM;YACV,CAAC;YAED,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,EAAE,GAAG,KAAK,GAAG,KAAK,EAAE,CAAC,CAAC;QAC7C,CAAC;QAGD,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YACjD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC;gBACrD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;gBACzC,KAAK,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,KAAK,IAAI,KAAK,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AApdD,oCAodC;AASD,MAAa,YAAY;IAAzB;QACU,UAAK,GAAgB,EAAE,CAAC;QACxB,UAAK,GAAgB,EAAE,CAAC;IAmDlC,CAAC;IAjDC,MAAM,CAAC,MAAM;QACX,OAAO,IAAI,YAAY,EAAE,CAAC;IAC5B,CAAC;IAKD,OAAO,CACL,EAAU,EACV,IAAc,EACd,MAAgD;QAEhD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAKD,OAAO,CACL,IAAY,EACZ,EAAU,EACV,MAAgD;QAEhD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAKD,kBAAkB,CAAC,MAAuB;QACxC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,EAAE;YACzC,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,KAAK,EAAE,MAAM;SACd,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,SAAS,EAAE;YAC1C,SAAS,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,GAAG,CAAC;YAC1D,KAAK,EAAE,OAAO;SACf,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAKD,KAAK;QACH,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC;CACF;AArDD,oCAqDC"}
@@ -0,0 +1,28 @@
1
+ export * from './types';
2
+ export * from './config';
3
+ export * from './errors';
4
+ export * from './logger';
5
+ export * from './retry';
6
+ export * from './worker-pool';
7
+ export * from './models';
8
+ export { MessageBus, RoleBuilder, AgentBuilder, StepBuilder, WorkflowConfigBuilder, DefaultWorkflowExecutor, } from './society';
9
+ export { Society, SocietyPatterns, AggregationStrategies, FluentRoleBuilder, FluentAgentBuilder, FluentStepBuilder, FluentPipelineBuilder, createRole, createAgent, } from './builder';
10
+ export type { PipelineConfig, PipelinePattern } from './builder';
11
+ export { MiddlewareChain, ComposedMiddleware, MiddlewareWrappedModel, Middlewares, StepMiddlewares, InMemoryMetricsCollector, } from './middleware';
12
+ export type { Middleware, MiddlewareFn, MiddlewareContext, MiddlewareResult, NextFunction, MetricsCollector, StepMiddleware, StepMiddlewareFn, StepMiddlewareContext, } from './middleware';
13
+ export { StrategyBuilder, Strategies, createStrategy, chainStrategies, parallelStrategies, } from './strategies';
14
+ export type { AggregationStrategy, AggregationResult, StrategyFactory, } from './strategies';
15
+ export { PipelineBuilder as PipelinePatternBuilder, Pipelines, composePipelines, parallelPipelines, } from './pipeline';
16
+ export type { Pipeline, PipelineContext, PipelineResult, PipelineStep, } from './pipeline';
17
+ export { createContextToken, isContextToken, ContextProvider, ContextProviderBuilder, ContextScope, CommonContexts, ContextStore, ContextMap, selectContext, fromObject, toObject, mergeContexts, AgentContextInjector, ContextAwarePromptBuilder, } from './context';
18
+ export type { ContextToken, IContextProvider, IMutableContextProvider, } from './context';
19
+ export { SocietyEventEmitter, FilteredEventEmitter, ProgressTracker, EventLogger, EventAggregator, createEventEmitter, createProgressTracker, createEventLogger, } from './events';
20
+ export type { BaseEvent, SocietyEvent, SocietyEventMap, WorkflowStartEvent, WorkflowCompleteEvent, WorkflowErrorEvent, StepStartEvent, StepCompleteEvent, StepErrorEvent, StepSkippedEvent, AgentStartEvent, AgentCompleteEvent, AgentErrorEvent, AgentRetryEvent, ProgressEvent, MessageSentEvent, MessageReceivedEvent, DebugEvent, CustomEvent, EventHandler, EventFilter, EventTransformer, EventSummary, } from './events';
21
+ export { SocietyGraph, GraphBuilder, NodeType, GraphNode, GraphEdge, ConditionalEdge, GraphContext, GraphResult, } from './graph';
22
+ export { ToolRegistry, ToolExecutor, ToolBuilder, BuiltInTools, Tool, ToolCall, ToolResult, ToolContext, ToolParameterSchema, } from './tools';
23
+ export { MemorySystem, MemoryBuilder, ShortTermMemory, LongTermMemory, EntityMemory, MemoryEntry, MemoryQuery, MemoryRetrievalResult, VectorProvider, Entity, } from './memory';
24
+ export { StructuredOutputValidator, StructuredOutputBuilder, validateJSON, createSchema, JSONSchema, ValidationError, ValidationResult, } from './validation';
25
+ export { MetricsTracker, MetricsBuilder, TokenCounter, PerformanceProfiler, CommonCostConfigs, TokenMetrics, ExecutionMetrics, CostMetrics, MetricsSnapshot, AggregatedMetrics, CostConfig, OTelTrace, } from './metrics';
26
+ export * from './graph';
27
+ export * from './patterns';
28
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAqCA,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AAMzB,cAAc,UAAU,CAAC;AAMzB,cAAc,UAAU,CAAC;AAMzB,cAAc,SAAS,CAAC;AAMxB,cAAc,eAAe,CAAC;AAM9B,cAAc,UAAU,CAAC;AAOzB,OAAO,EACL,UAAU,EACV,WAAW,EACX,YAAY,EACZ,WAAW,EACX,qBAAqB,EACrB,uBAAuB,GACxB,MAAM,WAAW,CAAC;AAMnB,OAAO,EAEL,OAAO,EACP,eAAe,EACf,qBAAqB,EAErB,iBAAiB,EACjB,kBAAkB,EAClB,iBAAiB,EACjB,qBAAqB,EAErB,UAAU,EACV,WAAW,GACZ,MAAM,WAAW,CAAC;AAMnB,YAAY,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAMjE,OAAO,EAEL,eAAe,EACf,kBAAkB,EAClB,sBAAsB,EAEtB,WAAW,EAEX,eAAe,EAEf,wBAAwB,GACzB,MAAM,cAAc,CAAC;AAGtB,YAAY,EACV,UAAU,EACV,YAAY,EACZ,iBAAiB,EACjB,gBAAgB,EAChB,YAAY,EACZ,gBAAgB,EAChB,cAAc,EACd,gBAAgB,EAChB,qBAAqB,GACtB,MAAM,cAAc,CAAC;AAMtB,OAAO,EAEL,eAAe,EAEf,UAAU,EAEV,cAAc,EACd,eAAe,EACf,kBAAkB,GACnB,MAAM,cAAc,CAAC;AAGtB,YAAY,EACV,mBAAmB,EACnB,iBAAiB,EACjB,eAAe,GAChB,MAAM,cAAc,CAAC;AAMtB,OAAO,EAEL,eAAe,IAAI,sBAAsB,EAEzC,SAAS,EAET,gBAAgB,EAChB,iBAAiB,GAClB,MAAM,YAAY,CAAC;AAGpB,YAAY,EACV,QAAQ,EACR,eAAe,EACf,cAAc,EACd,YAAY,GACb,MAAM,YAAY,CAAC;AAMpB,OAAO,EAEL,kBAAkB,EAClB,cAAc,EAEd,eAAe,EACf,sBAAsB,EAEtB,YAAY,EAEZ,cAAc,EAEd,YAAY,EACZ,UAAU,EACV,aAAa,EACb,UAAU,EACV,QAAQ,EACR,aAAa,EAEb,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,WAAW,CAAC;AAGnB,YAAY,EACV,YAAY,EACZ,gBAAgB,EAChB,uBAAuB,GACxB,MAAM,WAAW,CAAC;AAMnB,OAAO,EAEL,mBAAmB,EACnB,oBAAoB,EAEpB,eAAe,EAEf,WAAW,EAEX,eAAe,EAEf,kBAAkB,EAClB,qBAAqB,EACrB,iBAAiB,GAClB,MAAM,UAAU,CAAC;AAGlB,YAAY,EAEV,SAAS,EACT,YAAY,EACZ,eAAe,EAEf,kBAAkB,EAClB,qBAAqB,EACrB,kBAAkB,EAElB,cAAc,EACd,iBAAiB,EACjB,cAAc,EACd,gBAAgB,EAEhB,eAAe,EACf,kBAAkB,EAClB,eAAe,EACf,eAAe,EAEf,aAAa,EACb,gBAAgB,EAChB,oBAAoB,EACpB,UAAU,EACV,WAAW,EAEX,YAAY,EACZ,WAAW,EACX,gBAAgB,EAEhB,YAAY,GACb,MAAM,UAAU,CAAC;AAMlB,OAAO,EAEL,YAAY,EACZ,YAAY,EAEZ,QAAQ,EACR,SAAS,EACT,SAAS,EACT,eAAe,EACf,YAAY,EACZ,WAAW,GACZ,MAAM,SAAS,CAAC;AAMjB,OAAO,EAEL,YAAY,EACZ,YAAY,EACZ,WAAW,EAEX,YAAY,EAEZ,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,WAAW,EACX,mBAAmB,GACpB,MAAM,SAAS,CAAC;AAMjB,OAAO,EAEL,YAAY,EACZ,aAAa,EACb,eAAe,EACf,cAAc,EACd,YAAY,EAEZ,WAAW,EACX,WAAW,EACX,qBAAqB,EACrB,cAAc,EACd,MAAM,GACP,MAAM,UAAU,CAAC;AAMlB,OAAO,EAEL,yBAAyB,EACzB,uBAAuB,EAEvB,YAAY,EACZ,YAAY,EAEZ,UAAU,EACV,eAAe,EACf,gBAAgB,GACjB,MAAM,cAAc,CAAC;AAMtB,OAAO,EAEL,cAAc,EACd,cAAc,EAEd,YAAY,EACZ,mBAAmB,EAEnB,iBAAiB,EAEjB,YAAY,EACZ,gBAAgB,EAChB,WAAW,EACX,eAAe,EACf,iBAAiB,EACjB,UAAU,EACV,SAAS,GACV,MAAM,WAAW,CAAC;AACnB,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC"}