graphai 0.3.2 → 0.4.0

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 (51) hide show
  1. package/README.md +3 -3
  2. package/lib/experimental_agents/array_agents/index.d.ts +1 -3
  3. package/lib/experimental_agents/array_agents/index.js +15 -7
  4. package/lib/experimental_agents/array_agents/vanilla.d.ts +4 -0
  5. package/lib/experimental_agents/array_agents/vanilla.js +12 -0
  6. package/lib/experimental_agents/data_agents/copy_agent.d.ts +2 -0
  7. package/lib/experimental_agents/data_agents/copy_agent.js +2 -0
  8. package/lib/experimental_agents/data_agents/index.d.ts +3 -4
  9. package/lib/experimental_agents/data_agents/index.js +21 -9
  10. package/lib/experimental_agents/data_agents/packages.d.ts +2 -1
  11. package/lib/experimental_agents/data_agents/packages.js +3 -1
  12. package/lib/experimental_agents/data_agents/vanilla.d.ts +5 -0
  13. package/lib/experimental_agents/data_agents/vanilla.js +14 -0
  14. package/lib/experimental_agents/graph_agents/index.d.ts +1 -2
  15. package/lib/experimental_agents/graph_agents/index.js +15 -5
  16. package/lib/experimental_agents/graph_agents/vanilla.d.ts +3 -0
  17. package/lib/experimental_agents/graph_agents/vanilla.js +10 -0
  18. package/lib/experimental_agents/index.d.ts +10 -3
  19. package/lib/experimental_agents/index.js +17 -5
  20. package/lib/experimental_agents/llm_agents/index.d.ts +5 -4
  21. package/lib/experimental_agents/llm_agents/index.js +11 -17
  22. package/lib/experimental_agents/matrix_agents/index.d.ts +1 -2
  23. package/lib/experimental_agents/matrix_agents/index.js +15 -5
  24. package/lib/experimental_agents/matrix_agents/vanilla.d.ts +3 -0
  25. package/lib/experimental_agents/matrix_agents/vanilla.js +10 -0
  26. package/lib/experimental_agents/service_agents/index.d.ts +3 -2
  27. package/lib/experimental_agents/service_agents/index.js +8 -5
  28. package/lib/experimental_agents/sleeper_agents/index.d.ts +3 -2
  29. package/lib/experimental_agents/sleeper_agents/index.js +7 -4
  30. package/lib/experimental_agents/string_agents/index.d.ts +1 -2
  31. package/lib/experimental_agents/string_agents/index.js +15 -5
  32. package/lib/experimental_agents/string_agents/vanilla.d.ts +3 -0
  33. package/lib/experimental_agents/string_agents/vanilla.js +10 -0
  34. package/lib/experimental_agents/test_agents/index.d.ts +1 -7
  35. package/lib/experimental_agents/test_agents/index.js +15 -15
  36. package/lib/experimental_agents/test_agents/vanilla.d.ts +8 -0
  37. package/lib/experimental_agents/test_agents/vanilla.js +20 -0
  38. package/lib/experimental_agents/vanilla.d.ts +6 -6
  39. package/lib/experimental_agents/vanilla.js +6 -6
  40. package/lib/graphai.d.ts +5 -5
  41. package/lib/graphai.js +6 -6
  42. package/lib/index.d.ts +1 -1
  43. package/lib/node.js +2 -2
  44. package/lib/type.d.ts +2 -2
  45. package/lib/utils/test_agents.d.ts +2 -2
  46. package/lib/utils/test_agents.js +2 -2
  47. package/lib/utils/test_utils.d.ts +28 -0
  48. package/lib/utils/test_utils.js +24 -1
  49. package/package.json +1 -1
  50. package/lib/experimental_agents/packages.d.ts +0 -13
  51. package/lib/experimental_agents/packages.js +0 -35
package/README.md CHANGED
@@ -120,7 +120,7 @@ An *agent function* is a TypeScript function, which implements a particular *age
120
120
  There are additional optional parameters for developers of nested agents and agent filters.
121
121
 
122
122
  - *graphData*: an optional GraphData (for nested agents)
123
- - *agents*: AgentFunctionDictonary (for nested agents)
123
+ - *agents*: AgentFunctionInfoDictonary (for nested agents)
124
124
  - *taskManager*: TaskManager (for nested agents)
125
125
  - *log*: TransactionLog[] (for nested agents)
126
126
  - *filterParams*: agent filter parameters (for agent filters)
@@ -379,11 +379,11 @@ Negative priority values are allowed, enabling you to fine-tune the execution or
379
379
 
380
380
  ## GraphAI class
381
381
 
382
- ### ```constructor(data: GraphData, callbackDictonary: AgentFunctionDictonary)```
382
+ ### ```constructor(data: GraphData, callbackDictonary: AgentFunctionInfoDictonary)```
383
383
  Initializes a new instance of the GraphAI object with the specified graph data and a dictionary of callback functions.
384
384
 
385
385
  - ```data: GraphData```: The graph data including nodes and optional concurrency limit.
386
- - ```callbackDictonary: AgentFunctionDictonary | AgentFunction<any, any, any>```: A dictionary mapping agent IDs to their respective callback functions, or a single default callback function to be used for all nodes.
386
+ - ```callbackDictonary: AgentFunctionInfoDictonary```: A dictionary mapping agent IDs to their respective callback functions info to be used for all nodes.
387
387
 
388
388
  ### ```async run(): Promise<ResultDataDictonary<ResultData>>```
389
389
  Executes the graph asynchronously, starting with nodes that have no dependencies or whose dependencies have been met. The method continues to execute nodes as their dependencies are satisfied until all nodes have been executed or an error occurs.
@@ -1,3 +1 @@
1
- export { pushAgent } from "../../experimental_agents/array_agents/push_agent";
2
- export { popAgent } from "../../experimental_agents/array_agents/pop_agent";
3
- export { shiftAgent } from "../../experimental_agents/array_agents/shift_agent";
1
+ export * from "./vanilla";
@@ -1,9 +1,17 @@
1
1
  "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
