graphai 0.2.0 → 0.2.1

package/lib/experimental_agents/array_agents/pop_agent.d.ts

@@ -1,9 +1,9 @@
 import { AgentFunction } from "../../graphai";
-export declare const popAgent: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
+export declare const popAgent: AgentFunction<Record<string, any>, Record<string, any>, Array<any>>;
 declare const popAgentInfo: {
     name: string;
-    agent: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
-    mock: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
+    agent: AgentFunction<Record<string, any>, Record<string, any>, any[]>;
+    mock: AgentFunction<Record<string, any>, Record<string, any>, any[]>;
     samples: ({
         inputs: string[][];
         params: {};

package/lib/experimental_agents/array_agents/pop_agent.js

@@ -1,14 +1,9 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.popAgent = void 0;
-const deepmerge_1 = __importDefault(require("deepmerge"));
 const popAgent = async (context) => {
     const { inputs } = context;
-    const [array] = (0, deepmerge_1.default)({ inputs }, {}).inputs;
-    // TODO: Validation
+    const array = inputs[0].map((item) => item); // shallow copy
     const item = array.pop();
     return { array, item };
 };

package/lib/experimental_agents/array_agents/push_agent.d.ts

@@ -1,9 +1,9 @@
 import { AgentFunction } from "../../graphai";
-export declare const pushAgent: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
+export declare const pushAgent: AgentFunction<Record<string, any>, Record<string, any>, Array<any>>;
 declare const pushAgentInfo: {
     name: string;
-    agent: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
-    mock: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
+    agent: AgentFunction<Record<string, any>, Record<string, any>, any[]>;
+    mock: AgentFunction<Record<string, any>, Record<string, any>, any[]>;
     samples: {
         inputs: (number | number[])[];
         params: {};

package/lib/experimental_agents/array_agents/push_agent.js

@@ -1,14 +1,9 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.pushAgent = void 0;
-const deepmerge_1 = __importDefault(require("deepmerge"));
 const pushAgent = async ({ inputs }) => {
-    const [array, item] = (0, deepmerge_1.default)({ inputs }, {}).inputs;
-    // TODO: Validation
-    array.push(item);
+    const array = inputs[0].map((item) => item); // shallow copy
+    array.push(inputs[1]);
     return array;
 };
 exports.pushAgent = pushAgent;

package/lib/experimental_agents/array_agents/shift_agent.d.ts

@@ -1,9 +1,9 @@
 import { AgentFunction } from "../../graphai";
-export declare const shiftAgent: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
+export declare const shiftAgent: AgentFunction<Record<string, any>, Record<string, any>, Array<any>>;
 declare const shiftAgentInfo: {
     name: string;
-    agent: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
-    mock: AgentFunction<Record<string, any>, Record<string, any>, Record<string, any>>;
+    agent: AgentFunction<Record<string, any>, Record<string, any>, any[]>;
+    mock: AgentFunction<Record<string, any>, Record<string, any>, any[]>;
     samples: ({
         inputs: number[][];
         params: {};

package/lib/experimental_agents/array_agents/shift_agent.js

@@ -1,14 +1,9 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.shiftAgent = void 0;
-const deepmerge_1 = __importDefault(require("deepmerge"));
 const shiftAgent = async (context) => {
     const { inputs } = context;
-    const [array] = (0, deepmerge_1.default)({ inputs }, {}).inputs;
-    // TODO: Validation
+    const array = inputs[0].map((item) => item); // shallow copy
     const item = array.shift();
     return { array, item };
 };

package/lib/experimental_agents/data_agents/data_object_merge_template_agent.d.ts

@@ -75,7 +75,7 @@ declare const dataObjectMergeTemplateAgentInfo: {
         };
     })[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/data_agents/data_object_merge_template_agent.js

@@ -42,7 +42,7 @@ const dataObjectMergeTemplateAgentInfo = {
         },
     ],
     description: "Merge object",
-    category: [],
+    category: ["data"],
     author: "Satoshi Nakajima",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

package/lib/experimental_agents/data_agents/data_sum_template_agent.d.ts

@@ -10,7 +10,7 @@ declare const dataSumTemplateAgentInfo: {
         result: number;
     }[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/data_agents/data_sum_template_agent.js

@@ -23,7 +23,7 @@ const dataSumTemplateAgentInfo = {
         },
     ],
     description: "Returns the sum of input values",
-    category: [],
+    category: ["data"],
     author: "Satoshi Nakajima",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

package/lib/experimental_agents/data_agents/index.d.ts

@@ -1,3 +1,4 @@
 export { totalAgent } from "../../experimental_agents/data_agents/total_agent";
 export { dataObjectMergeTemplateAgent } from "../../experimental_agents/data_agents/data_object_merge_template_agent";
 export { dataSumTemplateAgent } from "../../experimental_agents/data_agents/data_sum_template_agent";
+export { propertyFilterAgent } from "../../experimental_agents/data_agents/property_filter_agent";

package/lib/experimental_agents/data_agents/index.js

@@ -1,9 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.dataSumTemplateAgent = exports.dataObjectMergeTemplateAgent = exports.totalAgent = void 0;
+exports.propertyFilterAgent = exports.dataSumTemplateAgent = exports.dataObjectMergeTemplateAgent = exports.totalAgent = void 0;
 var total_agent_1 = require("../../experimental_agents/data_agents/total_agent");
 Object.defineProperty(exports, "totalAgent", { enumerable: true, get: function () { return total_agent_1.totalAgent; } });
 var data_object_merge_template_agent_1 = require("../../experimental_agents/data_agents/data_object_merge_template_agent");
 Object.defineProperty(exports, "dataObjectMergeTemplateAgent", { enumerable: true, get: function () { return data_object_merge_template_agent_1.dataObjectMergeTemplateAgent; } });
 var data_sum_template_agent_1 = require("../../experimental_agents/data_agents/data_sum_template_agent");
 Object.defineProperty(exports, "dataSumTemplateAgent", { enumerable: true, get: function () { return data_sum_template_agent_1.dataSumTemplateAgent; } });
+var property_filter_agent_1 = require("../../experimental_agents/data_agents/property_filter_agent");
+Object.defineProperty(exports, "propertyFilterAgent", { enumerable: true, get: function () { return property_filter_agent_1.propertyFilterAgent; } });

package/lib/experimental_agents/data_agents/packages.d.ts

@@ -1,4 +1,5 @@
 import totalAgent from "../../experimental_agents/data_agents/total_agent";
 import dataObjectMergeTemplateAgent from "../../experimental_agents/data_agents/data_object_merge_template_agent";
 import dataSumTemplateAgent from "../../experimental_agents/data_agents/data_sum_template_agent";
-export { totalAgent, dataObjectMergeTemplateAgent, dataSumTemplateAgent };
+import propertyFilterAgent from "../../experimental_agents/data_agents/property_filter_agent";
+export { totalAgent, dataObjectMergeTemplateAgent, dataSumTemplateAgent, propertyFilterAgent };

package/lib/experimental_agents/data_agents/packages.js

@@ -3,10 +3,12 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.dataSumTemplateAgent = exports.dataObjectMergeTemplateAgent = exports.totalAgent = void 0;
+exports.propertyFilterAgent = exports.dataSumTemplateAgent = exports.dataObjectMergeTemplateAgent = exports.totalAgent = void 0;
 const total_agent_1 = __importDefault(require("../../experimental_agents/data_agents/total_agent"));
 exports.totalAgent = total_agent_1.default;
 const data_object_merge_template_agent_1 = __importDefault(require("../../experimental_agents/data_agents/data_object_merge_template_agent"));
 exports.dataObjectMergeTemplateAgent = data_object_merge_template_agent_1.default;
 const data_sum_template_agent_1 = __importDefault(require("../../experimental_agents/data_agents/data_sum_template_agent"));
 exports.dataSumTemplateAgent = data_sum_template_agent_1.default;
+const property_filter_agent_1 = __importDefault(require("../../experimental_agents/data_agents/property_filter_agent"));
+exports.propertyFilterAgent = property_filter_agent_1.default;

package/lib/experimental_agents/data_agents/property_filter_agent.d.ts

@@ -0,0 +1,61 @@
+import { AgentFunction } from "../../graphai";
+export declare const propertyFilterAgent: AgentFunction<{
+    include?: Array<string>;
+    exclude?: Array<string>;
+}>;
+declare const propertyFilterAgentInfo: {
+    name: string;
+    agent: AgentFunction<{
+        include?: string[] | undefined;
+        exclude?: string[] | undefined;
+    }>;
+    mock: AgentFunction<{
+        include?: string[] | undefined;
+        exclude?: string[] | undefined;
+    }>;
+    samples: ({
+        inputs: {
+            color: string;
+            model: string;
+            type: string;
+            maker: string;
+            range: number;
+        }[];
+        params: {
+            include: string[];
+            exclude?: undefined;
+        };
+        result: {
+            color: string;
+            model: string;
+            type?: undefined;
+            maker?: undefined;
+            range?: undefined;
+        };
+    } | {
+        inputs: {
+            color: string;
+            model: string;
+            type: string;
+            maker: string;
+            range: number;
+        }[];
+        params: {
+            exclude: string[];
+            include?: undefined;
+        };
+        result: {
+            type: string;
+            maker: string;
+            range: number;
+            color?: undefined;
+            model?: undefined;
+        };
+    })[];
+    description: string;
+    category: string[];
+    author: string;
+    repository: string;
+    license: string;
+};
+export default propertyFilterAgentInfo;

package/lib/experimental_agents/data_agents/property_filter_agent.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.propertyFilterAgent = void 0;
+const propertyFilterAgent = async ({ inputs, params }) => {
+    const [input] = inputs;
+    const { include, exclude } = params;
+    const propIds = include ? include : Object.keys(input);
+    const excludeSet = new Set(exclude ?? []);
+    return propIds.reduce((tmp, propId) => {
+        if (!excludeSet.has(propId)) {
+            tmp[propId] = input[propId];
+        }
+        return tmp;
+    }, {});
+};
+exports.propertyFilterAgent = propertyFilterAgent;
+const inputs = [{ color: "red", model: "Model 3", type: "EV", maker: "Tesla", range: 300 }];
+const propertyFilterAgentInfo = {
+    name: "propertyFilterAgent",
+    agent: exports.propertyFilterAgent,
+    mock: exports.propertyFilterAgent,
+    samples: [
+        {
+            inputs,
+            params: { include: ["color", "model"] },
+            result: { color: "red", model: "Model 3" },
+        },
+        {
+            inputs,
+            params: { exclude: ["color", "model"] },
+            result: { type: "EV", maker: "Tesla", range: 300 },
+        },
+    ],
+    description: "Filter properties based on property name either with 'include' or 'exclude'",
+    category: ["data"],
+    author: "Receptron team",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = propertyFilterAgentInfo;

package/lib/experimental_agents/data_agents/total_agent.d.ts

@@ -39,7 +39,7 @@ declare const totalAgentInfo: {
         };
     })[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/data_agents/total_agent.js

@@ -44,7 +44,7 @@ const totalAgentInfo = {
         },
     ],
     description: "Returns the sum of input values",
-    category: [],
+    category: ["data"],
     author: "Satoshi Nakajima",
     repository: "https://github.com/snakajima/graphai",
     license: "MIT",

package/lib/experimental_agents/function_agent.d.ts

@@ -0,0 +1,42 @@
+import { AgentFunction } from "../graphai";
+export declare const functionAgent: AgentFunction<{
+    function: (...args: any[]) => any;
+}>;
+declare const functionAgentInfo: {
+    name: string;
+    agent: AgentFunction<{
+        function: (...args: any[]) => any;
+    }>;
+    mock: AgentFunction<{
+        function: (...args: any[]) => any;
+    }>;
+    samples: ({
+        inputs: {
+            model: string;
+            maker: string;
+            range: number;
+            price: number;
+        }[];
+        params: {
+            function: (info: Record<string, any>) => string;
+        };
+        result: string;
+    } | {
+        inputs: string[];
+        params: {
+            function: (str: string) => any;
+        };
+        result: {
+            model: string;
+            maker: string;
+            range: number;
+            price: number;
+        };
+    })[];
+    description: string;
+    category: string[];
+    author: string;
+    repository: string;
+    license: string;
+};
+export default functionAgentInfo;

package/lib/experimental_agents/function_agent.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.functionAgent = void 0;
+const functionAgent = async ({ inputs, params }) => {
+    return params.function(...inputs);
+};
+exports.functionAgent = functionAgent;
+const carInfo = { model: "Model 3", maker: "Tesla", range: 300, price: 35000 };
+const functionAgentInfo = {
+    name: "functionAgent",
+    agent: exports.functionAgent,
+    mock: exports.functionAgent,
+    samples: [
+        {
+            inputs: [carInfo],
+            params: {
+                function: (info) => {
+                    const { model, maker, range, price } = info;
+                    return `A ${maker} ${model} has the range of ${range} miles. It costs $${price}.`;
+                },
+            },
+            result: "A Tesla Model 3 has the range of 300 miles. It costs $35000.",
+        },
+        {
+            inputs: [JSON.stringify(carInfo)],
+            params: {
+                function: (str) => {
+                    return JSON.parse(str);
+                },
+            },
+            result: carInfo,
+        },
+    ],
+    description: "It allows developers to implement the agent function within the graph itself.",
+    category: ["data"],
+    author: "Receptron team",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = functionAgentInfo;

package/lib/experimental_agents/index.d.ts

@@ -5,6 +5,7 @@ export * from "./array_agents";
 export * from "./matrix_agents";
 export * from "./test_agents";
 export * from "./graph_agents";
-export * from "./slashgpt_agent";
+export * from "./llm_agents";
 export * from "./embedding_agent";
 export * from "./token_agent";
+export * from "./function_agent";

package/lib/experimental_agents/index.js

@@ -21,6 +21,7 @@ __exportStar(require("./array_agents"), exports);
 __exportStar(require("./matrix_agents"), exports);
 __exportStar(require("./test_agents"), exports);
 __exportStar(require("./graph_agents"), exports);
-__exportStar(require("./slashgpt_agent"), exports);
+__exportStar(require("./llm_agents"), exports);
 __exportStar(require("./embedding_agent"), exports);
 __exportStar(require("./token_agent"), exports);
+__exportStar(require("./function_agent"), exports);

package/lib/experimental_agents/llm_agents/groq_agent.d.ts

@@ -0,0 +1,26 @@
+import { AgentFunction } from "../../graphai";
+export declare const gloqAgent: AgentFunction<{
+    model: string;
+    query?: string;
+    system?: string;
+}, Record<string, any> | string, string>;
+declare const gloqAgentInfo: {
+    name: string;
+    agent: AgentFunction<{
+        model: string;
+        query?: string | undefined;
+        system?: string | undefined;
+    }, string | Record<string, any>, string>;
+    mock: AgentFunction<{
+        model: string;
+        query?: string | undefined;
+        system?: string | undefined;
+    }, string | Record<string, any>, string>;
+    samples: never[];
+    description: string;
+    category: string[];
+    author: string;
+    repository: string;
+    license: string;
+};
+export default gloqAgentInfo;

package/lib/experimental_agents/llm_agents/groq_agent.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.gloqAgent = void 0;
+const groq_sdk_1 = require("groq-sdk");
+const utils_1 = require("../../utils/utils");
+const groq = process.env.GROQ_API_KEY
+    ? new groq_sdk_1.Groq({
+        apiKey: process.env.GROQ_API_KEY,
+    })
+    : undefined;
+const gloqAgent = async ({ params, inputs }) => {
+    (0, utils_1.assert)(groq !== undefined, "The GROQ_API_KEY environment variable is missing.");
+    const query = params?.query ? [params.query] : [];
+    const content = query.concat(inputs).join("\n");
+    const messages = params?.system ? [{ role: "system", content: params.system }] : [];
+    messages.push({
+        role: "user",
+        content,
+    });
+    console.log(messages);
+    const result = await groq.chat.completions.create({
+        messages,
+        model: params.model,
+    });
+    return result;
+};
+exports.gloqAgent = gloqAgent;
+const gloqAgentInfo = {
+    name: "gloqAgent",
+    agent: exports.gloqAgent,
+    mock: exports.gloqAgent,
+    samples: [],
+    description: "Groq Agent",
+    category: ["llm"],
+    author: "Receptron team",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = gloqAgentInfo;

package/lib/experimental_agents/llm_agents/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./slashgpt_agent";
+export * from "./groq_agent";

package/lib/experimental_agents/llm_agents/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./slashgpt_agent"), exports);
+__exportStar(require("./groq_agent"), exports);

package/lib/experimental_agents/llm_agents/packages.d.ts

@@ -0,0 +1,3 @@
+import gloqAgent from "../../experimental_agents/llm_agents/groq_agent";
+import slashGPTAgent from "../../experimental_agents/llm_agents/slashgpt_agent";
+export { gloqAgent, slashGPTAgent };

package/lib/experimental_agents/llm_agents/packages.js

@@ -0,0 +1,10 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.slashGPTAgent = exports.gloqAgent = void 0;
+const groq_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/groq_agent"));
+exports.gloqAgent = groq_agent_1.default;
+const slashgpt_agent_1 = __importDefault(require("../../experimental_agents/llm_agents/slashgpt_agent"));
+exports.slashGPTAgent = slashgpt_agent_1.default;

package/lib/experimental_agents/{slashgpt_agent.d.ts → llm_agents/slashgpt_agent.d.ts}

@@ -1,31 +1,25 @@
-import { AgentFunction } from "../graphai";
-import { ManifestData } from "slashgpt";
+import { AgentFunction } from "../../graphai";
+import { ManifestData, ChatData } from "slashgpt";
 export declare const slashGPTAgent: AgentFunction<{
     manifest: ManifestData;
     query?: string;
     function_result?: boolean;
-}, {
-    content: string;
-}, string>;
+}, ChatData[], string>;
 declare const slashGPTAgentInfo: {
     name: string;
     agent: AgentFunction<{
         manifest: ManifestData;
         query?: string | undefined;
         function_result?: boolean | undefined;
-    }, {
-        content: string;
-    }, string>;
+    }, ChatData[], string>;
     mock: AgentFunction<{
         manifest: ManifestData;
         query?: string | undefined;
         function_result?: boolean | undefined;
-    }, {
-        content: string;
-    }, string>;
+    }, ChatData[], string>;
     samples: never[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/{slashgpt_agent.js → llm_agents/slashgpt_agent.js}

@@ -16,16 +16,7 @@ const slashGPTAgent = async ({ params, inputs, debugInfo: { verbose, nodeId } })
     const contents = query.concat(inputs);
     session.append_user_question(contents.join("\n"));
     await session.call_loop(() => { });
-    const message = (() => {
-        if (params?.function_result) {
-            return session.history.messages().find((m) => m.role === "function_result");
-        }
-        return session.history.last_message();
-    })();
-    if (message === undefined) {
-        throw new Error("No message in the history");
-    }
-    return message;
+    return session.history.messages();
 };
 exports.slashGPTAgent = slashGPTAgent;
 const slashGPTAgentInfo = {
@@ -34,7 +25,7 @@ const slashGPTAgentInfo = {
     mock: exports.slashGPTAgent,
     samples: [],
     description: "Slash GPT Agent",
-    category: [],
+    category: ["llm"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

package/lib/experimental_agents/packages.d.ts

@@ -5,7 +5,8 @@ export * from "./array_agents/packages";
 export * from "./matrix_agents/packages";
 export * from "./test_agents/packages";
 export * from "./graph_agents/packages";
-import slashGPTAgent from "./slashgpt_agent";
+export * from "./llm_agents/packages";
 import stringEmbeddingsAgent from "./embedding_agent";
 import tokenBoundStringsAgent from "./token_agent";
-export { slashGPTAgent, stringEmbeddingsAgent, tokenBoundStringsAgent };
+import functionAgentInfo from "./function_agent";
+export { stringEmbeddingsAgent, tokenBoundStringsAgent, functionAgentInfo };

package/lib/experimental_agents/packages.js

@@ -17,7 +17,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.tokenBoundStringsAgent = exports.stringEmbeddingsAgent = exports.slashGPTAgent = void 0;
+exports.functionAgentInfo = exports.tokenBoundStringsAgent = exports.stringEmbeddingsAgent = void 0;
 __exportStar(require("./string_agents/packages"), exports);
 __exportStar(require("./sleeper_agents/packages"), exports);
 __exportStar(require("./data_agents/packages"), exports);
@@ -25,9 +25,10 @@ __exportStar(require("./array_agents/packages"), exports);
 __exportStar(require("./matrix_agents/packages"), exports);
 __exportStar(require("./test_agents/packages"), exports);
 __exportStar(require("./graph_agents/packages"), exports);
-const slashgpt_agent_1 = __importDefault(require("./slashgpt_agent"));
-exports.slashGPTAgent = slashgpt_agent_1.default;
+__exportStar(require("./llm_agents/packages"), exports);
 const embedding_agent_1 = __importDefault(require("./embedding_agent"));
 exports.stringEmbeddingsAgent = embedding_agent_1.default;
 const token_agent_1 = __importDefault(require("./token_agent"));
 exports.tokenBoundStringsAgent = token_agent_1.default;
+const function_agent_1 = __importDefault(require("./function_agent"));
+exports.functionAgentInfo = function_agent_1.default;

package/lib/experimental_agents/string_agents/string_template_agent.d.ts

@@ -1,23 +1,21 @@
 import { AgentFunction } from "../../graphai";
 export declare const stringTemplateAgent: AgentFunction<{
     template: string;
-}, Record<string, any> | string, string>;
+}, string, string>;
 declare const stringTemplateAgentInfo: {
     name: string;
     agent: AgentFunction<{
         template: string;
-    }, string | Record<string, any>, string>;
+    }, string, string>;
     mock: AgentFunction<{
         template: string;
-    }, string | Record<string, any>, string>;
+    }, string, string>;
     samples: {
         inputs: string[];
         params: {
             template: string;
         };
-        result: {
-            content: string;
-        };
+        result: string;
     }[];
     description: string;
     category: never[];

package/lib/experimental_agents/string_agents/string_template_agent.js

@@ -7,12 +7,12 @@ const stringTemplateAgent = async ({ params, inputs }) => {
     const content = inputs.reduce((template, input, index) => {
         return template.replace("${" + index + "}", input);
     }, params.template);
-    return { content };
+    return content;
 };
 exports.stringTemplateAgent = stringTemplateAgent;
 const sampleInput = ["hello", "test"];
 const sampleParams = { template: "${0}: ${1}" };
-const sampleResult = { content: "hello: test" };
+const sampleResult = "hello: test";
 // for test and document
 const stringTemplateAgentInfo = {
     name: "stringTemplateAgent",

package/lib/experimental_agents/test_agents/bypass_agent.d.ts

@@ -1,5 +1,8 @@
 import { AgentFunction } from "../../graphai";
 import { AgentFunctionInfo } from "../../type";
-export declare const bypassAgent: AgentFunction;
+export declare const bypassAgent: AgentFunction<{
+    flat?: number;
+    firstElement?: boolean;
+}>;
 declare const bypassAgentInfo: AgentFunctionInfo;
 export default bypassAgentInfo;

package/lib/experimental_agents/test_agents/bypass_agent.js

@@ -1,11 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.bypassAgent = void 0;
-const bypassAgent = async (context) => {
-    if (context.inputs.length === 1) {
-        return context.inputs[0];
+const bypassAgent = async ({ params, inputs }) => {
+    if (params && params.firstElement) {
+        return inputs[0];
     }
-    return context.inputs;
+    if (params && params.flat) {
+        return inputs.flat(params.flat || 1);
+    }
+    return inputs;
 };
 exports.bypassAgent = bypassAgent;
 // for test and document
@@ -17,13 +20,35 @@ const bypassAgentInfo = {
         {
             inputs: [{ a: "123" }],
             params: {},
-            result: { a: "123" },
+            result: [{ a: "123" }],
         },
         {
-            inputs: [{ a: "123" }, { b: "abc" }],
+            inputs: [
+                [{ a: "123" }, { b: "abc" }],
+                [{ c: "987" }, { d: "xyz" }],
+            ],
             params: {},
+            result: [
+                [{ a: "123" }, { b: "abc" }],
+                [{ c: "987" }, { d: "xyz" }],
+            ],
+        },
+        {
+            inputs: [
+                [{ a: "123" }, { b: "abc" }],
+                [{ c: "987" }, { d: "xyz" }],
+            ],
+            params: { firstElement: true },
             result: [{ a: "123" }, { b: "abc" }],
         },
+        {
+            inputs: [
+                [{ a: "123" }, { b: "abc" }],
+                [{ c: "987" }, { d: "xyz" }],
+            ],
+            params: { flat: 1 },
+            result: [{ a: "123" }, { b: "abc" }, { c: "987" }, { d: "xyz" }],
+        },
     ],
     description: "bypass agent",
     category: [],

package/lib/graphai.d.ts

@@ -5,6 +5,7 @@ import { ComputedNode, StaticNode } from "./node";
 import { TaskManager } from "./task_manager";
 type GraphNodes = Record<string, ComputedNode | StaticNode>;
 export declare class GraphAI {
+    private readonly version;
     private readonly graphId;
     private readonly data;
     private readonly loop?;

package/lib/graphai.js

@@ -40,7 +40,7 @@ class GraphAI {
     }
     getValueFromResults(key, results) {
         const source = (0, utils_1.parseNodeName)(key);
-        return (0, utils_1.getDataFromSource)(results[source.nodeId], source.propIds);
+        return (0, utils_1.getDataFromSource)(source.nodeId ? results[source.nodeId] : undefined, source);
     }
     // for static
     initializeNodes(previousResults) {
@@ -68,6 +68,10 @@ class GraphAI {
         this.logs = [];
         this.onLogCallback = (__log, __isUpdate) => { };
         this.repeatCount = 0;
+        if (!data.version) {
+            console.log("------------ no version");
+        }
+        this.version = data.version ?? 0.2;
         this.graphId = URL.createObjectURL(new Blob()).slice(-36);
         this.data = data;
         this.callbackDictonary = callbackDictonary;
@@ -229,8 +233,8 @@ class GraphAI {
     }
     resultsOf(sources) {
         return sources.map((source) => {
-            const { result } = this.nodes[source.nodeId];
-            return (0, utils_1.getDataFromSource)(result, source.propIds);
+            const { result } = source.nodeId ? this.nodes[source.nodeId] : { result: undefined };
+            return (0, utils_1.getDataFromSource)(result, source);
         });
     }
 }

package/lib/node.js

@@ -44,8 +44,8 @@ class ComputedNode extends Node {
         this.timeout = data.timeout;
         this.isResult = data.isResult ?? false;
         this.anyInput = data.anyInput ?? false;
-        this.dataSources = (data.inputs ?? []).map(utils_2.parseNodeName);
-        this.pendings = new Set(this.dataSources.map((source) => source.nodeId));
+        this.dataSources = (data.inputs ?? []).map((input) => (0, utils_2.parseNodeName)(input));
+        this.pendings = new Set(this.dataSources.filter((source) => source.nodeId).map((source) => source.nodeId));
         this.log.initForComputedNode(this);
     }
     isReadyNode() {

package/lib/transaction_log.js

@@ -40,7 +40,7 @@ class TransactionLog {
         this.state = node.state;
         this.retryCount = node.retryCount > 0 ? node.retryCount : undefined;
         this.startTime = transactionId;
-        this.inputs = node.dataSources.map((source) => source.nodeId);
+        this.inputs = node.dataSources.filter((source) => source.nodeId).map((source) => source.nodeId);
         this.inputsData = inputs.length > 0 ? inputs : undefined;
         graph.setLoopLog(this);
         graph.appendLog(this);

package/lib/type.d.ts

@@ -17,7 +17,8 @@ export type ResultDataDictonary<ResultType = DefaultResultData> = Record<string,
 export type DefaultParamsType = Record<string, any>;
 export type NodeDataParams<ParamsType = DefaultParamsType> = ParamsType;
 export type DataSource = {
-    nodeId: string;
+    nodeId?: string;
+    value?: any;
     propIds?: string[];
 };
 export type StaticNodeData = {
@@ -27,7 +28,7 @@ export type StaticNodeData = {
 };
 export type ComputedNodeData = {
     agentId: string;
-    inputs?: Array<string>;
+    inputs?: Array<any>;
     anyInput?: boolean;
     params?: NodeDataParams;
     retry?: number;
@@ -41,6 +42,7 @@ export type LoopData = {
     while?: string;
 };
 export type GraphData = {
+    version?: number;
     nodes: Record<string, NodeData>;
     concurrency?: number;
     loop?: LoopData;

package/lib/utils/test_agents.js

@@ -17,4 +17,5 @@ exports.defaultTestAgents = {
     copy2ArrayAgent: experimental_agents_1.copy2ArrayAgent,
     pushAgent: experimental_agents_1.pushAgent,
     popAgent: experimental_agents_1.popAgent,
+    shiftAgent: experimental_agents_1.shiftAgent,
 };

package/lib/utils/test_utils.js

@@ -18,17 +18,22 @@ exports.defaultTestContext = {
 };
 // for agent
 const agentTestRunner = async (agentInfo) => {
-    (0, node_test_1.default)(`test ${agentInfo.name}`, async () => {
-        const { agent, samples } = agentInfo;
-        for await (const sample of samples) {
-            const { params, inputs, result } = sample;
-            const actual = await agent({
-                ...exports.defaultTestContext,
-                params,
-                inputs,
+    const { agent, samples } = agentInfo;
+    if (samples.length === 0) {
+        console.log(`test ${agentInfo.name}: No test`);
+    }
+    else {
+        for await (const sampleKey of samples.keys()) {
+            (0, node_test_1.default)(`test ${agentInfo.name} ${sampleKey}`, async () => {
+                const { params, inputs, result } = samples[sampleKey];
+                const actual = await agent({
+                    ...exports.defaultTestContext,
+                    params,
+                    inputs,
+                });
+                node_assert_1.default.deepStrictEqual(actual, result);
             });
-            node_assert_1.default.deepStrictEqual(actual, result);
         }
-    });
+    }
 };
 exports.agentTestRunner = agentTestRunner;

package/lib/utils/utils.d.ts

@@ -1,7 +1,7 @@
 import { DataSource, ResultData } from "../type";
 export declare const sleep: (milliseconds: number) => Promise<unknown>;
-export declare const parseNodeName: (inputNodeId: string) => DataSource;
+export declare const parseNodeName: (inputNodeId: any) => DataSource;
 export declare function assert(condition: boolean, message: string, isWarn?: boolean): asserts condition;
 export declare const isObject: (x: unknown) => boolean;
-export declare const getDataFromSource: (result: ResultData, propIds: string[] | undefined) => ResultData | undefined;
+export declare const getDataFromSource: (result: ResultData | undefined, source: DataSource) => ResultData | undefined;
 export declare const strIntentionalError = "Intentional Error for Debugging";

package/lib/utils/utils.js

@@ -6,11 +6,19 @@ const sleep = async (milliseconds) => {
 };
 exports.sleep = sleep;
 const parseNodeName = (inputNodeId) => {
-    const parts = inputNodeId.split(".");
-    if (parts.length == 1) {
-        return { nodeId: parts[0] };
+    if (typeof inputNodeId === "string") {
+        const regex = /^"(.*)"$/;
+        const match = inputNodeId.match(regex);
+        if (match) {
+            return { value: match[1] }; // string literal
+        }
+        const parts = inputNodeId.split(".");
+        if (parts.length == 1) {
+            return { nodeId: parts[0] };
+        }
+        return { nodeId: parts[0], propIds: parts.slice(1) };
     }
-    return { nodeId: parts[0], propIds: parts.slice(1) };
+    return { value: inputNodeId }; // non-string literal
 };
 exports.parseNodeName = parseNodeName;
 function assert(condition, message, isWarn = false) {
@@ -41,16 +49,22 @@ const getNestedData = (result, propId) => {
     assert((0, exports.isObject)(result), "result is not object.");
     return result[propId];
 };
-const getDataFromSource = (result, propIds) => {
+const innerGetDataFromSource = (result, propIds) => {
     if (result && propIds && propIds.length > 0) {
         const propId = propIds[0];
         const ret = getNestedData(result, propId);
         if (propIds.length > 1) {
-            return (0, exports.getDataFromSource)(ret, propIds.slice(1));
+            return innerGetDataFromSource(ret, propIds.slice(1));
         }
         return ret;
     }
     return result;
 };
+const getDataFromSource = (result, source) => {
+    if (!source.nodeId) {
+        return source.value;
+    }
+    return innerGetDataFromSource(result, source.propIds);
+};
 exports.getDataFromSource = getDataFromSource;
 exports.strIntentionalError = "Intentional Error for Debugging";

package/lib/validators/relation_validator.js

@@ -12,13 +12,15 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
         pendings[computedNodeId] = new Set();
         if (nodeData.inputs) {
             nodeData.inputs.forEach((inputNodeId) => {
-                const input = (0, utils_1.parseNodeName)(inputNodeId).nodeId;
-                if (!nodeIds.has(input)) {
-                    throw new Error(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${input}`);
+                const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId).nodeId;
+                if (sourceNodeId) {
+                    if (!nodeIds.has(sourceNodeId)) {
+                        throw new Error(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${sourceNodeId}`);
+                    }
+                    waitlist[sourceNodeId] === undefined && (waitlist[sourceNodeId] = new Set());
+                    pendings[computedNodeId].add(sourceNodeId);
+                    waitlist[sourceNodeId].add(computedNodeId);
                 }
-                waitlist[input] === undefined && (waitlist[input] = new Set());
-                pendings[computedNodeId].add(input);
-                waitlist[input].add(computedNodeId);
             });
         }
     });
@@ -28,6 +30,9 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
         const update = nodeData.update;
         if (update) {
             const updateNodeId = (0, utils_1.parseNodeName)(update).nodeId;
+            if (!updateNodeId) {
+                throw new Error("Update it a literal");
+            }
             if (!nodeIds.has(updateNodeId)) {
                 throw new Error(`Update not match: NodeId ${staticNodeId}, update: ${update}`);
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [
@@ -38,6 +38,7 @@
     "eslint": "^7.32.0 || ^8.2.0",
     "eslint-plugin-import": "^2.25.2",
     "express": "^4.19.2",
+    "groq-sdk": "^0.3.3",
     "openai": "^4.12.4",
     "prettier": "^3.0.3",
     "slashgpt": "^0.0.8",