graphai 0.4.0 → 0.4.1

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

@@ -3,7 +3,8 @@ export declare const propertyFilterAgent: AgentFunction<{
     include?: Array<string>;
     exclude?: Array<string>;
     alter?: Record<string, Record<string, string>>;
-    inject?: Record<string, Record<string, any>>;
+    inject?: Array<Record<string, any>>;
+    swap?: Record<string, string>;
 }>;
 declare const propertyFilterAgentInfo: {
     name: string;
@@ -11,13 +12,15 @@ declare const propertyFilterAgentInfo: {
         include?: string[] | undefined;
         exclude?: string[] | undefined;
         alter?: Record<string, Record<string, string>> | undefined;
-        inject?: Record<string, Record<string, any>> | undefined;
+        inject?: Record<string, any>[] | undefined;
+        swap?: Record<string, string> | undefined;
     }>;
     mock: AgentFunction<{
         include?: string[] | undefined;
         exclude?: string[] | undefined;
         alter?: Record<string, Record<string, string>> | undefined;
-        inject?: Record<string, Record<string, any>> | undefined;
+        inject?: Record<string, any>[] | undefined;
+        swap?: Record<string, string> | undefined;
     }>;
     samples: ({
         inputs: (string | {
@@ -32,6 +35,7 @@ declare const propertyFilterAgentInfo: {
             exclude?: undefined;
             alter?: undefined;
             inject?: undefined;
+            swap?: undefined;
         };
         result: {
             color: string;
@@ -50,6 +54,7 @@ declare const propertyFilterAgentInfo: {
             exclude?: undefined;
             alter?: undefined;
             inject?: undefined;
+            swap?: undefined;
         };
         result: {
             color: string;
@@ -68,6 +73,7 @@ declare const propertyFilterAgentInfo: {
             include?: undefined;
             alter?: undefined;
             inject?: undefined;
+            swap?: undefined;
         };
         result: {
             type: string;
@@ -92,6 +98,7 @@ declare const propertyFilterAgentInfo: {
             include?: undefined;
             exclude?: undefined;
             inject?: undefined;
+            swap?: undefined;
         };
         result: {
             color: string;
@@ -110,14 +117,13 @@ declare const propertyFilterAgentInfo: {
         }[])[];
         params: {
             inject: {
-                maker: {
-                    from: number;
-                    index?: undefined;
-                };
-            };
+                propId: string;
+                from: number;
+            }[];
             include?: undefined;
             exclude?: undefined;
             alter?: undefined;
+            swap?: undefined;
         };
         result: {
             color: string;
@@ -136,14 +142,38 @@ declare const propertyFilterAgentInfo: {
         }[])[];
         params: {
             inject: {
-                maker: {
-                    index: number;
-                    from: number;
-                };
+                propId: string;
+                from: number;
+                index: number;
+            }[];
+            include?: undefined;
+            exclude?: undefined;
+            alter?: undefined;
+            swap?: undefined;
+        };
+        result: {
+            color: string;
+            model: string;
+            type: string;
+            maker: string;
+            range: number;
+        }[];
+    } | {
+        inputs: (string | {
+            color: string;
+            model: string;
+            type: string;
+            maker: string;
+            range: number;
+        }[])[];
+        params: {
+            swap: {
+                maker: string;
             };
             include?: undefined;
             exclude?: undefined;
             alter?: undefined;
+            inject?: undefined;
         };
         result: {
             color: string;

package/lib/experimental_agents/data_agents/property_filter_agent.js

@@ -1,17 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.propertyFilterAgent = void 0;
-const applyFilter = (input, index, inputs, include, exclude, alter, inject) => {
+const applyFilter = (input, index, inputs, include, exclude, alter, inject, swap) => {
     const propIds = include ? include : Object.keys(input);
     const excludeSet = new Set(exclude ?? []);
-    return propIds.reduce((tmp, propId) => {
+    const result = propIds.reduce((tmp, propId) => {
         if (!excludeSet.has(propId)) {
-            const injection = inject && inject[propId];
             const mapping = alter && alter[propId];
-            if (injection && (injection.index === undefined || injection.index === index)) {
-                tmp[propId] = inputs[injection.from];
-            }
-            else if (mapping && mapping[input[propId]]) {
+            if (mapping && mapping[input[propId]]) {
                 tmp[propId] = mapping[input[propId]];
             }
             else {
@@ -20,14 +16,29 @@ const applyFilter = (input, index, inputs, include, exclude, alter, inject) => {
         }
         return tmp;
     }, {});
+    if (inject) {
+        inject.forEach((item) => {
+            if (item.index === undefined || item.index === index) {
+                result[item.propId] = inputs[item.from];
+            }
+        });
+    }
+    if (swap) {
+        Object.keys(swap).forEach((key) => {
+            const tmp = result[key];
+            result[key] = result[swap[key]];
+            result[swap[key]] = tmp;
+        });
+    }
+    return result;
 };
 const propertyFilterAgent = async ({ inputs, params }) => {
     const [input] = inputs;
-    const { include, exclude, alter, inject } = params;
+    const { include, exclude, alter, inject, swap } = params;
     if (Array.isArray(input)) {
-        return input.map((item, index) => applyFilter(item, index, inputs, include, exclude, alter, inject));
+        return input.map((item, index) => applyFilter(item, index, inputs, include, exclude, alter, inject, swap));
     }
-    return applyFilter(input, 0, inputs, include, exclude, alter);
+    return applyFilter(input, 0, inputs, include, exclude, alter, inject, swap);
 };
 exports.propertyFilterAgent = propertyFilterAgent;
 const testInputs = [
@@ -73,7 +84,7 @@ const propertyFilterAgentInfo = {
         },
         {
             inputs: testInputs,
-            params: { inject: { maker: { from: 1 } } },
+            params: { inject: [{ propId: "maker", from: 1 }] },
             result: [
                 { color: "red", model: "Model 3", type: "EV", maker: "Tesla Motors", range: 300 },
                 { color: "blue", model: "Model Y", type: "EV", maker: "Tesla Motors", range: 400 },
@@ -81,12 +92,20 @@ const propertyFilterAgentInfo = {
         },
         {
             inputs: testInputs,
-            params: { inject: { maker: { index: 0, from: 1 } } },
+            params: { inject: [{ propId: "maker", from: 1, index: 0 }] },
             result: [
                 { color: "red", model: "Model 3", type: "EV", maker: "Tesla Motors", range: 300 },
                 { color: "blue", model: "Model Y", type: "EV", maker: "Tesla", range: 400 },
             ],
         },
+        {
+            inputs: testInputs,
+            params: { swap: { maker: "model" } },
+            result: [
+                { color: "red", model: "Tesla", type: "EV", maker: "Model 3", range: 300 },
+                { color: "blue", model: "Tesla", type: "EV", maker: "Model Y", range: 400 },
+            ],
+        },
     ],
     description: "Filter properties based on property name either with 'include' or 'exclude'",
     category: ["data"],

package/lib/experimental_agents/graph_agents/map_agent.d.ts

@@ -1,16 +1,16 @@
 import { AgentFunction } from "../../graphai";
 export declare const mapAgent: AgentFunction<{
-    injectionTo?: Array<string>;
+    namedInputs?: Array<string>;
     limit?: number;
 }, Record<string, Array<any>>, any>;
 declare const mapAgentInfo: {
     name: string;
     agent: AgentFunction<{
-        injectionTo?: string[] | undefined;
+        namedInputs?: string[] | undefined;
         limit?: number | undefined;
     }, Record<string, any[]>, any>;
     mock: AgentFunction<{
-        injectionTo?: string[] | undefined;
+        namedInputs?: string[] | undefined;
         limit?: number | undefined;
     }, Record<string, any[]>, any>;
     samples: never[];

package/lib/experimental_agents/graph_agents/map_agent.js

@@ -14,11 +14,11 @@ const mapAgent = async ({ params, inputs, agents, log, taskManager, graphData, a
     if (params.limit && params.limit < input.length) {
         input.length = params.limit; // trim
     }
-    const injectionTo = params.injectionTo ??
+    const namedInputs = params.namedInputs ??
         inputs.map((__input, index) => {
             return `$${index}`;
         });
-    injectionTo.forEach((nodeId) => {
+    namedInputs.forEach((nodeId) => {
         if (nestedGraphData.nodes[nodeId] === undefined) {
             // If the input node does not exist, automatically create a static node
             nestedGraphData.nodes[nodeId] = { value: {} };
@@ -27,7 +27,7 @@ const mapAgent = async ({ params, inputs, agents, log, taskManager, graphData, a
     const graphs = input.map((data) => {
         const graphAI = new graphai_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters: agentFilters || [] });
         // Only the first input will be mapped
-        injectionTo.forEach((injectToNodeId, index) => {
+        namedInputs.forEach((injectToNodeId, index) => {
             graphAI.injectValue(injectToNodeId, index === 0 ? data : inputs[index], "__mapAgent_inputs__");
         });
         return graphAI;

package/lib/experimental_agents/graph_agents/nested_agent.d.ts

@@ -2,15 +2,15 @@ import { AgentFunction } from "../../graphai";
 import { GraphData } from "../../type";
 export declare const getNestedGraphData: (graphData: GraphData | string | undefined, inputs: Array<any>) => GraphData;
 export declare const nestedAgent: AgentFunction<{
-    injectionTo?: Array<string>;
+    namedInputs?: Array<string>;
 }>;
 declare const nestedAgentInfo: {
     name: string;
     agent: AgentFunction<{
-        injectionTo?: string[] | undefined;
+        namedInputs?: string[] | undefined;
     }>;
     mock: AgentFunction<{
-        injectionTo?: string[] | undefined;
+        namedInputs?: string[] | undefined;
     }>;
     samples: never[];
     description: string;

package/lib/experimental_agents/graph_agents/nested_agent.js

@@ -27,8 +27,8 @@ const nestedAgent = async ({ params, inputs, agents, log, taskManager, graphData
         (0, utils_1.assert)(status.concurrency > status.running, `nestedAgent: Concurrency is too low: ${status.concurrency}`);
     }
     const nestedGraphData = (0, exports.getNestedGraphData)(graphData, inputs);
-    const injectionTo = params.injectionTo ?? inputs.map((__input, index) => `$${index}`);
-    injectionTo.forEach((nodeId, index) => {
+    const namedInputs = params.namedInputs ?? inputs.map((__input, index) => `$${index}`);
+    namedInputs.forEach((nodeId, index) => {
         if (nestedGraphData.nodes[nodeId] === undefined) {
             // If the input node does not exist, automatically create a static node
             nestedGraphData.nodes[nodeId] = { value: inputs[index] };

package/lib/experimental_agents/vanilla.d.ts

@@ -4,5 +4,6 @@ export * from "./matrix_agents/vanilla";
 export * from "./test_agents/vanilla";
 export * from "./graph_agents/vanilla";
 export * from "./data_agents/vanilla";
-export * from "./function_agent";
-export * from "./embedding_agent";
+import functionAgent from "./function_agent";
+import stringEmbeddingsAgent from "./embedding_agent";
+export { functionAgent, stringEmbeddingsAgent };

package/lib/experimental_agents/vanilla.js

@@ -14,7 +14,11 @@ var __createBinding = (this && this.__createBinding) || (Object.create ? (functi
 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);
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.stringEmbeddingsAgent = exports.functionAgent = void 0;
 // Please refrain from adding agents that require npm. Those should be added to the index.ts.
 __exportStar(require("./string_agents/vanilla"), exports);
 __exportStar(require("./array_agents/vanilla"), exports);
@@ -22,5 +26,7 @@ __exportStar(require("./matrix_agents/vanilla"), exports);
 __exportStar(require("./test_agents/vanilla"), exports);
 __exportStar(require("./graph_agents/vanilla"), exports);
 __exportStar(require("./data_agents/vanilla"), exports);
-__exportStar(require("./function_agent"), exports);
-__exportStar(require("./embedding_agent"), exports);
+const function_agent_1 = __importDefault(require("./function_agent"));
+exports.functionAgent = function_agent_1.default;
+const embedding_agent_1 = __importDefault(require("./embedding_agent"));
+exports.stringEmbeddingsAgent = embedding_agent_1.default;

package/lib/utils/test_utils.d.ts

@@ -1,28 +1,4 @@
 import { AgentFunctionInfo, AgentFunctionContext, AgentFunction, AgentFilterInfo, ResultData } from "../type";
-export declare const defaultTestContext: {
-    debugInfo: {
-        nodeId: string;
-        retry: number;
-        verbose: boolean;
-    };
-    params: {};
-    filterParams: {};
-    agents: {};
-    log: never[];
-};
-export declare const defaultAgentInfo: {
-    name: string;
-    samples: {
-        inputs: never[];
-        params: {};
-        result: {};
-    }[];
-    description: string;
-    category: never[];
-    author: string;
-    repository: string;
-    license: string;
-};
 export declare const getAgentInfo: (agent: AgentFunction<any, any, any>) => {
     name: string;
     samples: {
@@ -38,5 +14,16 @@ export declare const getAgentInfo: (agent: AgentFunction<any, any, any>) => {
     agent: AgentFunction<any, any, any>;
     mock: AgentFunction<any, any, any>;
 };
+export declare const defaultTestContext: {
+    debugInfo: {
+        nodeId: string;
+        retry: number;
+        verbose: boolean;
+    };
+    params: {};
+    filterParams: {};
+    agents: {};
+    log: never[];
+};
 export declare const agentTestRunner: (agentInfo: AgentFunctionInfo) => Promise<void>;
 export declare const agentFilterRunnerBuilder: (__agentFilters: AgentFilterInfo[]) => (context: AgentFunctionContext, agent: AgentFunction) => Promise<ResultData>;

package/lib/utils/test_utils.js

@@ -3,9 +3,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.agentFilterRunnerBuilder = exports.agentTestRunner = exports.getAgentInfo = exports.defaultAgentInfo = exports.defaultTestContext = void 0;
+exports.agentFilterRunnerBuilder = exports.agentTestRunner = exports.defaultTestContext = exports.getAgentInfo = void 0;
+const utils_1 = require("../utils/utils");
 const node_assert_1 = __importDefault(require("node:assert"));
 const node_test_1 = __importDefault(require("node:test"));
+exports.getAgentInfo = utils_1.agentInfoWrapper;
 exports.defaultTestContext = {
     debugInfo: {
         nodeId: "test",
@@ -17,29 +19,6 @@ exports.defaultTestContext = {
     agents: {},
     log: [],
 };
-exports.defaultAgentInfo = {
-    name: "defaultAgentInfo",
-    samples: [
-        {
-            inputs: [],
-            params: {},
-            result: {},
-        },
-    ],
-    description: "",
-    category: [],
-    author: "",
-    repository: "",
-    license: "",
-};
-const getAgentInfo = (agent) => {
-    return {
-        agent,
-        mock: agent,
-        ...exports.defaultAgentInfo,
-    };
-};
-exports.getAgentInfo = getAgentInfo;
 // for agent
 const agentTestRunner = async (agentInfo) => {
     const { agent, samples, skipTest } = agentInfo;

package/lib/utils/utils.d.ts

@@ -1,7 +1,35 @@
-import { DataSource, ResultData } from "../type";
+import { DataSource, ResultData, AgentFunction } from "../type";
 export declare const sleep: (milliseconds: number) => Promise<unknown>;
 export declare const parseNodeName: (inputNodeId: any, version: number) => 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 | undefined, source: DataSource) => ResultData | undefined;
 export declare const strIntentionalError = "Intentional Error for Debugging";
+export declare const defaultAgentInfo: {
+    name: string;
+    samples: {
+        inputs: never[];
+        params: {};
+        result: {};
+    }[];
+    description: string;
+    category: never[];
+    author: string;
+    repository: string;
+    license: string;
+};
+export declare const agentInfoWrapper: (agent: AgentFunction<any, any, any>) => {
+    name: string;
+    samples: {
+        inputs: never[];
+        params: {};
+        result: {};
+    }[];
+    description: string;
+    category: never[];
+    author: string;
+    repository: string;
+    license: string;
+    agent: AgentFunction<any, any, any>;
+    mock: AgentFunction<any, any, any>;
+};

package/lib/utils/utils.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.strIntentionalError = exports.getDataFromSource = exports.isObject = exports.assert = exports.parseNodeName = exports.sleep = void 0;
+exports.agentInfoWrapper = exports.defaultAgentInfo = exports.strIntentionalError = exports.getDataFromSource = exports.isObject = exports.assert = exports.parseNodeName = exports.sleep = void 0;
 const sleep = async (milliseconds) => {
     return await new Promise((resolve) => setTimeout(resolve, milliseconds));
 };
@@ -86,3 +86,26 @@ const getDataFromSource = (result, source) => {
 };
 exports.getDataFromSource = getDataFromSource;
 exports.strIntentionalError = "Intentional Error for Debugging";
+exports.defaultAgentInfo = {
+    name: "defaultAgentInfo",
+    samples: [
+        {
+            inputs: [],
+            params: {},
+            result: {},
+        },
+    ],
+    description: "",
+    category: [],
+    author: "",
+    repository: "",
+    license: "",
+};
+const agentInfoWrapper = (agent) => {
+    return {
+        agent,
+        mock: agent,
+        ...exports.defaultAgentInfo,
+    };
+};
+exports.agentInfoWrapper = agentInfoWrapper;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [
@@ -40,7 +40,7 @@
     "express": "^4.19.2",
     "globals": "^15.2.0",
     "groq-sdk": "^0.3.3",
-    "openai": "^4.12.4",
+    "openai": "^4.47.1",
     "prettier": "^3.2.5",
     "slashgpt": "^0.0.9",
     "tiktoken": "^1.0.14",