2
16
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.shiftAgent = exports.popAgent = exports.pushAgent = void 0;
4
- var push_agent_1 = require("../../experimental_agents/array_agents/push_agent");
5
- Object.defineProperty(exports, "pushAgent", { enumerable: true, get: function () { return push_agent_1.pushAgent; } });
6
- var pop_agent_1 = require("../../experimental_agents/array_agents/pop_agent");
7
- Object.defineProperty(exports, "popAgent", { enumerable: true, get: function () { return pop_agent_1.popAgent; } });
8
- var shift_agent_1 = require("../../experimental_agents/array_agents/shift_agent");
9
- Object.defineProperty(exports, "shiftAgent", { enumerable: true, get: function () { return shift_agent_1.shiftAgent; } });
17
+ __exportStar(require("./vanilla"), exports);
@@ -0,0 +1,4 @@
1
+ import pushAgent from "../../experimental_agents/array_agents/push_agent";
2
+ import popAgent from "../../experimental_agents/array_agents/pop_agent";
3
+ import shiftAgent from "../../experimental_agents/array_agents/shift_agent";
4
+ export { pushAgent, popAgent, shiftAgent };
@@ -0,0 +1,12 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.shiftAgent = exports.popAgent = exports.pushAgent = void 0;
7
+ const push_agent_1 = __importDefault(require("../../experimental_agents/array_agents/push_agent"));
8
+ exports.pushAgent = push_agent_1.default;
9
+ const pop_agent_1 = __importDefault(require("../../experimental_agents/array_agents/pop_agent"));
10
+ exports.popAgent = pop_agent_1.default;
11
+ const shift_agent_1 = __importDefault(require("../../experimental_agents/array_agents/shift_agent"));
12
+ exports.shiftAgent = shift_agent_1.default;
@@ -9,12 +9,14 @@ declare const copyAgentInfo: {
9
9
  color: string;
10
10
  model: string;
11
11
  }[];
12
+ params: {};
12
13
  result: {
13
14
  color: string;
14
15
  model: string;
15
16
  };
16
17
  } | {
17
18
  inputs: string[];
19
+ params: {};
18
20
  result: string;
19
21
  })[];
20
22
  description: string;
@@ -13,10 +13,12 @@ const copyAgentInfo = {
13
13
  samples: [
14
14
  {
15
15
  inputs: [{ color: "red", model: "Model 3" }],
16
+ params: {},
16
17
  result: { color: "red", model: "Model 3" },
17
18
  },
18
19
  {
19
20
  inputs: ["Hello World"],
21
+ params: {},
20
22
  result: "Hello World",
21
23
  },
22
24
  ],
@@ -1,4 +1,3 @@
1
- export { totalAgent } from "../../experimental_agents/data_agents/total_agent";
2
- export { dataSumTemplateAgent } from "../../experimental_agents/data_agents/data_sum_template_agent";
3
- export { propertyFilterAgent } from "../../experimental_agents/data_agents/property_filter_agent";
4
- export { copyAgent } from "../../experimental_agents/data_agents/copy_agent";
1
+ export * from "./vanilla";
2
+ import dataObjectMergeTemplateAgent from "./data_object_merge_template_agent";
3
+ export { dataObjectMergeTemplateAgent };
@@ -1,11 +1,23 @@
1
1
  "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
16
+ var __importDefault = (this && this.__importDefault) || function (mod) {
17
+ return (mod && mod.__esModule) ? mod : { "default": mod };
18
+ };
2
19
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.copyAgent = exports.propertyFilterAgent = exports.dataSumTemplateAgent = exports.totalAgent = void 0;
4
- var total_agent_1 = require("../../experimental_agents/data_agents/total_agent");
5
- Object.defineProperty(exports, "totalAgent", { enumerable: true, get: function () { return total_agent_1.totalAgent; } });
6
- var data_sum_template_agent_1 = require("../../experimental_agents/data_agents/data_sum_template_agent");
7
- Object.defineProperty(exports, "dataSumTemplateAgent", { enumerable: true, get: function () { return data_sum_template_agent_1.dataSumTemplateAgent; } });
8
- var property_filter_agent_1 = require("../../experimental_agents/data_agents/property_filter_agent");
9
- Object.defineProperty(exports, "propertyFilterAgent", { enumerable: true, get: function () { return property_filter_agent_1.propertyFilterAgent; } });
10
- var copy_agent_1 = require("../../experimental_agents/data_agents/copy_agent");
11
- Object.defineProperty(exports, "copyAgent", { enumerable: true, get: function () { return copy_agent_1.copyAgent; } });
20
+ exports.dataObjectMergeTemplateAgent = void 0;
21
+ __exportStar(require("./vanilla"), exports);
22
+ const data_object_merge_template_agent_1 = __importDefault(require("./data_object_merge_template_agent"));
23
+ exports.dataObjectMergeTemplateAgent = data_object_merge_template_agent_1.default;
@@ -2,4 +2,5 @@ import totalAgent from "../../experimental_agents/data_agents/total_agent";
2
2
  import dataObjectMergeTemplateAgent from "../../experimental_agents/data_agents/data_object_merge_template_agent";
3
3
  import dataSumTemplateAgent from "../../experimental_agents/data_agents/data_sum_template_agent";
4
4
  import propertyFilterAgent from "../../experimental_agents/data_agents/property_filter_agent";
5
- export { totalAgent, dataObjectMergeTemplateAgent, dataSumTemplateAgent, propertyFilterAgent };
5
+ import copyAgent from "../../experimental_agents/data_agents/copy_agent";
6
+ export { totalAgent, dataObjectMergeTemplateAgent, dataSumTemplateAgent, propertyFilterAgent, copyAgent };
@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
3
3
  return (mod && mod.__esModule) ? mod : { "default": mod };
4
4
  };
5
5
  Object.defineProperty(exports, "__esModule", { value: true });
6
- exports.propertyFilterAgent = exports.dataSumTemplateAgent = exports.dataObjectMergeTemplateAgent = exports.totalAgent = void 0;
6
+ exports.copyAgent = exports.propertyFilterAgent = exports.dataSumTemplateAgent = exports.dataObjectMergeTemplateAgent = exports.totalAgent = void 0;
7
7
  const total_agent_1 = __importDefault(require("../../experimental_agents/data_agents/total_agent"));
