graphai 0.3.1 → 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.
- package/README.md +30 -11
- package/lib/experimental_agents/array_agents/index.d.ts +1 -3
- package/lib/experimental_agents/array_agents/index.js +15 -7
- package/lib/experimental_agents/array_agents/vanilla.d.ts +4 -0
- package/lib/experimental_agents/array_agents/vanilla.js +12 -0
- package/lib/experimental_agents/data_agents/copy_agent.d.ts +2 -0
- package/lib/experimental_agents/data_agents/copy_agent.js +2 -0
- package/lib/experimental_agents/data_agents/data_object_merge_template_agent.d.ts +43 -0
- package/lib/experimental_agents/data_agents/data_object_merge_template_agent.js +22 -0
- package/lib/experimental_agents/data_agents/data_sum_template_agent.js +15 -0
- package/lib/experimental_agents/data_agents/index.d.ts +3 -4
- package/lib/experimental_agents/data_agents/index.js +21 -9
- package/lib/experimental_agents/data_agents/packages.d.ts +2 -1
- package/lib/experimental_agents/data_agents/packages.js +3 -1
- package/lib/experimental_agents/data_agents/property_filter_agent.d.ts +109 -8
- package/lib/experimental_agents/data_agents/property_filter_agent.js +60 -9
- package/lib/experimental_agents/data_agents/total_agent.d.ts +13 -0
- package/lib/experimental_agents/data_agents/total_agent.js +29 -0
- package/lib/experimental_agents/data_agents/vanilla.d.ts +5 -0
- package/lib/experimental_agents/data_agents/vanilla.js +14 -0
- package/lib/experimental_agents/graph_agents/index.d.ts +1 -2
- package/lib/experimental_agents/graph_agents/index.js +15 -5
- package/lib/experimental_agents/graph_agents/map_agent.js +2 -2
- package/lib/experimental_agents/graph_agents/nested_agent.js +9 -12
- package/lib/experimental_agents/graph_agents/vanilla.d.ts +3 -0
- package/lib/experimental_agents/graph_agents/vanilla.js +10 -0
- package/lib/experimental_agents/index.d.ts +10 -3
- package/lib/experimental_agents/index.js +17 -5
- package/lib/experimental_agents/llm_agents/groq_stream_agent.d.ts +42 -0
- package/lib/experimental_agents/llm_agents/groq_stream_agent.js +84 -0
- package/lib/experimental_agents/llm_agents/index.d.ts +5 -2
- package/lib/experimental_agents/llm_agents/index.js +11 -15
- package/lib/experimental_agents/llm_agents/openai_agent.d.ts +1 -0
- package/lib/experimental_agents/llm_agents/openai_agent.js +1 -0
- package/lib/experimental_agents/llm_agents/packages.d.ts +3 -1
- package/lib/experimental_agents/llm_agents/packages.js +5 -1
- package/lib/experimental_agents/matrix_agents/dot_product_agent.js +11 -0
- package/lib/experimental_agents/matrix_agents/index.d.ts +1 -2
- package/lib/experimental_agents/matrix_agents/index.js +15 -5
- package/lib/experimental_agents/matrix_agents/sort_by_values_agent.d.ts +13 -1
- package/lib/experimental_agents/matrix_agents/sort_by_values_agent.js +20 -1
- package/lib/experimental_agents/matrix_agents/vanilla.d.ts +3 -0
- package/lib/experimental_agents/matrix_agents/vanilla.js +10 -0
- package/lib/experimental_agents/service_agents/fetch_agent.js +1 -1
- package/lib/experimental_agents/service_agents/index.d.ts +3 -2
- package/lib/experimental_agents/service_agents/index.js +8 -5
- package/lib/experimental_agents/service_agents/packages.d.ts +2 -1
- package/lib/experimental_agents/service_agents/packages.js +3 -1
- package/lib/experimental_agents/service_agents/wikipedia.d.ts +1 -0
- package/lib/experimental_agents/service_agents/wikipedia.js +1 -0
- package/lib/experimental_agents/sleeper_agents/index.d.ts +3 -2
- package/lib/experimental_agents/sleeper_agents/index.js +7 -4
- package/lib/experimental_agents/string_agents/index.d.ts +1 -2
- package/lib/experimental_agents/string_agents/index.js +15 -5
- package/lib/experimental_agents/string_agents/vanilla.d.ts +3 -0
- package/lib/experimental_agents/string_agents/vanilla.js +10 -0
- package/lib/experimental_agents/test_agents/index.d.ts +1 -7
- package/lib/experimental_agents/test_agents/index.js +15 -15
- package/lib/experimental_agents/test_agents/vanilla.d.ts +8 -0
- package/lib/experimental_agents/test_agents/vanilla.js +20 -0
- package/lib/experimental_agents/token_agent.d.ts +11 -1
- package/lib/experimental_agents/token_agent.js +30 -1
- package/lib/experimental_agents/vanilla.d.ts +6 -6
- package/lib/experimental_agents/vanilla.js +6 -6
- package/lib/graphai.d.ts +5 -5
- package/lib/graphai.js +6 -6
- package/lib/index.d.ts +1 -1
- package/lib/node.js +11 -10
- package/lib/task_manager.js +3 -6
- package/lib/transaction_log.d.ts +1 -1
- package/lib/transaction_log.js +2 -1
- package/lib/type.d.ts +4 -2
- package/lib/utils/test_agents.d.ts +2 -2
- package/lib/utils/test_agents.js +2 -2
- package/lib/utils/test_utils.d.ts +28 -0
- package/lib/utils/test_utils.js +26 -3
- package/package.json +10 -11
- package/lib/experimental_agents/packages.d.ts +0 -13
- package/lib/experimental_agents/packages.js +0 -35
|
@@ -42,6 +42,35 @@ const totalAgentInfo = {
|
|
|
42
42
|
params: sample2Params,
|
|
43
43
|
result: sample2Result,
|
|
44
44
|
},
|
|
45
|
+
{
|
|
46
|
+
inputs: [{ a: 1 }],
|
|
47
|
+
params: {},
|
|
48
|
+
result: { a: 1 },
|
|
49
|
+
},
|
|
50
|
+
{
|
|
51
|
+
inputs: [{ a: 1 }, { a: 2 }],
|
|
52
|
+
params: {},
|
|
53
|
+
result: { a: 3 },
|
|
54
|
+
},
|
|
55
|
+
{
|
|
56
|
+
inputs: [{ a: 1 }, { a: 2 }, { a: 3 }],
|
|
57
|
+
params: {},
|
|
58
|
+
result: { a: 6 },
|
|
59
|
+
},
|
|
60
|
+
{
|
|
61
|
+
inputs: [
|
|
62
|
+
{ a: 1, b: 1 },
|
|
63
|
+
{ a: 2, b: 2 },
|
|
64
|
+
{ a: 3, b: 0 },
|
|
65
|
+
],
|
|
66
|
+
params: {},
|
|
67
|
+
result: { a: 6, b: 3 },
|
|
68
|
+
},
|
|
69
|
+
{
|
|
70
|
+
inputs: [{ a: 1 }, { a: 2, b: 2 }, { a: 3, b: 0 }],
|
|
71
|
+
params: {},
|
|
72
|
+
result: { a: 6, b: 2 },
|
|
73
|
+
},
|
|
45
74
|
],
|
|
46
75
|
description: "Returns the sum of input values",
|
|
47
76
|
category: ["data"],
|
|
@@ -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
|
|
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
|
-
|
|
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);
|
|
@@ -4,7 +4,7 @@ exports.mapAgent = void 0;
|
|
|
4
4
|
const graphai_1 = require("../../graphai");
|
|
5
5
|
const utils_1 = require("../../utils/utils");
|
|
6
6
|
const nested_agent_1 = require("./nested_agent");
|
|
7
|
-
const mapAgent = async ({ params, inputs, agents, log, taskManager, graphData }) => {
|
|
7
|
+
const mapAgent = async ({ params, inputs, agents, log, taskManager, graphData, agentFilters }) => {
|
|
8
8
|
if (taskManager) {
|
|
9
9
|
const status = taskManager.getStatus();
|
|
10
10
|
(0, utils_1.assert)(status.concurrency > status.running, `mapAgent: Concurrency is too low: ${status.concurrency}`);
|
|
@@ -25,7 +25,7 @@ const mapAgent = async ({ params, inputs, agents, log, taskManager, graphData })
|
|
|
25
25
|
}
|
|
26
26
|
});
|
|
27
27
|
const graphs = input.map((data) => {
|
|
28
|
-
const graphAI = new graphai_1.GraphAI(nestedGraphData, agents || {}, { taskManager });
|
|
28
|
+
const graphAI = new graphai_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters: agentFilters || [] });
|
|
29
29
|
// Only the first input will be mapped
|
|
30
30
|
injectionTo.forEach((injectToNodeId, index) => {
|
|
31
31
|
graphAI.injectValue(injectToNodeId, index === 0 ? data : inputs[index], "__mapAgent_inputs__");
|
|
@@ -21,27 +21,24 @@ const getNestedGraphData = (graphData, inputs) => {
|
|
|
21
21
|
return graphData;
|
|
22
22
|
};
|
|
23
23
|
exports.getNestedGraphData = getNestedGraphData;
|
|
24
|
-
const nestedAgent = async ({ params, inputs, agents, log, taskManager, graphData }) => {
|
|
24
|
+
const nestedAgent = async ({ params, inputs, agents, log, taskManager, graphData, agentFilters }) => {
|
|
25
25
|
if (taskManager) {
|
|
26
26
|
const status = taskManager.getStatus(false);
|
|
27
27
|
(0, utils_1.assert)(status.concurrency > status.running, `nestedAgent: Concurrency is too low: ${status.concurrency}`);
|
|
28
28
|
}
|
|
29
29
|
const nestedGraphData = (0, exports.getNestedGraphData)(graphData, inputs);
|
|
30
|
-
const injectionTo = params.injectionTo ??
|
|
31
|
-
|
|
32
|
-
return `$${index}`;
|
|
33
|
-
});
|
|
34
|
-
injectionTo.forEach((nodeId) => {
|
|
30
|
+
const injectionTo = params.injectionTo ?? inputs.map((__input, index) => `$${index}`);
|
|
31
|
+
injectionTo.forEach((nodeId, index) => {
|
|
35
32
|
if (nestedGraphData.nodes[nodeId] === undefined) {
|
|
36
33
|
// If the input node does not exist, automatically create a static node
|
|
37
|
-
nestedGraphData.nodes[nodeId] = { value:
|
|
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];
|
|
38
39
|
}
|
|
39
40
|
});
|
|
40
|
-
const graphAI = new graphai_1.GraphAI(nestedGraphData, agents || {}, { taskManager });
|
|
41
|
-
// Inject inputs to specified source nodes
|
|
42
|
-
injectionTo.forEach((injectToNodeId, index) => {
|
|
43
|
-
graphAI.injectValue(injectToNodeId, inputs[index]);
|
|
44
|
-
});
|
|
41
|
+
const graphAI = new graphai_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters });
|
|
45
42
|
const results = await graphAI.run(false);
|
|
46
43
|
log?.push(...graphAI.transactionLogs());
|
|
47
44
|
return results;
|
|
@@ -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 "./
|
|
2
|
-
export
|
|
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
|
-
|
|
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.
|
|
18
|
-
__exportStar(require("./
|
|
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
|
-
|
|
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;
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { AgentFunction } from "../../graphai";
|
|
2
|
+
import { Groq } from "groq-sdk";
|
|
3
|
+
export declare const groqStreamAgent: 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
|
+
}, any, string | Array<Groq.Chat.CompletionCreateParams.Message>>;
|
|
13
|
+
declare const groqStreamAgentInfo: {
|
|
14
|
+
name: string;
|
|
15
|
+
agent: AgentFunction<{
|
|
16
|
+
model: string;
|
|
17
|
+
query?: string | undefined;
|
|
18
|
+
system?: string | undefined;
|
|
19
|
+
verbose?: boolean | undefined;
|
|
20
|
+
tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
|
|
21
|
+
temperature?: number | undefined;
|
|
22
|
+
max_tokens?: number | undefined;
|
|
23
|
+
tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
|
|
24
|
+
}, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
|
|
25
|
+
mock: AgentFunction<{
|
|
26
|
+
model: string;
|
|
27
|
+
query?: string | undefined;
|
|
28
|
+
system?: string | undefined;
|
|
29
|
+
verbose?: boolean | undefined;
|
|
30
|
+
tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
|
|
31
|
+
temperature?: number | undefined;
|
|
32
|
+
max_tokens?: number | undefined;
|
|
33
|
+
tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
|
|
34
|
+
}, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
|
|
35
|
+
samples: never[];
|
|
36
|
+
description: string;
|
|
37
|
+
category: string[];
|
|
38
|
+
author: string;
|
|
39
|
+
repository: string;
|
|
40
|
+
license: string;
|
|
41
|
+
};
|
|
42
|
+
export default groqStreamAgentInfo;
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.groqStreamAgent = void 0;
|
|
4
|
+
const groq_sdk_1 = require("groq-sdk");
|
|
5
|
+
const utils_1 = require("../../utils/utils");
|
|
6
|
+
const groq = process.env.GROQ_API_KEY ? new groq_sdk_1.Groq({ apiKey: process.env.GROQ_API_KEY }) : undefined;
|
|
7
|
+
//
|
|
8
|
+
// This agent takes two optional inputs, and following parameters.
|
|
9
|
+
// inputs:
|
|
10
|
+
// - [0]: query string (typically from the user), optional
|
|
11
|
+
// - [1]: array of messages from previous conversation, optional
|
|
12
|
+
//
|
|
13
|
+
// params:
|
|
14
|
+
// - model: LLM model (Llama3-8b-8192, Llama3-70b-8192, Mixtral-8x7b-32768), required.
|
|
15
|
+
// - query: Additional query string from the app to prepend the query from the user, optional.
|
|
16
|
+
// - system: System prompt (ignored if inputs[1] is specified), optional
|
|
17
|
+
// - tools: Function definitions, optional
|
|
18
|
+
// - tool_choice: Tool choice parameter, optional (default = "auto")
|
|
19
|
+
// - temperature: Controls randomness of responses, optional (default = 0.7)
|
|
20
|
+
// - max_tokens: The maximum number of tokens that the model can process in a single response, optional.
|
|
21
|
+
// - verbose: dumps the message array to the debug console, before sending it the LLM.
|
|
22
|
+
//
|
|
23
|
+
// https://console.groq.com/docs/quickstart
|
|
24
|
+
//
|
|
25
|
+
const groqStreamAgent = async ({ params, inputs, filterParams }) => {
|
|
26
|
+
(0, utils_1.assert)(groq !== undefined, "The GROQ_API_KEY environment variable is missing.");
|
|
27
|
+
const { verbose, query, system, tools, tool_choice, max_tokens, temperature } = params;
|
|
28
|
+
const [input_query, previous_messages] = inputs;
|
|
29
|
+
// Notice that we ignore params.system if previous_message exists.
|
|
30
|
+
const messages = previous_messages && Array.isArray(previous_messages) ? previous_messages : system ? [{ role: "system", content: system }] : [];
|
|
31
|
+
const content = (query ? [query] : []).concat(input_query && typeof input_query === "string" ? [input_query] : []).join("\n");
|
|
32
|
+
if (content) {
|
|
33
|
+
messages.push({
|
|
34
|
+
role: "user",
|
|
35
|
+
content,
|
|
36
|
+
});
|
|
37
|
+
}
|
|
38
|
+
if (verbose) {
|
|
39
|
+
console.log(messages);
|
|
40
|
+
}
|
|
41
|
+
const options = {
|
|
42
|
+
messages,
|
|
43
|
+
model: params.model,
|
|
44
|
+
temperature: temperature ?? 0.7,
|
|
45
|
+
stream: true,
|
|
46
|
+
};
|
|
47
|
+
if (max_tokens) {
|
|
48
|
+
options.max_tokens = max_tokens;
|
|
49
|
+
}
|
|
50
|
+
if (tools) {
|
|
51
|
+
options.tools = tools;
|
|
52
|
+
options.tool_choice = tool_choice ?? "auto";
|
|
53
|
+
}
|
|
54
|
+
const stream = await groq.chat.completions.create(options);
|
|
55
|
+
let lastMessage = null;
|
|
56
|
+
const contents = [];
|
|
57
|
+
for await (const message of stream) {
|
|
58
|
+
const token = message.choices[0].delta.content;
|
|
59
|
+
if (token) {
|
|
60
|
+
if (filterParams && filterParams.streamTokenCallback) {
|
|
61
|
+
filterParams.streamTokenCallback(token);
|
|
62
|
+
}
|
|
63
|
+
contents.push(token);
|
|
64
|
+
}
|
|
65
|
+
lastMessage = message;
|
|
66
|
+
}
|
|
67
|
+
if (lastMessage) {
|
|
68
|
+
lastMessage.choices[0]["message"] = [{ role: "assistant", content: contents.join("") }];
|
|
69
|
+
}
|
|
70
|
+
return lastMessage;
|
|
71
|
+
};
|
|
72
|
+
exports.groqStreamAgent = groqStreamAgent;
|
|
73
|
+
const groqStreamAgentInfo = {
|
|
74
|
+
name: "groqStreamAgent",
|
|
75
|
+
agent: exports.groqStreamAgent,
|
|
76
|
+
mock: exports.groqStreamAgent,
|
|
77
|
+
samples: [],
|
|
78
|
+
description: "Groq Stream Agent",
|
|
79
|
+
category: ["llm"],
|
|
80
|
+
author: "Receptron team",
|
|
81
|
+
repository: "https://github.com/receptron/graphai",
|
|
82
|
+
license: "MIT",
|
|
83
|
+
};
|
|
84
|
+
exports.default = groqStreamAgentInfo;
|
|
@@ -1,2 +1,5 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
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,18 +1,14 @@
|
|
|
1
1
|
"use strict";
|
|
2
|
-
var
|
|
3
|
-
|
|
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
|
-
|
|
18
|
-
|
|
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,3 +1,5 @@
|
|
|
1
1
|
import groqAgent from "../../experimental_agents/llm_agents/groq_agent";
|
|
2
|
+
import groqStreamAgent from "../../experimental_agents/llm_agents/groq_stream_agent";
|
|
2
3
|
import slashGPTAgent from "../../experimental_agents/llm_agents/slashgpt_agent";
|
|
3
|
-
|
|
4
|
+
import openAIAgent from "../../experimental_agents/llm_agents/openai_agent";
|
|
5
|
+
export { groqAgent, groqStreamAgent, slashGPTAgent, openAIAgent };
|
|
@@ -3,8 +3,12 @@ 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.slashGPTAgent = exports.groqAgent = void 0;
|
|
6
|
+
exports.openAIAgent = exports.slashGPTAgent = exports.groqStreamAgent = exports.groqAgent = void 0;
|
|
7
7
|
const groq_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/groq_agent"));
|
|
8
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;
|
|
9
11
|
const slashgpt_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/slashgpt_agent"));
|
|
10
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;
|
|
@@ -39,6 +39,17 @@ const dotProductAgentInfo = {
|
|
|
39
39
|
params: {},
|
|
40
40
|
result: [7, 17, 27],
|
|
41
41
|
},
|
|
42
|
+
{
|
|
43
|
+
inputs: [
|
|
44
|
+
[
|
|
45
|
+
[1, 2],
|
|
46
|
+
[2, 3],
|
|
47
|
+
],
|
|
48
|
+
[1, 2],
|
|
49
|
+
],
|
|
50
|
+
params: {},
|
|
51
|
+
result: [5, 8],
|
|
52
|
+
},
|
|
42
53
|
],
|
|
43
54
|
description: "dotProduct Agent",
|
|
44
55
|
category: [],
|
|
@@ -1,2 +1 @@
|
|
|
1
|
-
export
|
|
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
|
-
|
|
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);
|
|
@@ -10,7 +10,19 @@ declare const sortByValuesAgentInfo: {
|
|
|
10
10
|
mock: AgentFunction<{
|
|
11
11
|
assendant?: boolean | undefined;
|
|
12
12
|
}, any[], any[]>;
|
|
13
|
-
samples:
|
|
13
|
+
samples: ({
|
|
14
|
+
inputs: (string[] | number[])[];
|
|
15
|
+
params: {
|
|
16
|
+
assendant?: undefined;
|
|
17
|
+
};
|
|
18
|
+
result: string[];
|
|
19
|
+
} | {
|
|
20
|
+
inputs: (string[] | number[])[];
|
|
21
|
+
params: {
|
|
22
|
+
assendant: boolean;
|
|
23
|
+
};
|
|
24
|
+
result: string[];
|
|
25
|
+
})[];
|
|
14
26
|
description: string;
|
|
15
27
|
category: never[];
|
|
16
28
|
author: string;
|
|
@@ -31,7 +31,26 @@ const sortByValuesAgentInfo = {
|
|
|
31
31
|
name: "sortByValuesAgent",
|
|
32
32
|
agent: exports.sortByValuesAgent,
|
|
33
33
|
mock: exports.sortByValuesAgent,
|
|
34
|
-
samples: [
|
|
34
|
+
samples: [
|
|
35
|
+
{
|
|
36
|
+
inputs: [
|
|
37
|
+
["banana", "orange", "lemon", "apple"],
|
|
38
|
+
[2, 5, 6, 4],
|
|
39
|
+
],
|
|
40
|
+
params: {},
|
|
41
|
+
result: ["lemon", "orange", "apple", "banana"],
|
|
42
|
+
},
|
|
43
|
+
{
|
|
44
|
+
inputs: [
|
|
45
|
+
["banana", "orange", "lemon", "apple"],
|
|
46
|
+
[2, 5, 6, 4],
|
|
47
|
+
],
|
|
48
|
+
params: {
|
|
49
|
+
assendant: true,
|
|
50
|
+
},
|
|
51
|
+
result: ["banana", "apple", "orange", "lemon"],
|
|
52
|
+
},
|
|
53
|
+
],
|
|
35
54
|
description: "sortByValues Agent",
|
|
36
55
|
category: [],
|
|
37
56
|
author: "Receptron team",
|
|
@@ -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
|
-
|
|
2
|
-
|
|
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.
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
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;
|
|
@@ -3,6 +3,8 @@ 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.fetchAgent = void 0;
|
|
6
|
+
exports.wikipediaAgent = exports.fetchAgent = void 0;
|
|
7
|
+
const wikipedia_1 = __importDefault(require("../../experimental_agents/service_agents/wikipedia"));
|
|
8
|
+
exports.wikipediaAgent = wikipedia_1.default;
|
|
7
9
|
const fetch_agent_1 = __importDefault(require("../../experimental_agents/service_agents/fetch_agent"));
|
|
8
10
|
exports.fetchAgent = fetch_agent_1.default;
|
|
@@ -1,2 +1,3 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
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
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
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;
|