graphai 0.4.4 → 0.4.5
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.
- package/lib/graphai.d.ts +6 -6
- package/lib/graphai.js +14 -2
- package/lib/index.d.ts +1 -1
- package/lib/type.d.ts +5 -0
- package/lib/utils/test_utils.d.ts +0 -2
- package/lib/utils/test_utils.js +1 -28
- package/lib/utils/utils.js +4 -2
- package/package.json +4 -5
- package/README.md +0 -388
- package/lib/experimental_agents/array_agents/index.d.ts +0 -1
- package/lib/experimental_agents/array_agents/index.js +0 -17
- package/lib/experimental_agents/array_agents/packages.d.ts +0 -4
- package/lib/experimental_agents/array_agents/packages.js +0 -12
- package/lib/experimental_agents/array_agents/pop_agent.d.ts +0 -28
- package/lib/experimental_agents/array_agents/pop_agent.js +0 -50
- package/lib/experimental_agents/array_agents/push_agent.d.ts +0 -32
- package/lib/experimental_agents/array_agents/push_agent.js +0 -41
- package/lib/experimental_agents/array_agents/shift_agent.d.ts +0 -28
- package/lib/experimental_agents/array_agents/shift_agent.js +0 -39
- package/lib/experimental_agents/array_agents/vanilla.d.ts +0 -4
- package/lib/experimental_agents/array_agents/vanilla.js +0 -12
- package/lib/experimental_agents/data_agents/copy_agent.d.ts +0 -28
- package/lib/experimental_agents/data_agents/copy_agent.js +0 -31
- package/lib/experimental_agents/data_agents/data_object_merge_template_agent.d.ts +0 -126
- package/lib/experimental_agents/data_agents/data_object_merge_template_agent.js +0 -72
- package/lib/experimental_agents/data_agents/data_sum_template_agent.d.ts +0 -18
- package/lib/experimental_agents/data_agents/data_sum_template_agent.js +0 -46
- package/lib/experimental_agents/data_agents/index.d.ts +0 -3
- package/lib/experimental_agents/data_agents/index.js +0 -23
- package/lib/experimental_agents/data_agents/packages.d.ts +0 -6
- package/lib/experimental_agents/data_agents/packages.js +0 -16
- package/lib/experimental_agents/data_agents/property_filter_agent.d.ts +0 -237
- package/lib/experimental_agents/data_agents/property_filter_agent.js +0 -140
- package/lib/experimental_agents/data_agents/total_agent.d.ts +0 -60
- package/lib/experimental_agents/data_agents/total_agent.js +0 -81
- package/lib/experimental_agents/data_agents/vanilla.d.ts +0 -5
- package/lib/experimental_agents/data_agents/vanilla.js +0 -14
- package/lib/experimental_agents/embedding_agent.d.ts +0 -20
- package/lib/experimental_agents/embedding_agent.js +0 -55
- package/lib/experimental_agents/graph_agents/index.d.ts +0 -1
- package/lib/experimental_agents/graph_agents/index.js +0 -17
- package/lib/experimental_agents/graph_agents/map_agent.d.ts +0 -23
- package/lib/experimental_agents/graph_agents/map_agent.js +0 -70
- package/lib/experimental_agents/graph_agents/nested_agent.d.ts +0 -22
- package/lib/experimental_agents/graph_agents/nested_agent.js +0 -58
- package/lib/experimental_agents/graph_agents/packages.d.ts +0 -4
- package/lib/experimental_agents/graph_agents/packages.js +0 -12
- package/lib/experimental_agents/graph_agents/vanilla.d.ts +0 -4
- package/lib/experimental_agents/graph_agents/vanilla.js +0 -12
- package/lib/experimental_agents/graph_agents/worker_agent.d.ts +0 -39
- package/lib/experimental_agents/graph_agents/worker_agent.js +0 -111
- package/lib/experimental_agents/index.d.ts +0 -13
- package/lib/experimental_agents/index.js +0 -36
- package/lib/experimental_agents/input_agents/index.d.ts +0 -2
- package/lib/experimental_agents/input_agents/index.js +0 -8
- package/lib/experimental_agents/input_agents/packages.d.ts +0 -2
- package/lib/experimental_agents/input_agents/packages.js +0 -8
- package/lib/experimental_agents/input_agents/text_input_agent.d.ts +0 -33
- package/lib/experimental_agents/input_agents/text_input_agent.js +0 -30
- package/lib/experimental_agents/llm_agents/anthropic_agent.d.ts +0 -34
- package/lib/experimental_agents/llm_agents/anthropic_agent.js +0 -47
- package/lib/experimental_agents/llm_agents/groq_agent.d.ts +0 -47
- package/lib/experimental_agents/llm_agents/groq_agent.js +0 -97
- package/lib/experimental_agents/llm_agents/index.d.ts +0 -5
- package/lib/experimental_agents/llm_agents/index.js +0 -14
- package/lib/experimental_agents/llm_agents/openai_agent.d.ts +0 -70
- package/lib/experimental_agents/llm_agents/openai_agent.js +0 -100
- package/lib/experimental_agents/llm_agents/packages.d.ts +0 -5
- package/lib/experimental_agents/llm_agents/packages.js +0 -14
- package/lib/experimental_agents/llm_agents/slashgpt_agent.d.ts +0 -29
- package/lib/experimental_agents/llm_agents/slashgpt_agent.js +0 -39
- package/lib/experimental_agents/matrix_agents/dot_product_agent.d.ts +0 -18
- package/lib/experimental_agents/matrix_agents/dot_product_agent.js +0 -60
- package/lib/experimental_agents/matrix_agents/index.d.ts +0 -1
- package/lib/experimental_agents/matrix_agents/index.js +0 -17
- package/lib/experimental_agents/matrix_agents/packages.d.ts +0 -3
- package/lib/experimental_agents/matrix_agents/packages.js +0 -10
- package/lib/experimental_agents/matrix_agents/sort_by_values_agent.d.ts +0 -32
- package/lib/experimental_agents/matrix_agents/sort_by_values_agent.js +0 -60
- package/lib/experimental_agents/matrix_agents/vanilla.d.ts +0 -3
- package/lib/experimental_agents/matrix_agents/vanilla.js +0 -10
- package/lib/experimental_agents/service_agents/fetch_agent.d.ts +0 -62
- package/lib/experimental_agents/service_agents/fetch_agent.js +0 -96
- package/lib/experimental_agents/service_agents/index.d.ts +0 -3
- package/lib/experimental_agents/service_agents/index.js +0 -10
- package/lib/experimental_agents/service_agents/packages.d.ts +0 -3
- package/lib/experimental_agents/service_agents/packages.js +0 -10
- package/lib/experimental_agents/service_agents/wikipedia.d.ts +0 -23
- package/lib/experimental_agents/service_agents/wikipedia.js +0 -39
- package/lib/experimental_agents/sleeper_agents/index.d.ts +0 -3
- package/lib/experimental_agents/sleeper_agents/index.js +0 -10
- package/lib/experimental_agents/sleeper_agents/packages.d.ts +0 -3
- package/lib/experimental_agents/sleeper_agents/packages.js +0 -10
- package/lib/experimental_agents/sleeper_agents/sleeper_agent.d.ts +0 -23
- package/lib/experimental_agents/sleeper_agents/sleeper_agent.js +0 -28
- package/lib/experimental_agents/sleeper_agents/sleeper_agent_debug.d.ts +0 -26
- package/lib/experimental_agents/sleeper_agents/sleeper_agent_debug.js +0 -31
- package/lib/experimental_agents/string_agents/index.d.ts +0 -1
- package/lib/experimental_agents/string_agents/index.js +0 -17
- package/lib/experimental_agents/string_agents/json_parser_agent.d.ts +0 -38
- package/lib/experimental_agents/string_agents/json_parser_agent.js +0 -39
- package/lib/experimental_agents/string_agents/packages.d.ts +0 -4
- package/lib/experimental_agents/string_agents/packages.js +0 -12
- package/lib/experimental_agents/string_agents/string_splitter_agent.d.ts +0 -9
- package/lib/experimental_agents/string_agents/string_splitter_agent.js +0 -65
- package/lib/experimental_agents/string_agents/string_template_agent.d.ts +0 -70
- package/lib/experimental_agents/string_agents/string_template_agent.js +0 -64
- package/lib/experimental_agents/string_agents/vanilla.d.ts +0 -4
- package/lib/experimental_agents/string_agents/vanilla.js +0 -12
- package/lib/experimental_agents/test_agents/bypass_agent.d.ts +0 -8
- package/lib/experimental_agents/test_agents/bypass_agent.js +0 -59
- package/lib/experimental_agents/test_agents/copy2array_agent.d.ts +0 -7
- package/lib/experimental_agents/test_agents/copy2array_agent.js +0 -39
- package/lib/experimental_agents/test_agents/copy_message_agent.d.ts +0 -9
- package/lib/experimental_agents/test_agents/copy_message_agent.js +0 -30
- package/lib/experimental_agents/test_agents/counting_agent.d.ts +0 -8
- package/lib/experimental_agents/test_agents/counting_agent.js +0 -30
- package/lib/experimental_agents/test_agents/echo_agent.d.ts +0 -4
- package/lib/experimental_agents/test_agents/echo_agent.js +0 -34
- package/lib/experimental_agents/test_agents/index.d.ts +0 -1
- package/lib/experimental_agents/test_agents/index.js +0 -17
- package/lib/experimental_agents/test_agents/merge_node_id_agent.d.ts +0 -4
- package/lib/experimental_agents/test_agents/merge_node_id_agent.js +0 -32
- package/lib/experimental_agents/test_agents/packages.d.ts +0 -8
- package/lib/experimental_agents/test_agents/packages.js +0 -20
- package/lib/experimental_agents/test_agents/stream_mock_agent.d.ts +0 -4
- package/lib/experimental_agents/test_agents/stream_mock_agent.js +0 -35
- package/lib/experimental_agents/test_agents/vanilla.d.ts +0 -8
- package/lib/experimental_agents/test_agents/vanilla.js +0 -20
- package/lib/experimental_agents/token_agent.d.ts +0 -36
- package/lib/experimental_agents/token_agent.js +0 -74
- package/lib/experimental_agents/vanilla.d.ts +0 -8
- package/lib/experimental_agents/vanilla.js +0 -30
- package/lib/utils/test_agents.d.ts +0 -2
- package/lib/utils/test_agents.js +0 -24
- package/lib/vanilla_agents.d.ts +0 -1
- package/lib/vanilla_agents.js +0 -17
|
@@ -1,70 +0,0 @@
|
|
|
1
|
-
"use strict";
|
|
2
|
-
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.mapAgent = void 0;
|
|
4
|
-
const index_1 = require("../../index");
|
|
5
|
-
const utils_1 = require("../../utils/utils");
|
|
6
|
-
const nested_agent_1 = require("./nested_agent");
|
|
7
|
-
const mapAgent = async ({ params, inputs, agents, log, taskManager, graphData, agentFilters }) => {
|
|
8
|
-
if (taskManager) {
|
|
9
|
-
const status = taskManager.getStatus();
|
|
10
|
-
(0, utils_1.assert)(status.concurrency > status.running, `mapAgent: Concurrency is too low: ${status.concurrency}`);
|
|
11
|
-
}
|
|
12
|
-
const nestedGraphData = (0, nested_agent_1.getNestedGraphData)(graphData, inputs);
|
|
13
|
-
const input = (Array.isArray(inputs[0]) ? inputs[0] : inputs).map((item) => item);
|
|
14
|
-
if (params.limit && params.limit < input.length) {
|
|
15
|
-
input.length = params.limit; // trim
|
|
16
|
-
}
|
|
17
|
-
const namedInputs = params.namedInputs ??
|
|
18
|
-
inputs.map((__input, index) => {
|
|
19
|
-
return `$${index}`;
|
|
20
|
-
});
|
|
21
|
-
namedInputs.forEach((nodeId) => {
|
|
22
|
-
if (nestedGraphData.nodes[nodeId] === undefined) {
|
|
23
|
-
// If the input node does not exist, automatically create a static node
|
|
24
|
-
nestedGraphData.nodes[nodeId] = { value: {} };
|
|
25
|
-
}
|
|
26
|
-
});
|
|
27
|
-
const graphs = input.map((data) => {
|
|
28
|
-
const graphAI = new index_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters: agentFilters || [] });
|
|
29
|
-
// Only the first input will be mapped
|
|
30
|
-
namedInputs.forEach((injectToNodeId, index) => {
|
|
31
|
-
graphAI.injectValue(injectToNodeId, index === 0 ? data : inputs[index], "__mapAgent_inputs__");
|
|
32
|
-
});
|
|
33
|
-
return graphAI;
|
|
34
|
-
});
|
|
35
|
-
const runs = graphs.map((graph) => {
|
|
36
|
-
return graph.run(false);
|
|
37
|
-
});
|
|
38
|
-
const results = await Promise.all(runs);
|
|
39
|
-
const nodeIds = Object.keys(results[0]);
|
|
40
|
-
// assert(nodeIds.length > 0, "mapAgent: no return values (missing isResult)");
|
|
41
|
-
const compositeResult = nodeIds.reduce((tmp, nodeId) => {
|
|
42
|
-
tmp[nodeId] = results.map((result) => {
|
|
43
|
-
return result[nodeId];
|
|
44
|
-
});
|
|
45
|
-
return tmp;
|
|
46
|
-
}, {});
|
|
47
|
-
if (log) {
|
|
48
|
-
const logs = graphs.map((graph, index) => {
|
|
49
|
-
return graph.transactionLogs().map((log) => {
|
|
50
|
-
log.mapIndex = index;
|
|
51
|
-
return log;
|
|
52
|
-
});
|
|
53
|
-
});
|
|
54
|
-
log.push(...logs.flat());
|
|
55
|
-
}
|
|
56
|
-
return compositeResult;
|
|
57
|
-
};
|
|
58
|
-
exports.mapAgent = mapAgent;
|
|
59
|
-
const mapAgentInfo = {
|
|
60
|
-
name: "mapAgent",
|
|
61
|
-
agent: exports.mapAgent,
|
|
62
|
-
mock: exports.mapAgent,
|
|
63
|
-
samples: [],
|
|
64
|
-
description: "Map Agent",
|
|
65
|
-
category: ["graph"],
|
|
66
|
-
author: "Receptron team",
|
|
67
|
-
repository: "https://github.com/receptron/graphai",
|
|
68
|
-
license: "MIT",
|
|
69
|
-
};
|
|
70
|
-
exports.default = mapAgentInfo;
|
|
@@ -1,22 +0,0 @@
|
|
|
1
|
-
import { AgentFunction } from "../../index";
|
|
2
|
-
import { GraphData } from "../../type";
|
|
3
|
-
export declare const getNestedGraphData: (graphData: GraphData | string | undefined, inputs: Array<any>) => GraphData;
|
|
4
|
-
export declare const nestedAgent: AgentFunction<{
|
|
5
|
-
namedInputs?: Array<string>;
|
|
6
|
-
}>;
|
|
7
|
-
declare const nestedAgentInfo: {
|
|
8
|
-
name: string;
|
|
9
|
-
agent: AgentFunction<{
|
|
10
|
-
namedInputs?: string[] | undefined;
|
|
11
|
-
}>;
|
|
12
|
-
mock: AgentFunction<{
|
|
13
|
-
namedInputs?: string[] | undefined;
|
|
14
|
-
}>;
|
|
15
|
-
samples: never[];
|
|
16
|
-
description: string;
|
|
17
|
-
category: string[];
|
|
18
|
-
author: string;
|
|
19
|
-
repository: string;
|
|
20
|
-
license: string;
|
|
21
|
-
};
|
|
22
|
-
export default nestedAgentInfo;
|
|
@@ -1,58 +0,0 @@
|
|
|
1
|
-
"use strict";
|
|
2
|
-
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.nestedAgent = exports.getNestedGraphData = void 0;
|
|
4
|
-
const index_1 = require("../../index");
|
|
5
|
-
const utils_1 = require("../../utils/utils");
|
|
6
|
-
// This function allows us to use one of inputs as the graph data for this nested agent,
|
|
7
|
-
// which is equivalent to "eval" of JavaScript.
|
|
8
|
-
const getNestedGraphData = (graphData, inputs) => {
|
|
9
|
-
(0, utils_1.assert)(graphData !== undefined, "nestedAgent: graphData is required");
|
|
10
|
-
if (typeof graphData === "string") {
|
|
11
|
-
const regex = /^\$(\d+)$/;
|
|
12
|
-
const match = graphData.match(regex);
|
|
13
|
-
if (match) {
|
|
14
|
-
const index = parseInt(match[1], 10);
|
|
15
|
-
if (index < inputs.length) {
|
|
16
|
-
return inputs[index];
|
|
17
|
-
}
|
|
18
|
-
}
|
|
19
|
-
(0, utils_1.assert)(false, `getNestedGraphData: Invalid graphData string: ${graphData}`);
|
|
20
|
-
}
|
|
21
|
-
return graphData;
|
|
22
|
-
};
|
|
23
|
-
exports.getNestedGraphData = getNestedGraphData;
|
|
24
|
-
const nestedAgent = async ({ params, inputs, agents, log, taskManager, graphData, agentFilters }) => {
|
|
25
|
-
if (taskManager) {
|
|
26
|
-
const status = taskManager.getStatus(false);
|
|
27
|
-
(0, utils_1.assert)(status.concurrency > status.running, `nestedAgent: Concurrency is too low: ${status.concurrency}`);
|
|
28
|
-
}
|
|
29
|
-
const nestedGraphData = (0, exports.getNestedGraphData)(graphData, inputs);
|
|
30
|
-
const namedInputs = params.namedInputs ?? inputs.map((__input, index) => `$${index}`);
|
|
31
|
-
namedInputs.forEach((nodeId, index) => {
|
|
32
|
-
if (nestedGraphData.nodes[nodeId] === undefined) {
|
|
33
|
-
// If the input node does not exist, automatically create a static node
|
|
34
|
-
nestedGraphData.nodes[nodeId] = { value: inputs[index] };
|
|
35
|
-
}
|
|
36
|
-
else {
|
|
37
|
-
// Otherwise, inject the proper data here (instead of calling injectTo method later)
|
|
38
|
-
nestedGraphData.nodes[nodeId]["value"] = inputs[index];
|
|
39
|
-
}
|
|
40
|
-
});
|
|
41
|
-
const graphAI = new index_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters });
|
|
42
|
-
const results = await graphAI.run(false);
|
|
43
|
-
log?.push(...graphAI.transactionLogs());
|
|
44
|
-
return results;
|
|
45
|
-
};
|
|
46
|
-
exports.nestedAgent = nestedAgent;
|
|
47
|
-
const nestedAgentInfo = {
|
|
48
|
-
name: "nestedAgent",
|
|
49
|
-
agent: exports.nestedAgent,
|
|
50
|
-
mock: exports.nestedAgent,
|
|
51
|
-
samples: [],
|
|
52
|
-
description: "nested Agent",
|
|
53
|
-
category: ["graph"],
|
|
54
|
-
author: "Receptron team",
|
|
55
|
-
repository: "https://github.com/receptron/graphai",
|
|
56
|
-
license: "MIT",
|
|
57
|
-
};
|
|
58
|
-
exports.default = nestedAgentInfo;
|
|
@@ -1,4 +0,0 @@
|
|
|
1
|
-
import nestedAgent from "../../experimental_agents/graph_agents/nested_agent";
|
|
2
|
-
import mapAgent from "../../experimental_agents/graph_agents/map_agent";
|
|
3
|
-
import workerAgent from "../../experimental_agents/graph_agents/worker_agent";
|
|
4
|
-
export { nestedAgent, mapAgent, workerAgent };
|
|
@@ -1,12 +0,0 @@
|
|
|
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.workerAgent = 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;
|
|
11
|
-
const worker_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/worker_agent"));
|
|
12
|
-
exports.workerAgent = worker_agent_1.default;
|
|
@@ -1,4 +0,0 @@
|
|
|
1
|
-
import nestedAgent from "../../experimental_agents/graph_agents/nested_agent";
|
|
2
|
-
import mapAgent from "../../experimental_agents/graph_agents/map_agent";
|
|
3
|
-
import workerAgent from "../../experimental_agents/graph_agents/worker_agent";
|
|
4
|
-
export { nestedAgent, mapAgent, workerAgent };
|
|
@@ -1,12 +0,0 @@
|
|
|
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.workerAgent = 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;
|
|
11
|
-
const worker_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/worker_agent"));
|
|
12
|
-
exports.workerAgent = worker_agent_1.default;
|
|
@@ -1,39 +0,0 @@
|
|
|
1
|
-
import { AgentFunction } from "../../index";
|
|
2
|
-
export declare const workerAgent: AgentFunction<{
|
|
3
|
-
namedInputs?: Array<string>;
|
|
4
|
-
}, any, any>;
|
|
5
|
-
declare const workerAgentInfo: {
|
|
6
|
-
name: string;
|
|
7
|
-
agent: AgentFunction<{
|
|
8
|
-
namedInputs?: string[] | undefined;
|
|
9
|
-
}, any, any>;
|
|
10
|
-
mock: AgentFunction<{
|
|
11
|
-
namedInputs?: string[] | undefined;
|
|
12
|
-
}, any, any>;
|
|
13
|
-
samples: {
|
|
14
|
-
inputs: string[];
|
|
15
|
-
params: {};
|
|
16
|
-
result: {
|
|
17
|
-
message: string;
|
|
18
|
-
};
|
|
19
|
-
graph: {
|
|
20
|
-
version: number;
|
|
21
|
-
nodes: {
|
|
22
|
-
source: {
|
|
23
|
-
value: string;
|
|
24
|
-
};
|
|
25
|
-
message: {
|
|
26
|
-
agent: string;
|
|
27
|
-
inputs: string[];
|
|
28
|
-
isResult: boolean;
|
|
29
|
-
};
|
|
30
|
-
};
|
|
31
|
-
};
|
|
32
|
-
}[];
|
|
33
|
-
description: string;
|
|
34
|
-
category: string[];
|
|
35
|
-
author: string;
|
|
36
|
-
repository: string;
|
|
37
|
-
license: string;
|
|
38
|
-
};
|
|
39
|
-
export default workerAgentInfo;
|
|
@@ -1,111 +0,0 @@
|
|
|
1
|
-
"use strict";
|
|
2
|
-
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.workerAgent = void 0;
|
|
4
|
-
const index_1 = require("../../index");
|
|
5
|
-
const worker_threads_1 = require("worker_threads");
|
|
6
|
-
const vanilla_1 = require("../../experimental_agents/vanilla");
|
|
7
|
-
const utils_1 = require("../../utils/utils");
|
|
8
|
-
const vanillaAgents = {
|
|
9
|
-
totalAgent: vanilla_1.totalAgent,
|
|
10
|
-
dataSumTemplateAgent: vanilla_1.dataSumTemplateAgent,
|
|
11
|
-
propertyFilterAgent: vanilla_1.propertyFilterAgent,
|
|
12
|
-
copyAgent: vanilla_1.copyAgent,
|
|
13
|
-
pushAgent: vanilla_1.pushAgent,
|
|
14
|
-
popAgent: vanilla_1.popAgent,
|
|
15
|
-
shiftAgent: vanilla_1.shiftAgent,
|
|
16
|
-
nestedAgent: vanilla_1.nestedAgent,
|
|
17
|
-
mapAgent: vanilla_1.mapAgent,
|
|
18
|
-
dotProductAgent: vanilla_1.dotProductAgent,
|
|
19
|
-
sortByValuesAgent: vanilla_1.sortByValuesAgent,
|
|
20
|
-
stringSplitterAgent: vanilla_1.stringSplitterAgent,
|
|
21
|
-
stringTemplateAgent: vanilla_1.stringTemplateAgent,
|
|
22
|
-
jsonParserAgent: vanilla_1.jsonParserAgent,
|
|
23
|
-
};
|
|
24
|
-
if (!worker_threads_1.isMainThread && worker_threads_1.parentPort) {
|
|
25
|
-
const port = worker_threads_1.parentPort;
|
|
26
|
-
port.on("message", async (data) => {
|
|
27
|
-
const { graphData } = data;
|
|
28
|
-
const graphAI = new index_1.GraphAI(graphData, vanillaAgents);
|
|
29
|
-
const result = await graphAI.run();
|
|
30
|
-
port.postMessage(result);
|
|
31
|
-
});
|
|
32
|
-
}
|
|
33
|
-
const workerAgent = async ({ inputs, params, /* agents, log, */ graphData }) => {
|
|
34
|
-
const namedInputs = params.namedInputs ?? inputs.map((__input, index) => `$${index}`);
|
|
35
|
-
(0, utils_1.assert)(!!graphData, "required");
|
|
36
|
-
(0, utils_1.assert)(typeof graphData === "object", "required");
|
|
37
|
-
namedInputs.forEach((nodeId, index) => {
|
|
38
|
-
if (graphData.nodes[nodeId] === undefined) {
|
|
39
|
-
// If the input node does not exist, automatically create a static node
|
|
40
|
-
graphData.nodes[nodeId] = { value: inputs[index] };
|
|
41
|
-
}
|
|
42
|
-
else {
|
|
43
|
-
// Otherwise, inject the proper data here (instead of calling injectTo method later)
|
|
44
|
-
graphData.nodes[nodeId]["value"] = inputs[index];
|
|
45
|
-
}
|
|
46
|
-
});
|
|
47
|
-
return new Promise((resolve, reject) => {
|
|
48
|
-
const worker = new worker_threads_1.Worker("./lib/experimental_agents/graph_agents/worker_agent.js");
|
|
49
|
-
worker.on("message", (result) => {
|
|
50
|
-
worker.terminate();
|
|
51
|
-
resolve(result);
|
|
52
|
-
});
|
|
53
|
-
worker.on("error", reject);
|
|
54
|
-
worker.on("exit", (code) => {
|
|
55
|
-
if (code !== 0)
|
|
56
|
-
reject(new Error(`Worker stopped with exit code ${code}`));
|
|
57
|
-
});
|
|
58
|
-
// copyAgent is required for test case
|
|
59
|
-
worker.postMessage({ graphData });
|
|
60
|
-
});
|
|
61
|
-
};
|
|
62
|
-
exports.workerAgent = workerAgent;
|
|
63
|
-
const workerAgentInfo = {
|
|
64
|
-
name: "workerAgent",
|
|
65
|
-
agent: exports.workerAgent,
|
|
66
|
-
mock: exports.workerAgent,
|
|
67
|
-
samples: [
|
|
68
|
-
{
|
|
69
|
-
inputs: [],
|
|
70
|
-
params: {},
|
|
71
|
-
result: { message: "May the force be with you" },
|
|
72
|
-
graph: {
|
|
73
|
-
version: 0.3,
|
|
74
|
-
nodes: {
|
|
75
|
-
source: {
|
|
76
|
-
value: "May the force be with you",
|
|
77
|
-
},
|
|
78
|
-
message: {
|
|
79
|
-
agent: "copyAgent",
|
|
80
|
-
inputs: [":source"],
|
|
81
|
-
isResult: true,
|
|
82
|
-
},
|
|
83
|
-
},
|
|
84
|
-
},
|
|
85
|
-
},
|
|
86
|
-
{
|
|
87
|
-
inputs: ["May the force be with you"],
|
|
88
|
-
params: {},
|
|
89
|
-
result: { message: "May the force be with you" },
|
|
90
|
-
graph: {
|
|
91
|
-
version: 0.3,
|
|
92
|
-
nodes: {
|
|
93
|
-
source: {
|
|
94
|
-
value: "TypeScript compiler fails without this node for some reason.",
|
|
95
|
-
},
|
|
96
|
-
message: {
|
|
97
|
-
agent: "copyAgent",
|
|
98
|
-
inputs: [":$0"],
|
|
99
|
-
isResult: true,
|
|
100
|
-
},
|
|
101
|
-
},
|
|
102
|
-
},
|
|
103
|
-
},
|
|
104
|
-
],
|
|
105
|
-
description: "Map Agent",
|
|
106
|
-
category: ["graph"],
|
|
107
|
-
author: "Receptron team",
|
|
108
|
-
repository: "https://github.com/receptron/graphai",
|
|
109
|
-
license: "MIT",
|
|
110
|
-
};
|
|
111
|
-
exports.default = workerAgentInfo;
|
|
@@ -1,13 +0,0 @@
|
|
|
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";
|
|
7
|
-
export * from "./sleeper_agents";
|
|
8
|
-
export * from "./llm_agents";
|
|
9
|
-
export * from "./service_agents";
|
|
10
|
-
export * from "./input_agents";
|
|
11
|
-
import stringEmbeddingsAgent from "./embedding_agent";
|
|
12
|
-
import tokenBoundStringsAgent from "./token_agent";
|
|
13
|
-
export { stringEmbeddingsAgent, tokenBoundStringsAgent };
|
|
@@ -1,36 +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.tokenBoundStringsAgent = exports.stringEmbeddingsAgent = 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);
|
|
27
|
-
// Agents that use npm modules will be added here.
|
|
28
|
-
__exportStar(require("./sleeper_agents"), exports);
|
|
29
|
-
__exportStar(require("./llm_agents"), exports);
|
|
30
|
-
__exportStar(require("./service_agents"), exports);
|
|
31
|
-
__exportStar(require("./input_agents"), exports);
|
|
32
|
-
// TODO sub folder
|
|
33
|
-
const embedding_agent_1 = __importDefault(require("./embedding_agent"));
|
|
34
|
-
exports.stringEmbeddingsAgent = embedding_agent_1.default;
|
|
35
|
-
const token_agent_1 = __importDefault(require("./token_agent"));
|
|
36
|
-
exports.tokenBoundStringsAgent = token_agent_1.default;
|
|
@@ -1,8 +0,0 @@
|
|
|
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.textInputAgent = void 0;
|
|
7
|
-
const text_input_agent_1 = __importDefault(require("../../experimental_agents/input_agents/text_input_agent"));
|
|
8
|
-
exports.textInputAgent = text_input_agent_1.default;
|
|
@@ -1,8 +0,0 @@
|
|
|
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.textInputAgent = void 0;
|
|
7
|
-
const text_input_agent_1 = __importDefault(require("../../experimental_agents/input_agents/text_input_agent"));
|
|
8
|
-
exports.textInputAgent = text_input_agent_1.default;
|
|
@@ -1,33 +0,0 @@
|
|
|
1
|
-
import { AgentFunction } from "../../index";
|
|
2
|
-
export declare const textInputAgent: AgentFunction<{
|
|
3
|
-
message?: string;
|
|
4
|
-
}, string | {
|
|
5
|
-
[x: string]: string;
|
|
6
|
-
}>;
|
|
7
|
-
declare const textInputAgentInfo: {
|
|
8
|
-
name: string;
|
|
9
|
-
agent: AgentFunction<{
|
|
10
|
-
message?: string | undefined;
|
|
11
|
-
}, string | {
|
|
12
|
-
[x: string]: string;
|
|
13
|
-
}>;
|
|
14
|
-
mock: AgentFunction<{
|
|
15
|
-
message?: string | undefined;
|
|
16
|
-
}, string | {
|
|
17
|
-
[x: string]: string;
|
|
18
|
-
}>;
|
|
19
|
-
samples: {
|
|
20
|
-
inputs: never[];
|
|
21
|
-
params: {
|
|
22
|
-
message: string;
|
|
23
|
-
};
|
|
24
|
-
result: string;
|
|
25
|
-
}[];
|
|
26
|
-
skipTest: boolean;
|
|
27
|
-
description: string;
|
|
28
|
-
category: string[];
|
|
29
|
-
author: string;
|
|
30
|
-
repository: string;
|
|
31
|
-
license: string;
|
|
32
|
-
};
|
|
33
|
-
export default textInputAgentInfo;
|
|
@@ -1,30 +0,0 @@
|
|
|
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.textInputAgent = void 0;
|
|
7
|
-
const input_1 = __importDefault(require("@inquirer/input"));
|
|
8
|
-
const textInputAgent = async ({ params }) => {
|
|
9
|
-
return await (0, input_1.default)({ message: params.message ?? "Enter" });
|
|
10
|
-
};
|
|
11
|
-
exports.textInputAgent = textInputAgent;
|
|
12
|
-
const textInputAgentInfo = {
|
|
13
|
-
name: "textInputAgent",
|
|
14
|
-
agent: exports.textInputAgent,
|
|
15
|
-
mock: exports.textInputAgent,
|
|
16
|
-
samples: [
|
|
17
|
-
{
|
|
18
|
-
inputs: [],
|
|
19
|
-
params: { message: "Enter your message to AI." },
|
|
20
|
-
result: "message from the user",
|
|
21
|
-
},
|
|
22
|
-
],
|
|
23
|
-
skipTest: true,
|
|
24
|
-
description: "Text Input Agent",
|
|
25
|
-
category: ["input"],
|
|
26
|
-
author: "Receptron team",
|
|
27
|
-
repository: "https://github.com/receptron/graphai",
|
|
28
|
-
license: "MIT",
|
|
29
|
-
};
|
|
30
|
-
exports.default = textInputAgentInfo;
|
|
@@ -1,34 +0,0 @@
|
|
|
1
|
-
import { AgentFunction } from "../../index";
|
|
2
|
-
export declare const anthropicAgent: AgentFunction<{
|
|
3
|
-
model?: string;
|
|
4
|
-
query?: string;
|
|
5
|
-
system?: string;
|
|
6
|
-
temperature?: number;
|
|
7
|
-
max_tokens?: number;
|
|
8
|
-
}, Record<string, any> | string, string | Array<any>>;
|
|
9
|
-
declare const anthropicAgentInfo: {
|
|
10
|
-
name: string;
|
|
11
|
-
agent: AgentFunction<{
|
|
12
|
-
model?: string | undefined;
|
|
13
|
-
query?: string | undefined;
|
|
14
|
-
system?: string | undefined;
|
|
15
|
-
temperature?: number | undefined;
|
|
16
|
-
max_tokens?: number | undefined;
|
|
17
|
-
}, string | Record<string, any>, string | any[]>;
|
|
18
|
-
mock: AgentFunction<{
|
|
19
|
-
model?: string | undefined;
|
|
20
|
-
query?: string | undefined;
|
|
21
|
-
system?: string | undefined;
|
|
22
|
-
temperature?: number | undefined;
|
|
23
|
-
max_tokens?: number | undefined;
|
|
24
|
-
}, string | Record<string, any>, string | any[]>;
|
|
25
|
-
samples: never[];
|
|
26
|
-
skipTest: boolean;
|
|
27
|
-
description: string;
|
|
28
|
-
category: string[];
|
|
29
|
-
author: string;
|
|
30
|
-
repository: string;
|
|
31
|
-
license: string;
|
|
32
|
-
npms: string[];
|
|
33
|
-
};
|
|
34
|
-
export default anthropicAgentInfo;
|
|
@@ -1,47 +0,0 @@
|
|
|
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.anthropicAgent = void 0;
|
|
7
|
-
const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
|
|
8
|
-
const anthropicAgent = async ({ params, inputs }) => {
|
|
9
|
-
const { query, system, temperature, max_tokens } = params;
|
|
10
|
-
const [input_query, previous_messages] = inputs;
|
|
11
|
-
// Notice that we ignore params.system if previous_message exists.
|
|
12
|
-
const messages = previous_messages && Array.isArray(previous_messages) ? previous_messages : system ? [{ role: "system", content: system }] : [];
|
|
13
|
-
const content = (query ? [query] : []).concat(input_query ? [input_query] : []).join("\n");
|
|
14
|
-
if (content) {
|
|
15
|
-
messages.push({
|
|
16
|
-
role: "user",
|
|
17
|
-
content,
|
|
18
|
-
});
|
|
19
|
-
}
|
|
20
|
-
const anthropic = new sdk_1.default({
|
|
21
|
-
apiKey: process.env["ANTHROPIC_API_KEY"], // This is the default and can be omitted
|
|
22
|
-
});
|
|
23
|
-
return await anthropic.messages.create({
|
|
24
|
-
model: params.model || "claude-3-haiku-20240307", // "claude-3-opus-20240229",
|
|
25
|
-
messages,
|
|
26
|
-
temperature: temperature ?? 0.7,
|
|
27
|
-
max_tokens: max_tokens ?? 1024,
|
|
28
|
-
// tools: params.tools,
|
|
29
|
-
// tool_choice: params.tool_choice,
|
|
30
|
-
});
|
|
31
|
-
};
|
|
32
|
-
exports.anthropicAgent = anthropicAgent;
|
|
33
|
-
const anthropicAgentInfo = {
|
|
34
|
-
name: "anthropicAgent",
|
|
35
|
-
agent: exports.anthropicAgent,
|
|
36
|
-
mock: exports.anthropicAgent,
|
|
37
|
-
samples: [],
|
|
38
|
-
skipTest: true,
|
|
39
|
-
description: "Anthropic Agent",
|
|
40
|
-
category: ["llm"],
|
|
41
|
-
author: "Receptron team",
|
|
42
|
-
repository: "https://github.com/receptron/graphai",
|
|
43
|
-
license: "MIT",
|
|
44
|
-
// stream: true,
|
|
45
|
-
npms: ["@anthropic-ai/sdk"],
|
|
46
|
-
};
|
|
47
|
-
exports.default = anthropicAgentInfo;
|
|
@@ -1,47 +0,0 @@
|
|
|
1
|
-
import { AgentFunction } from "../../index";
|
|
2
|
-
import { Groq } from "groq-sdk";
|
|
3
|
-
export declare const groqAgent: AgentFunction<{
|
|
4
|
-
model: string;
|
|
5
|
-
query?: string;
|
|
6
|
-
system?: string;
|
|
7
|
-
verbose?: boolean;
|
|
8
|
-
tools?: Groq.Chat.CompletionCreateParams.Tool[];
|
|
9
|
-
temperature?: number;
|
|
10
|
-
max_tokens?: number;
|
|
11
|
-
tool_choice?: Groq.Chat.CompletionCreateParams.ToolChoice;
|
|
12
|
-
stream?: boolean;
|
|
13
|
-
}, any, string | Array<Groq.Chat.CompletionCreateParams.Message>>;
|
|
14
|
-
declare const groqAgentInfo: {
|
|
15
|
-
name: string;
|
|
16
|
-
agent: AgentFunction<{
|
|
17
|
-
model: string;
|
|
18
|
-
query?: string | undefined;
|
|
19
|
-
system?: string | undefined;
|
|
20
|
-
verbose?: boolean | undefined;
|
|
21
|
-
tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
|
|
22
|
-
temperature?: number | undefined;
|
|
23
|
-
max_tokens?: number | undefined;
|
|
24
|
-
tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
|
|
25
|
-
stream?: boolean | undefined;
|
|
26
|
-
}, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
|
|
27
|
-
mock: AgentFunction<{
|
|
28
|
-
model: string;
|
|
29
|
-
query?: string | undefined;
|
|
30
|
-
system?: string | undefined;
|
|
31
|
-
verbose?: boolean | undefined;
|
|
32
|
-
tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
|
|
33
|
-
temperature?: number | undefined;
|
|
34
|
-
max_tokens?: number | undefined;
|
|
35
|
-
tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
|
|
36
|
-
stream?: boolean | undefined;
|
|
37
|
-
}, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
|
|
38
|
-
samples: never[];
|
|
39
|
-
description: string;
|
|
40
|
-
category: string[];
|
|
41
|
-
author: string;
|
|
42
|
-
repository: string;
|
|
43
|
-
license: string;
|
|
44
|
-
stream: boolean;
|
|
45
|
-
npms: string[];
|
|
46
|
-
};
|
|
47
|
-
export default groqAgentInfo;
|