8
8
  exports.totalAgent = total_agent_1.default;
9
9
  const data_object_merge_template_agent_1 = __importDefault(require("../../experimental_agents/data_agents/data_object_merge_template_agent"));
@@ -12,3 +12,5 @@ const data_sum_template_agent_1 = __importDefault(require("../../experimental_ag
12
12
  exports.dataSumTemplateAgent = data_sum_template_agent_1.default;
13
13
  const property_filter_agent_1 = __importDefault(require("../../experimental_agents/data_agents/property_filter_agent"));
14
14
  exports.propertyFilterAgent = property_filter_agent_1.default;
15
+ const copy_agent_1 = __importDefault(require("../../experimental_agents/data_agents/copy_agent"));
16
+ exports.copyAgent = copy_agent_1.default;
@@ -0,0 +1,5 @@
1
+ import totalAgent from "../../experimental_agents/data_agents/total_agent";
2
+ import dataSumTemplateAgent from "../../experimental_agents/data_agents/data_sum_template_agent";
3
+ import propertyFilterAgent from "../../experimental_agents/data_agents/property_filter_agent";
4
+ import copyAgent from "../../experimental_agents/data_agents/copy_agent";
5
+ export { totalAgent, dataSumTemplateAgent, propertyFilterAgent, copyAgent };
@@ -0,0 +1,14 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.copyAgent = exports.propertyFilterAgent = exports.dataSumTemplateAgent = exports.totalAgent = void 0;
7
+ const total_agent_1 = __importDefault(require("../../experimental_agents/data_agents/total_agent"));
8
+ exports.totalAgent = total_agent_1.default;
9
+ const data_sum_template_agent_1 = __importDefault(require("../../experimental_agents/data_agents/data_sum_template_agent"));
10
+ exports.dataSumTemplateAgent = data_sum_template_agent_1.default;
11
+ const property_filter_agent_1 = __importDefault(require("../../experimental_agents/data_agents/property_filter_agent"));
12
+ exports.propertyFilterAgent = property_filter_agent_1.default;
13
+ const copy_agent_1 = __importDefault(require("../../experimental_agents/data_agents/copy_agent"));
14
+ exports.copyAgent = copy_agent_1.default;
@@ -1,2 +1 @@
1
- export { nestedAgent } from "../../experimental_agents/graph_agents/nested_agent";
2
- export { mapAgent } from "../../experimental_agents/graph_agents/map_agent";
1
+ export * from "./vanilla";
@@ -1,7 +1,17 @@
1
1
  "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
2
16
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.mapAgent = exports.nestedAgent = void 0;
4
- var nested_agent_1 = require("../../experimental_agents/graph_agents/nested_agent");
5
- Object.defineProperty(exports, "nestedAgent", { enumerable: true, get: function () { return nested_agent_1.nestedAgent; } });
6
- var map_agent_1 = require("../../experimental_agents/graph_agents/map_agent");
7
- Object.defineProperty(exports, "mapAgent", { enumerable: true, get: function () { return map_agent_1.mapAgent; } });
17
+ __exportStar(require("./vanilla"), exports);
@@ -0,0 +1,3 @@
1
+ import nestedAgent from "../../experimental_agents/graph_agents/nested_agent";
2
+ import mapAgent from "../../experimental_agents/graph_agents/map_agent";
3
+ export { nestedAgent, mapAgent };
@@ -0,0 +1,10 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.mapAgent = exports.nestedAgent = void 0;
7
+ const nested_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/nested_agent"));
8
+ exports.nestedAgent = nested_agent_1.default;
9
+ const map_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/map_agent"));
10
+ exports.mapAgent = map_agent_1.default;
@@ -1,6 +1,13 @@
1
- export * from "./vanilla";
2
- export { dataObjectMergeTemplateAgent } from "../experimental_agents/data_agents/data_object_merge_template_agent";
1
+ export * from "./string_agents";
2
+ export * from "./array_agents";
3
+ export * from "./matrix_agents";
4
+ export * from "./test_agents";
5
+ export * from "./graph_agents";
6
+ export * from "./data_agents";
3
7
  export * from "./sleeper_agents";
4
8
  export * from "./llm_agents";
5
9
  export * from "./service_agents";
6
- export * from "./token_agent";
10
+ import functionAgent from "./function_agent";
11
+ import stringEmbeddingsAgent from "./embedding_agent";
12
+ import tokenBoundStringsAgent from "./token_agent";
13
+ export { functionAgent, stringEmbeddingsAgent, tokenBoundStringsAgent };
@@ -13,13 +13,25 @@ var __createBinding = (this && this.__createBinding) || (Object.create ? (functi
13
13
  var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
14
  for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
15
  };
16
+ var __importDefault = (this && this.__importDefault) || function (mod) {
17
+ return (mod && mod.__esModule) ? mod : { "default": mod };
18
+ };
16
19
  Object.defineProperty(exports, "__esModule", { value: true });
17
- exports.dataObjectMergeTemplateAgent = void 0;
18
- __exportStar(require("./vanilla"), exports);
20
+ exports.tokenBoundStringsAgent = exports.stringEmbeddingsAgent = exports.functionAgent = void 0;
21
+ __exportStar(require("./string_agents"), exports);
22
+ __exportStar(require("./array_agents"), exports);
23
+ __exportStar(require("./matrix_agents"), exports);
24
+ __exportStar(require("./test_agents"), exports);
25
+ __exportStar(require("./graph_agents"), exports);
26
+ __exportStar(require("./data_agents"), exports);
19
27
  // Agents that use npm modules will be added here.
20
- var data_object_merge_template_agent_1 = require("../experimental_agents/data_agents/data_object_merge_template_agent");
21
- Object.defineProperty(exports, "dataObjectMergeTemplateAgent", { enumerable: true, get: function () { return data_object_merge_template_agent_1.dataObjectMergeTemplateAgent; } });
22
28
  __exportStar(require("./sleeper_agents"), exports);
23
29
  __exportStar(require("./llm_agents"), exports);
24
30
  __exportStar(require("./service_agents"), exports);
25
- __exportStar(require("./token_agent"), exports);
31
+ // TODO sub folder
32
+ const function_agent_1 = __importDefault(require("./function_agent"));
33
+ exports.functionAgent = function_agent_1.default;
34
+ const embedding_agent_1 = __importDefault(require("./embedding_agent"));
35
+ exports.stringEmbeddingsAgent = embedding_agent_1.default;
36
+ const token_agent_1 = __importDefault(require("./token_agent"));
37
+ exports.tokenBoundStringsAgent = token_agent_1.default;
@@ -1,4 +1,5 @@
1
- export * from "./openai_agent";
2
- export * from "./slashgpt_agent";
3
- export * from "./groq_agent";
4
- export * from "./groq_stream_agent";
1
+ import groqAgent from "../../experimental_agents/llm_agents/groq_agent";
2
+ import groqStreamAgent from "../../experimental_agents/llm_agents/groq_stream_agent";
3
+ import slashGPTAgent from "../../experimental_agents/llm_agents/slashgpt_agent";
4
+ import openAIAgent from "../../experimental_agents/llm_agents/openai_agent";
5
+ export { groqAgent, groqStreamAgent, slashGPTAgent, openAIAgent };
@@ -1,20 +1,14 @@
1
1
  "use strict";
2
- var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
- if (k2 === undefined) k2 = k;
4
- var desc = Object.getOwnPropertyDescriptor(m, k);
5
- if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
- desc = { enumerable: true, get: function() { return m[k]; } };
7
- }
8
- Object.defineProperty(o, k2, desc);
9
- }) : (function(o, m, k, k2) {
10
- if (k2 === undefined) k2 = k;
11
- o[k2] = m[k];
12
- }));
13
- var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
- for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
15
4
  };
16
5
  Object.defineProperty(exports, "__esModule", { value: true });
17
- __exportStar(require("./openai_agent"), exports);
18
- __exportStar(require("./slashgpt_agent"), exports);
19
- __exportStar(require("./groq_agent"), exports);
20
- __exportStar(require("./groq_stream_agent"), exports);
6
+ exports.openAIAgent = exports.slashGPTAgent = exports.groqStreamAgent = exports.groqAgent = void 0;
7
+ const groq_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/groq_agent"));
8
+ exports.groqAgent = groq_agent_1.default;
9
+ const groq_stream_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/groq_stream_agent"));
10
+ exports.groqStreamAgent = groq_stream_agent_1.default;
11
+ const slashgpt_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/slashgpt_agent"));
12
+ exports.slashGPTAgent = slashgpt_agent_1.default;
13
+ const openai_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/openai_agent"));
14
+ exports.openAIAgent = openai_agent_1.default;
@@ -1,2 +1 @@
1
- export { dotProductAgent } from "../../experimental_agents/matrix_agents/dot_product_agent";
2
- export { sortByValuesAgent } from "../../experimental_agents/matrix_agents/sort_by_values_agent";
1
+ export * from "./vanilla";
@@ -1,7 +1,17 @@
1
1
  "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
2
16
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.sortByValuesAgent = exports.dotProductAgent = void 0;
4
- var dot_product_agent_1 = require("../../experimental_agents/matrix_agents/dot_product_agent");
5
- Object.defineProperty(exports, "dotProductAgent", { enumerable: true, get: function () { return dot_product_agent_1.dotProductAgent; } });
6
- var sort_by_values_agent_1 = require("../../experimental_agents/matrix_agents/sort_by_values_agent");
7
- Object.defineProperty(exports, "sortByValuesAgent", { enumerable: true, get: function () { return sort_by_values_agent_1.sortByValuesAgent; } });
17
+ __exportStar(require("./vanilla"), exports);
@@ -0,0 +1,3 @@
1
+ import dotProductAgent from "../../experimental_agents/matrix_agents/dot_product_agent";
2
+ import sortByValuesAgent from "../../experimental_agents/matrix_agents/sort_by_values_agent";
3
+ export { dotProductAgent, sortByValuesAgent };
@@ -0,0 +1,10 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.sortByValuesAgent = exports.dotProductAgent = void 0;
7
+ const dot_product_agent_1 = __importDefault(require("../../experimental_agents/matrix_agents/dot_product_agent"));
8
+ exports.dotProductAgent = dot_product_agent_1.default;
9
+ const sort_by_values_agent_1 = __importDefault(require("../../experimental_agents/matrix_agents/sort_by_values_agent"));
10
+ exports.sortByValuesAgent = sort_by_values_agent_1.default;
@@ -1,2 +1,3 @@
1
- export { wikipediaAgent } from "../../experimental_agents/service_agents/wikipedia";
2
- export { fetchAgent } from "../../experimental_agents/service_agents/fetch_agent";
1
+ import wikipediaAgent from "../../experimental_agents/service_agents/wikipedia";
2
+ import fetchAgent from "../../experimental_agents/service_agents/fetch_agent";
3
+ export { fetchAgent, wikipediaAgent };
@@ -1,7 +1,10 @@
1
1
  "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
2
5
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.fetchAgent = exports.wikipediaAgent = void 0;
4
- var wikipedia_1 = require("../../experimental_agents/service_agents/wikipedia");
5
- Object.defineProperty(exports, "wikipediaAgent", { enumerable: true, get: function () { return wikipedia_1.wikipediaAgent; } });
6
- var fetch_agent_1 = require("../../experimental_agents/service_agents/fetch_agent");
7
- Object.defineProperty(exports, "fetchAgent", { enumerable: true, get: function () { return fetch_agent_1.fetchAgent; } });
6
+ exports.wikipediaAgent = exports.fetchAgent = void 0;
7
+ const wikipedia_1 = __importDefault(require("../../experimental_agents/service_agents/wikipedia"));
8
+ exports.wikipediaAgent = wikipedia_1.default;
9
+ const fetch_agent_1 = __importDefault(require("../../experimental_agents/service_agents/fetch_agent"));
10
+ exports.fetchAgent = fetch_agent_1.default;
@@ -1,2 +1,3 @@
1
- export { sleeperAgent } from "../../experimental_agents/sleeper_agents/sleeper_agent";
2
- export { sleeperAgentDebug } from "../../experimental_agents/sleeper_agents/sleeper_agent_debug";
1
+ import sleeperAgent from "../../experimental_agents/sleeper_agents/sleeper_agent";
2
+ import sleeperAgentDebug from "../../experimental_agents/sleeper_agents/sleeper_agent_debug";
3
+ export { sleeperAgent, sleeperAgentDebug };
@@ -1,7 +1,10 @@
1
1
  "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
2
5
  Object.defineProperty(exports, "__esModule", { value: true });
3
6
  exports.sleeperAgentDebug = exports.sleeperAgent = void 0;
4
- var sleeper_agent_1 = require("../../experimental_agents/sleeper_agents/sleeper_agent");
5
- Object.defineProperty(exports, "sleeperAgent", { enumerable: true, get: function () { return sleeper_agent_1.sleeperAgent; } });
6
- var sleeper_agent_debug_1 = require("../../experimental_agents/sleeper_agents/sleeper_agent_debug");
7
- Object.defineProperty(exports, "sleeperAgentDebug", { enumerable: true, get: function () { return sleeper_agent_debug_1.sleeperAgentDebug; } });
7
+ const sleeper_agent_1 = __importDefault(require("../../experimental_agents/sleeper_agents/sleeper_agent"));
8
+ exports.sleeperAgent = sleeper_agent_1.default;
9
+ const sleeper_agent_debug_1 = __importDefault(require("../../experimental_agents/sleeper_agents/sleeper_agent_debug"));
10
+ exports.sleeperAgentDebug = sleeper_agent_debug_1.default;
@@ -1,2 +1 @@
1
- export { stringSplitterAgent } from "../../experimental_agents/string_agents/string_splitter_agent";
2
- export { stringTemplateAgent } from "../../experimental_agents/string_agents/string_template_agent";
1
+ export * from "./vanilla";
@@ -1,7 +1,17 @@
1
1
  "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
2
16
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.stringTemplateAgent = exports.stringSplitterAgent = void 0;
4
- var string_splitter_agent_1 = require("../../experimental_agents/string_agents/string_splitter_agent");
5
- Object.defineProperty(exports, "stringSplitterAgent", { enumerable: true, get: function () { return string_splitter_agent_1.stringSplitterAgent; } });
6
- var string_template_agent_1 = require("../../experimental_agents/string_agents/string_template_agent");
7
- Object.defineProperty(exports, "stringTemplateAgent", { enumerable: true, get: function () { return string_template_agent_1.stringTemplateAgent; } });
17
+ __exportStar(require("./vanilla"), exports);
@@ -0,0 +1,3 @@
1
+ import stringSplitterAgent from "../../experimental_agents/string_agents/string_splitter_agent";
2
+ import stringTemplateAgent from "../../experimental_agents/string_agents/string_template_agent";
3
+ export { stringSplitterAgent, stringTemplateAgent };
@@ -0,0 +1,10 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.stringTemplateAgent = exports.stringSplitterAgent = void 0;
7
+ const string_splitter_agent_1 = __importDefault(require("../../experimental_agents/string_agents/string_splitter_agent"));
8
+ exports.stringSplitterAgent = string_splitter_agent_1.default;
9
+ const string_template_agent_1 = __importDefault(require("../../experimental_agents/string_agents/string_template_agent"));
10
+ exports.stringTemplateAgent = string_template_agent_1.default;
@@ -1,7 +1 @@
1
- export { bypassAgent } from "../../experimental_agents/test_agents/bypass_agent";
2
- export { echoAgent } from "../../experimental_agents/test_agents/echo_agent";
3
- export { copyMessageAgent } from "../../experimental_agents/test_agents/copy_message_agent";
4
- export { mergeNodeIdAgent } from "../../experimental_agents/test_agents/merge_node_id_agent";
5
- export { countingAgent } from "../../experimental_agents/test_agents/counting_agent";
6
- export { copy2ArrayAgent } from "../../experimental_agents/test_agents/copy2array_agent";
7
- export { streamMockAgent } from "../../experimental_agents/test_agents/stream_mock_agent";
1
+ export * from "./vanilla";
@@ -1,17 +1,17 @@
1
1
  "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
2
16
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.streamMockAgent = exports.copy2ArrayAgent = exports.countingAgent = exports.mergeNodeIdAgent = exports.copyMessageAgent = exports.echoAgent = exports.bypassAgent = void 0;
4
- var bypass_agent_1 = require("../../experimental_agents/test_agents/bypass_agent");
5
- Object.defineProperty(exports, "bypassAgent", { enumerable: true, get: function () { return bypass_agent_1.bypassAgent; } });
6
- var echo_agent_1 = require("../../experimental_agents/test_agents/echo_agent");
7
- Object.defineProperty(exports, "echoAgent", { enumerable: true, get: function () { return echo_agent_1.echoAgent; } });
8
- var copy_message_agent_1 = require("../../experimental_agents/test_agents/copy_message_agent");
9
- Object.defineProperty(exports, "copyMessageAgent", { enumerable: true, get: function () { return copy_message_agent_1.copyMessageAgent; } });
10
- var merge_node_id_agent_1 = require("../../experimental_agents/test_agents/merge_node_id_agent");
11
- Object.defineProperty(exports, "mergeNodeIdAgent", { enumerable: true, get: function () { return merge_node_id_agent_1.mergeNodeIdAgent; } });
12
- var counting_agent_1 = require("../../experimental_agents/test_agents/counting_agent");
13
- Object.defineProperty(exports, "countingAgent", { enumerable: true, get: function () { return counting_agent_1.countingAgent; } });
14
- var copy2array_agent_1 = require("../../experimental_agents/test_agents/copy2array_agent");
15
- Object.defineProperty(exports, "copy2ArrayAgent", { enumerable: true, get: function () { return copy2array_agent_1.copy2ArrayAgent; } });
16
- var stream_mock_agent_1 = require("../../experimental_agents/test_agents/stream_mock_agent");
17
- Object.defineProperty(exports, "streamMockAgent", { enumerable: true, get: function () { return stream_mock_agent_1.streamMockAgent; } });
17
+ __exportStar(require("./vanilla"), exports);
@@ -0,0 +1,8 @@
1
+ import echoAgent from "../../experimental_agents/test_agents/echo_agent";
2
+ import bypassAgent from "../../experimental_agents/test_agents/bypass_agent";
3
+ import countingAgent from "../../experimental_agents/test_agents/counting_agent";
4
+ import copyMessageAgent from "../../experimental_agents/test_agents/copy_message_agent";
5
+ import copy2ArrayAgent from "../../experimental_agents/test_agents/copy2array_agent";
6
+ import mergeNodeIdAgent from "../../experimental_agents/test_agents/merge_node_id_agent";
7
+ import streamMockAgent from "../../experimental_agents/test_agents/stream_mock_agent";
8
+ export { echoAgent, bypassAgent, countingAgent, copyMessageAgent, copy2ArrayAgent, mergeNodeIdAgent, streamMockAgent };
@@ -0,0 +1,20 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.streamMockAgent = exports.mergeNodeIdAgent = exports.copy2ArrayAgent = exports.copyMessageAgent = exports.countingAgent = exports.bypassAgent = exports.echoAgent = void 0;
7
+ const echo_agent_1 = __importDefault(require("../../experimental_agents/test_agents/echo_agent"));
8
+ exports.echoAgent = echo_agent_1.default;
9
+ const bypass_agent_1 = __importDefault(require("../../experimental_agents/test_agents/bypass_agent"));
10
+ exports.bypassAgent = bypass_agent_1.default;
11
+ const counting_agent_1 = __importDefault(require("../../experimental_agents/test_agents/counting_agent"));
12
+ exports.countingAgent = counting_agent_1.default;
13
+ const copy_message_agent_1 = __importDefault(require("../../experimental_agents/test_agents/copy_message_agent"));
14
+ exports.copyMessageAgent = copy_message_agent_1.default;
15
+ const copy2array_agent_1 = __importDefault(require("../../experimental_agents/test_agents/copy2array_agent"));
16
+ exports.copy2ArrayAgent = copy2array_agent_1.default;
17
+ const merge_node_id_agent_1 = __importDefault(require("../../experimental_agents/test_agents/merge_node_id_agent"));
18
+ exports.mergeNodeIdAgent = merge_node_id_agent_1.default;
19
+ const stream_mock_agent_1 = __importDefault(require("../../experimental_agents/test_agents/stream_mock_agent"));
20
+ exports.streamMockAgent = stream_mock_agent_1.default;
@@ -1,8 +1,8 @@
1
- export * from "./string_agents";
2
- export * from "./array_agents";
3
- export * from "./matrix_agents";
4
- export * from "./test_agents";
5
- export * from "./graph_agents";
1
+ export * from "./string_agents/vanilla";
2
+ export * from "./array_agents/vanilla";
3
+ export * from "./matrix_agents/vanilla";
4
+ export * from "./test_agents/vanilla";
5
+ export * from "./graph_agents/vanilla";
6
+ export * from "./data_agents/vanilla";
6
7
  export * from "./function_agent";
7
8
  export * from "./embedding_agent";
8
- export * from "./data_agents";
@@ -16,11 +16,11 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
16
16
  };
17
17
  Object.defineProperty(exports, "__esModule", { value: true });
18
18
  // Please refrain from adding agents that require npm. Those should be added to the index.ts.
19
- __exportStar(require("./string_agents"), exports);
20
- __exportStar(require("./array_agents"), exports);
21
- __exportStar(require("./matrix_agents"), exports);
22
- __exportStar(require("./test_agents"), exports);
23
- __exportStar(require("./graph_agents"), exports);
19
+ __exportStar(require("./string_agents/vanilla"), exports);
20
+ __exportStar(require("./array_agents/vanilla"), exports);
21
+ __exportStar(require("./matrix_agents/vanilla"), exports);
22
+ __exportStar(require("./test_agents/vanilla"), exports);
23
+ __exportStar(require("./graph_agents/vanilla"), exports);
24
+ __exportStar(require("./data_agents/vanilla"), exports);
24
25
  __exportStar(require("./function_agent"), exports);
25
26
  __exportStar(require("./embedding_agent"), exports);
26
- __exportStar(require("./data_agents"), exports);
package/lib/graphai.d.ts CHANGED
@@ -1,5 +1,5 @@
1
- export { AgentFunction, AgentFunctionDictonary, GraphData } from "./type";
2
- import { AgentFunctionDictonary, AgentFilterInfo, GraphData, DataSource, ResultDataDictonary, ResultData, DefaultResultData } from "./type";
1
+ export { AgentFunction, GraphData, AgentFunctionInfoDictonary } from "./type";
2
+ import { AgentFunctionInfoDictonary, AgentFilterInfo, GraphData, DataSource, ResultDataDictonary, ResultData, DefaultResultData } from "./type";
3
3
  import { TransactionLog } from "./transaction_log";
4
4
  import { ComputedNode, StaticNode } from "./node";
5
5
  import { TaskManager } from "./task_manager";
@@ -10,7 +10,7 @@ export declare class GraphAI {
10
10
  private readonly data;
11
11
  private readonly loop?;
12
12
  private readonly logs;
13
- readonly agentFunctionDictionary: AgentFunctionDictonary;
13
+ readonly agentFunctionInfoDictionary: AgentFunctionInfoDictonary;
14
14
  readonly taskManager: TaskManager;
15
15
  readonly agentFilters: AgentFilterInfo[];
16
16
  readonly retryLimit?: number;
@@ -22,11 +22,11 @@ export declare class GraphAI {
22
22
  private createNodes;
23
23
  private getValueFromResults;
24
24
  private initializeNodes;
25
- constructor(data: GraphData, agentFunctionDictionary: AgentFunctionDictonary, options?: {
25
+ constructor(data: GraphData, agentFunctionInfoDictionary: AgentFunctionInfoDictonary, options?: {
26
26
  agentFilters?: AgentFilterInfo[] | undefined;
27
27
  taskManager?: TaskManager | undefined;
28
28
  });
29
- getAgentFunction(agentId?: string): import("./type").AgentFunction<any, any, any>;
29
+ getAgentFunctionInfo(agentId?: string): import("./type").AgentFunctionInfo;
30
30
  asString(): string;
31
31
  results<T = DefaultResultData>(all: boolean): ResultDataDictonary<T>;
32
32
  errors(): Record<string, Error>;
package/lib/graphai.js CHANGED
@@ -63,7 +63,7 @@ class GraphAI {
63
63
  }
64
64
  });
65
65
  }
66
- constructor(data, agentFunctionDictionary, options = { taskManager: undefined, agentFilters: [] }) {
66
+ constructor(data, agentFunctionInfoDictionary, options = { taskManager: undefined, agentFilters: [] }) {
67
67
  this.logs = [];
68
68
  this.onLogCallback = (__log, __isUpdate) => { };
69
69
  this.repeatCount = 0;
@@ -77,7 +77,7 @@ class GraphAI {
77
77
  this.retryLimit = data.retry; // optional
78
78
  this.graphId = URL.createObjectURL(new Blob()).slice(-36);
79
79
  this.data = data;
80
- this.agentFunctionDictionary = agentFunctionDictionary;
80
+ this.agentFunctionInfoDictionary = agentFunctionInfoDictionary;
81
81
  this.taskManager = options.taskManager ?? new task_manager_1.TaskManager(data.concurrency ?? defaultConcurrency);
82
82
  this.agentFilters = options.agentFilters ?? [];
83
83
  this.loop = data.loop;
@@ -85,13 +85,13 @@ class GraphAI {
85
85
  this.onComplete = () => {
86
86
  console.error("-- SOMETHING IS WRONG: onComplete is called without run()");
87
87
  };
88
- (0, validator_1.validateGraphData)(data, Object.keys(agentFunctionDictionary));
88
+ (0, validator_1.validateGraphData)(data, Object.keys(agentFunctionInfoDictionary));
89
89
  this.nodes = this.createNodes(data);
90
90
  this.initializeNodes();
91
91
  }
92
- getAgentFunction(agentId) {
93
- if (agentId && this.agentFunctionDictionary[agentId]) {
94
- return this.agentFunctionDictionary[agentId];
92
+ getAgentFunctionInfo(agentId) {
93
+ if (agentId && this.agentFunctionInfoDictionary[agentId]) {
94
+ return this.agentFunctionInfoDictionary[agentId];
95
95
  }
96
96
  throw new Error("No agent: " + agentId);
97
97
  }
package/lib/index.d.ts CHANGED
@@ -1,4 +1,4 @@
1
1
  import { GraphAI } from "./graphai";
2
2
  export { GraphAI };
3
- export { AgentFunction, AgentFunctionDictonary, GraphData, ResultDataDictonary, ResultData, NodeState } from "./type";
3
+ export { AgentFunction, AgentFunctionInfoDictonary, GraphData, ResultDataDictonary, ResultData, NodeState } from "./type";
4
4
  export type { TransactionLog } from "./transaction_log";
package/lib/node.js CHANGED
@@ -201,7 +201,7 @@ class ComputedNode extends Node {
201
201
  }, this.timeout);
202
202
  }
203
203
  try {
204
- const agentFunction = this.agentFunction ?? this.graph.getAgentFunction(this.agentId);
204
+ const agentFunction = this.agentFunction ?? this.graph.getAgentFunctionInfo(this.agentId).agent;
205
205
  const localLog = [];
206
206
  const params = Object.keys(this.dynamicParams).reduce((tmp, key) => {
207
207
  const [result] = this.graph.resultsOf([this.dynamicParams[key]]);
@@ -227,7 +227,7 @@ class ComputedNode extends Node {
227
227
  this.graph.taskManager.prepareForNesting();
228
228
  context.taskManager = this.graph.taskManager;
229
229
  context.graphData = this.nestedGraph;
230
- context.agents = this.graph.agentFunctionDictionary;
230
+ context.agents = this.graph.agentFunctionInfoDictionary;
231
231
  }
232
232
  const result = await this.agentFilterHandler(context, agentFunction);
233
233
  if (this.nestedGraph) {
package/lib/type.d.ts CHANGED
@@ -64,7 +64,7 @@ export type AgentFunctionContext<ParamsType = DefaultParamsType, InputDataType =
64
64
  agentId?: string;
65
65
  };
66
66
  graphData?: GraphData | string;
67
- agents?: AgentFunctionDictonary;
67
+ agents?: AgentFunctionInfoDictonary;
68
68
  taskManager?: TaskManager;
69
69
  filterParams: AgentFilterParams;
70
70
  agentFilters?: AgentFilterInfo[];
@@ -79,7 +79,6 @@ export type AgentFilterInfo = {
79
79
  nodeIds?: string[];
80
80
  filterParams?: AgentFilterParams;
81
81
  };
82
- export type AgentFunctionDictonary = Record<string, AgentFunction<any, any, any>>;
83
82
  export type AgentFunctionInfo = {
84
83
  name: string;
85
84
  agent: AgentFunction<any, any, any>;
@@ -96,3 +95,4 @@ export type AgentFunctionInfo = {
96
95
  repository: string;
97
96
  license: string;
98
97
  };
98
+ export type AgentFunctionInfoDictonary = Record<string, AgentFunctionInfo>;
@@ -1,2 +1,2 @@
1
- import { AgentFunctionDictonary } from "../graphai";
2
- export declare const defaultTestAgents: AgentFunctionDictonary;
1
+ import { AgentFunctionInfoDictonary } from "../graphai";
2
+ export declare const defaultTestAgents: AgentFunctionInfoDictonary;
@@ -2,7 +2,7 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.defaultTestAgents = void 0;
4
4
  const experimental_agents_1 = require("../experimental_agents");
5
- const openai_agent_1 = require("../experimental_agents/llm_agents/openai_agent");
5
+ // import openAIMockAgent from "../experimental_agents/llm_agents/openai_agent";
6
6
  exports.defaultTestAgents = {
7
7
  bypassAgent: experimental_agents_1.bypassAgent,
8
8
  echoAgent: experimental_agents_1.echoAgent,
@@ -20,5 +20,5 @@ exports.defaultTestAgents = {
20
20
  popAgent: experimental_agents_1.popAgent,
21
21
  shiftAgent: experimental_agents_1.shiftAgent,
22
22
  streamMockAgent: experimental_agents_1.streamMockAgent,
23
- openAIMockAgent: openai_agent_1.openAIMockAgent,
23
+ // openAIMockAgent,
24
24
  };
@@ -10,5 +10,33 @@ export declare const defaultTestContext: {
10
10
  agents: {};
11
11
  log: never[];
12
12
  };
13
+ export declare const defaultAgentInfo: {
14
+ name: string;
15
+ samples: {
16
+ inputs: never[];
17
+ params: {};
18
+ result: {};
19
+ }[];
20
+ description: string;
21
+ category: never[];
22
+ author: string;
23
+ repository: string;
24
+ license: string;
25
+ };
26
+ export declare const getAgentInfo: (agent: AgentFunction<any, any, any>) => {
27
+ name: string;
28
+ samples: {
29
+ inputs: never[];
30
+ params: {};
31
+ result: {};
32
+ }[];
33
+ description: string;
34
+ category: never[];
35
+ author: string;
36
+ repository: string;
37
+ license: string;
38
+ agent: AgentFunction<any, any, any>;
39
+ mock: AgentFunction<any, any, any>;
40
+ };
13
41
  export declare const agentTestRunner: (agentInfo: AgentFunctionInfo) => Promise<void>;
14
42
  export declare const agentFilterRunnerBuilder: (__agentFilters: AgentFilterInfo[]) => (context: AgentFunctionContext, agent: AgentFunction) => Promise<ResultData>;
@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
3
3
  return (mod && mod.__esModule) ? mod : { "default": mod };
4
4
  };
5
5
  Object.defineProperty(exports, "__esModule", { value: true });
6
- exports.agentFilterRunnerBuilder = exports.agentTestRunner = exports.defaultTestContext = void 0;
6
+ exports.agentFilterRunnerBuilder = exports.agentTestRunner = exports.getAgentInfo = exports.defaultAgentInfo = exports.defaultTestContext = void 0;
7
7
  const node_assert_1 = __importDefault(require("node:assert"));
8
8
  const node_test_1 = __importDefault(require("node:test"));
9
9
  exports.defaultTestContext = {
@@ -17,6 +17,29 @@ exports.defaultTestContext = {
17
17
  agents: {},
18
18
  log: [],
19
19
  };
20
+ exports.defaultAgentInfo = {
21
+ name: "defaultAgentInfo",
22
+ samples: [
23
+ {
24
+ inputs: [],
25
+ params: {},
26
+ result: {},
27
+ },
28
+ ],
29
+ description: "",
30
+ category: [],
31
+ author: "",
32
+ repository: "",
33
+ license: "",
34
+ };
35
+ const getAgentInfo = (agent) => {
36
+ return {
37
+ agent,
38
+ mock: agent,
39
+ ...exports.defaultAgentInfo,
40
+ };
41
+ };
42
+ exports.getAgentInfo = getAgentInfo;
20
43
  // for agent
21
44
  const agentTestRunner = async (agentInfo) => {
22
45
  const { agent, samples, skipTest } = agentInfo;
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "graphai",
3
- "version": "0.3.2",
3
+ "version": "0.4.0",
4
4
  "description": "Asynchronous data flow execution engine for agentic AI apps.",
5
5
  "main": "lib/index.js",
6
6
  "files": [
@@ -1,13 +0,0 @@
1
- export * from "./string_agents/packages";
2
- export * from "./sleeper_agents/packages";
3
- export * from "./data_agents/packages";
4
- export * from "./array_agents/packages";
5
- export * from "./matrix_agents/packages";
6
- export * from "./test_agents/packages";
7
- export * from "./graph_agents/packages";
8
- export * from "./llm_agents/packages";
9
- export * from "./service_agents/packages";
10
- import stringEmbeddingsAgent from "./embedding_agent";
11
- import tokenBoundStringsAgent from "./token_agent";
12
- import functionAgentInfo from "./function_agent";
13
- export { stringEmbeddingsAgent, tokenBoundStringsAgent, functionAgentInfo };
@@ -1,35 +0,0 @@
1
- "use strict";
2
- var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
- if (k2 === undefined) k2 = k;
4
- var desc = Object.getOwnPropertyDescriptor(m, k);
5
- if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
- desc = { enumerable: true, get: function() { return m[k]; } };
7
- }
8
- Object.defineProperty(o, k2, desc);
9
- }) : (function(o, m, k, k2) {
10
- if (k2 === undefined) k2 = k;
11
- o[k2] = m[k];
12
- }));
13
- var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
- for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
- };
16
- var __importDefault = (this && this.__importDefault) || function (mod) {
17
- return (mod && mod.__esModule) ? mod : { "default": mod };
18
- };
19
- Object.defineProperty(exports, "__esModule", { value: true });
20
- exports.functionAgentInfo = exports.tokenBoundStringsAgent = exports.stringEmbeddingsAgent = void 0;
21
- __exportStar(require("./string_agents/packages"), exports);
22
- __exportStar(require("./sleeper_agents/packages"), exports);
23
- __exportStar(require("./data_agents/packages"), exports);
24
- __exportStar(require("./array_agents/packages"), exports);
25
- __exportStar(require("./matrix_agents/packages"), exports);
26
- __exportStar(require("./test_agents/packages"), exports);
27
- __exportStar(require("./graph_agents/packages"), exports);
28
- __exportStar(require("./llm_agents/packages"), exports);
29
- __exportStar(require("./service_agents/packages"), exports);
30
- const embedding_agent_1 = __importDefault(require("./embedding_agent"));
31
- exports.stringEmbeddingsAgent = embedding_agent_1.default;
32
- const token_agent_1 = __importDefault(require("./token_agent"));
33
- exports.tokenBoundStringsAgent = token_agent_1.default;
34
- const function_agent_1 = __importDefault(require("./function_agent"));
35
- exports.functionAgentInfo = function_agent_1.default;