graphai 0.4.1 → 0.4.2

package/README.md

@@ -418,7 +418,7 @@ Step 1. Install git, node and yarn
 Step 2. Clone the project and install necessary node modules
 
 ```
-git clone git@github.com:snakajima/graphai.git
+git clone git@github.com:receptron/graphai.git
 cd graphai
 yarn install
 ```

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const popAgent: AgentFunction<Record<string, any>, Record<string, any>, Array<any>>;
 declare const popAgentInfo: {
     name: string;

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

@@ -1,14 +1,28 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 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>, any[]>;
     mock: AgentFunction<Record<string, any>, Record<string, any>, any[]>;
-    samples: {
+    samples: ({
         inputs: (number | number[])[];
         params: {};
         result: number[];
-    }[];
+    } | {
+        inputs: ({
+            apple: number;
+        }[] | {
+            lemon: number;
+        })[];
+        params: {};
+        result: ({
+            apple: number;
+            lemon?: undefined;
+        } | {
+            lemon: number;
+            apple?: undefined;
+        })[];
+    })[];
     description: string;
     category: never[];
     author: string;

package/lib/experimental_agents/array_agents/push_agent.js

@@ -3,7 +3,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.pushAgent = void 0;
 const pushAgent = async ({ inputs }) => {
     const array = inputs[0].map((item) => item); // shallow copy
-    array.push(inputs[1]);
+    inputs.forEach((input, index) => {
+        if (index > 0) {
+            array.push(input);
+        }
+    });
     return array;
 };
 exports.pushAgent = pushAgent;
@@ -17,6 +21,16 @@ const pushAgentInfo = {
             params: {},
             result: [1, 2, 3],
         },
+        {
+            inputs: [[1, 2], 3, 4, 5],
+            params: {},
+            result: [1, 2, 3, 4, 5],
+        },
+        {
+            inputs: [[{ apple: 1 }], { lemon: 2 }],
+            params: {},
+            result: [{ apple: 1 }, { lemon: 2 }],
+        },
     ],
     description: "push Agent",
     category: [],

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const shiftAgent: AgentFunction<Record<string, any>, Record<string, any>, Array<any>>;
 declare const shiftAgentInfo: {
     name: string;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const copyAgent: AgentFunction;
 declare const copyAgentInfo: {
     name: string;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const dataObjectMergeTemplateAgent: AgentFunction;
 declare const dataObjectMergeTemplateAgentInfo: {
     name: string;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const dataSumTemplateAgent: AgentFunction<Record<string, any>, number, number>;
 declare const dataSumTemplateAgentInfo: {
     name: string;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const propertyFilterAgent: AgentFunction<{
     include?: Array<string>;
     exclude?: Array<string>;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const totalAgent: AgentFunction<Record<never, never>, Record<string, number>>;
 declare const totalAgentInfo: {
     name: string;

package/lib/experimental_agents/embedding_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../graphai";
+import { AgentFunction } from "../index";
 export declare const stringEmbeddingsAgent: AgentFunction<{
     model?: string;
 }, any, Array<string> | string>;

package/lib/experimental_agents/function_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../graphai";
+import { AgentFunction } from "../index";
 export declare const functionAgent: AgentFunction<{
     function: (...args: any[]) => any;
 }>;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const mapAgent: AgentFunction<{
     namedInputs?: Array<string>;
     limit?: number;

package/lib/experimental_agents/graph_agents/map_agent.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.mapAgent = void 0;
-const graphai_1 = require("../../graphai");
+const index_1 = require("../../index");
 const utils_1 = require("../../utils/utils");
 const nested_agent_1 = require("./nested_agent");
 const mapAgent = async ({ params, inputs, agents, log, taskManager, graphData, agentFilters }) => {
@@ -25,7 +25,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 || [] });
+        const graphAI = new index_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters: agentFilters || [] });
         // Only the first input will be mapped
         namedInputs.forEach((injectToNodeId, index) => {
             graphAI.injectValue(injectToNodeId, index === 0 ? data : inputs[index], "__mapAgent_inputs__");

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { GraphData } from "../../type";
 export declare const getNestedGraphData: (graphData: GraphData | string | undefined, inputs: Array<any>) => GraphData;
 export declare const nestedAgent: AgentFunction<{

package/lib/experimental_agents/graph_agents/nested_agent.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.nestedAgent = exports.getNestedGraphData = void 0;
-const graphai_1 = require("../../graphai");
+const index_1 = require("../../index");
 const utils_1 = require("../../utils/utils");
 // This function allows us to use one of inputs as the graph data for this nested agent,
 // which is equivalent to "eval" of JavaScript.
@@ -38,7 +38,7 @@ const nestedAgent = async ({ params, inputs, agents, log, taskManager, graphData
             nestedGraphData.nodes[nodeId]["value"] = inputs[index];
         }
     });
-    const graphAI = new graphai_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters });
+    const graphAI = new index_1.GraphAI(nestedGraphData, agents || {}, { taskManager, agentFilters });
     const results = await graphAI.run(false);
     log?.push(...graphAI.transactionLogs());
     return results;

package/lib/experimental_agents/index.d.ts

@@ -7,6 +7,7 @@ export * from "./data_agents";
 export * from "./sleeper_agents";
 export * from "./llm_agents";
 export * from "./service_agents";
+export * from "./input_agents";
 import functionAgent from "./function_agent";
 import stringEmbeddingsAgent from "./embedding_agent";
 import tokenBoundStringsAgent from "./token_agent";

package/lib/experimental_agents/index.js

@@ -28,6 +28,7 @@ __exportStar(require("./data_agents"), exports);
 __exportStar(require("./sleeper_agents"), exports);
 __exportStar(require("./llm_agents"), exports);
 __exportStar(require("./service_agents"), exports);
+__exportStar(require("./input_agents"), exports);
 //  TODO sub folder
 const function_agent_1 = __importDefault(require("./function_agent"));
 exports.functionAgent = function_agent_1.default;

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

@@ -0,0 +1,2 @@
+import textInputAgent from "../../experimental_agents/input_agents/text_input_agent";
+export { textInputAgent };

package/lib/experimental_agents/input_agents/index.js

@@ -0,0 +1,8 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textInputAgent = void 0;
+const text_input_agent_1 = __importDefault(require("../../experimental_agents/input_agents/text_input_agent"));
+exports.textInputAgent = text_input_agent_1.default;

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

@@ -0,0 +1,2 @@
+import textInputAgent from "../../experimental_agents/input_agents/text_input_agent";
+export { textInputAgent };

package/lib/experimental_agents/input_agents/packages.js

@@ -0,0 +1,8 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textInputAgent = void 0;
+const text_input_agent_1 = __importDefault(require("../../experimental_agents/input_agents/text_input_agent"));
+exports.textInputAgent = text_input_agent_1.default;

package/lib/experimental_agents/input_agents/text_input_agent.d.ts

@@ -0,0 +1,26 @@
+import { AgentFunction } from "../../index";
+export declare const textInputAgent: AgentFunction<{
+    message?: string;
+}, string | {
+    [x: string]: string;
+}>;
+declare const textInputAgentInfo: {
+    name: string;
+    agent: AgentFunction<{
+        message?: string | undefined;
+    }, string | {
+        [x: string]: string;
+    }>;
+    mock: AgentFunction<{
+        message?: string | undefined;
+    }, string | {
+        [x: string]: string;
+    }>;
+    samples: never[];
+    description: string;
+    category: string[];
+    author: string;
+    repository: string;
+    license: string;
+};
+export default textInputAgentInfo;

package/lib/experimental_agents/input_agents/text_input_agent.js

@@ -0,0 +1,23 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textInputAgent = void 0;
+const input_1 = __importDefault(require("@inquirer/input"));
+const textInputAgent = async ({ params }) => {
+    return await (0, input_1.default)({ message: params.message ?? "Enter" });
+};
+exports.textInputAgent = textInputAgent;
+const textInputAgentInfo = {
+    name: "textInputAgent",
+    agent: exports.textInputAgent,
+    mock: exports.textInputAgent,
+    samples: [],
+    description: "Text Input Agent",
+    category: ["array"],
+    author: "Receptron team",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = textInputAgentInfo;

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

@@ -1,15 +1,16 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { Groq } from "groq-sdk";
 export declare const groqAgent: AgentFunction<{
     model: string;
     query?: string;
     system?: string;
     verbose?: boolean;
-    tools?: Record<string, Groq.Chat.CompletionCreateParams.Tool>;
+    tools?: Groq.Chat.CompletionCreateParams.Tool[];
     temperature?: number;
     max_tokens?: number;
-    tool_choice?: string | Record<string, Groq.Chat.CompletionCreateParams.ToolChoice>;
-}, Groq.Chat.ChatCompletion, string | Array<Groq.Chat.CompletionCreateParams.Message>>;
+    tool_choice?: Groq.Chat.CompletionCreateParams.ToolChoice;
+    isStreaming?: boolean;
+}, any, string | Array<Groq.Chat.CompletionCreateParams.Message>>;
 declare const groqAgentInfo: {
     name: string;
     agent: AgentFunction<{
@@ -17,21 +18,23 @@ declare const groqAgentInfo: {
         query?: string | undefined;
         system?: string | undefined;
         verbose?: boolean | undefined;
-        tools?: Record<string, Groq.Chat.Completions.CompletionCreateParams.Tool> | undefined;
+        tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
         temperature?: number | undefined;
         max_tokens?: number | undefined;
-        tool_choice?: string | Record<string, Groq.Chat.Completions.CompletionCreateParams.ToolChoice> | undefined;
-    }, Groq.Chat.Completions.ChatCompletion, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
+        tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
+        isStreaming?: boolean | undefined;
+    }, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
     mock: AgentFunction<{
         model: string;
         query?: string | undefined;
         system?: string | undefined;
         verbose?: boolean | undefined;
-        tools?: Record<string, Groq.Chat.Completions.CompletionCreateParams.Tool> | undefined;
+        tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
         temperature?: number | undefined;
         max_tokens?: number | undefined;
-        tool_choice?: string | Record<string, Groq.Chat.Completions.CompletionCreateParams.ToolChoice> | undefined;
-    }, Groq.Chat.Completions.ChatCompletion, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
+        tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
+        isStreaming?: boolean | undefined;
+    }, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
     samples: never[];
     description: string;
     category: string[];

package/lib/experimental_agents/llm_agents/groq_agent.js

@@ -22,9 +22,9 @@ const groq = process.env.GROQ_API_KEY ? new groq_sdk_1.Groq({ apiKey: process.en
 //
 // https://console.groq.com/docs/quickstart
 //
-const groqAgent = async ({ params, inputs }) => {
+const groqAgent = async ({ params, inputs, filterParams }) => {
     (0, utils_1.assert)(groq !== undefined, "The GROQ_API_KEY environment variable is missing.");
-    const { verbose, query, system, tools, tool_choice, max_tokens, temperature } = params;
+    const { verbose, query, system, tools, tool_choice, max_tokens, temperature, isStreaming } = params;
     const [input_query, previous_messages] = inputs;
     // Notice that we ignore params.system if previous_message exists.
     const messages = previous_messages && Array.isArray(previous_messages) ? previous_messages : system ? [{ role: "system", content: system }] : [];
@@ -38,11 +38,18 @@ const groqAgent = async ({ params, inputs }) => {
     if (verbose) {
         console.log(messages);
     }
-    const options = {
+    const streamOption = {
         messages,
         model: params.model,
         temperature: temperature ?? 0.7,
+        stream: true,
     };
+    const nonStreamOption = {
+        messages,
+        model: params.model,
+        temperature: temperature ?? 0.7,
+    };
+    const options = isStreaming ? streamOption : nonStreamOption;
     if (max_tokens) {
         options.max_tokens = max_tokens;
     }
@@ -50,8 +57,28 @@ const groqAgent = async ({ params, inputs }) => {
         options.tools = tools;
         options.tool_choice = tool_choice ?? "auto";
     }
-    const result = await groq.chat.completions.create(options);
-    return result;
+    if (!options.stream) {
+        const result = await groq.chat.completions.create(options);
+        return result;
+    }
+    // streaming
+    const stream = await groq.chat.completions.create(options);
+    let lastMessage = null;
+    const contents = [];
+    for await (const message of stream) {
+        const token = message.choices[0].delta.content;
+        if (token) {
+            if (filterParams && filterParams.streamTokenCallback) {
+                filterParams.streamTokenCallback(token);
+            }
+            contents.push(token);
+        }
+        lastMessage = message;
+    }
+    if (lastMessage) {
+        lastMessage.choices[0]["message"] = [{ role: "assistant", content: contents.join("") }];
+    }
+    return lastMessage;
 };
 exports.groqAgent = groqAgent;
 const groqAgentInfo = {

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { Groq } from "groq-sdk";
 export declare const groqStreamAgent: AgentFunction<{
     model: string;

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

@@ -1,8 +1,10 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const openAIAgent: AgentFunction<{
     model?: string;
     query?: string;
     system?: string;
+    tools?: any;
+    tool_choice?: any;
     verbose?: boolean;
     temperature?: number;
 }, Record<string, any> | string, string | Array<any>>;
@@ -19,6 +21,8 @@ declare const openaiAgentInfo: {
         model?: string | undefined;
         query?: string | undefined;
         system?: string | undefined;
+        tools?: any;
+        tool_choice?: any;
         verbose?: boolean | undefined;
         temperature?: number | undefined;
     }, string | Record<string, any>, string | any[]>;

package/lib/experimental_agents/llm_agents/openai_agent.js

@@ -25,6 +25,8 @@ const openAIAgent = async ({ filterParams, params, inputs }) => {
     const chatStream = await openai.beta.chat.completions.stream({
         model: params.model || "gpt-3.5-turbo",
         messages,
+        tools: params.tools,
+        tool_choice: params.tool_choice,
         temperature: temperature ?? 0.7,
         stream: true,
     });

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { ManifestData, ChatData } from "slashgpt";
 export declare const slashGPTAgent: AgentFunction<{
     manifest: ManifestData;

package/lib/experimental_agents/matrix_agents/dot_product_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const dotProductAgent: AgentFunction<Record<never, never>, Array<number>, Array<Array<number>> | Array<number>>;
 declare const dotProductAgentInfo: {
     name: string;

package/lib/experimental_agents/matrix_agents/sort_by_values_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const sortByValuesAgent: AgentFunction<{
     assendant?: boolean;
 }, Array<any>, Array<any>>;

package/lib/experimental_agents/service_agents/fetch_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const fetchAgent: AgentFunction<{
     debug?: boolean;
     type?: string;

package/lib/experimental_agents/service_agents/wikipedia.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const wikipediaAgent: AgentFunction<{
     lang?: string;
     summary?: boolean;

package/lib/experimental_agents/sleeper_agents/sleeper_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const sleeperAgent: AgentFunction<{
     duration?: number;
     value?: Record<string, any>;

package/lib/experimental_agents/sleeper_agents/sleeper_agent_debug.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const sleeperAgentDebug: AgentFunction<{
     duration: number;
     value?: Record<string, any>;

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

@@ -1,40 +1,9 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const stringSplitterAgent: AgentFunction<{
     chunkSize?: number;
     overlap?: number;
 }, {
     contents: Array<string>;
 }, string>;
-declare const stringSplitterAgentInfo: {
-    name: string;
-    agent: AgentFunction<{
-        chunkSize?: number | undefined;
-        overlap?: number | undefined;
-    }, {
-        contents: Array<string>;
-    }, string>;
-    mock: AgentFunction<{
-        chunkSize?: number | undefined;
-        overlap?: number | undefined;
-    }, {
-        contents: Array<string>;
-    }, string>;
-    samples: {
-        inputs: string[];
-        params: {
-            chunkSize: number;
-        };
-        result: {
-            contents: string[];
-            count: number;
-            chunkSize: number;
-            overlap: number;
-        };
-    }[];
-    description: string;
-    category: never[];
-    author: string;
-    repository: string;
-    license: string;
-};
+declare const stringSplitterAgentInfo: AgentFunctionInfo;
 export default stringSplitterAgentInfo;

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

@@ -1,22 +1,64 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const stringTemplateAgent: AgentFunction<{
-    template: string;
-}, string, string>;
+    template: any;
+}, any, string>;
 declare const stringTemplateAgentInfo: {
     name: string;
     agent: AgentFunction<{
-        template: string;
-    }, string, string>;
+        template: any;
+    }, any, string>;
     mock: AgentFunction<{
-        template: string;
-    }, string, string>;
-    samples: {
+        template: any;
+    }, any, string>;
+    samples: ({
         inputs: string[];
         params: {
             template: string;
         };
         result: string;
-    }[];
+    } | {
+        inputs: string[];
+        params: {
+            template: string[];
+        };
+        result: string[];
+    } | {
+        inputs: string[];
+        params: {
+            template: {
+                apple: string;
+                lemon: string;
+            };
+        };
+        result: {
+            apple: string;
+            lemon: string;
+        };
+    } | {
+        inputs: string[];
+        params: {
+            template: {
+                apple: string;
+                lemon: string;
+            }[];
+        };
+        result: {
+            apple: string;
+            lemon: string;
+        }[];
+    } | {
+        inputs: string[];
+        params: {
+            template: {
+                apple: string;
+                lemon: string[];
+            };
+        };
+        result: {
+            apple: string;
+            lemon: string[];
+        };
+    })[];
     description: string;
     category: never[];
     author: string;

package/lib/experimental_agents/string_agents/string_template_agent.js

@@ -1,18 +1,25 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.stringTemplateAgent = void 0;
-// see example
-//  tests/agents/test_string_agent.ts
+const processTemplate = (template, match, input) => {
+    if (typeof template === "string") {
+        return template.replace(match, input);
+    }
+    else if (Array.isArray(template)) {
+        return template.map((item) => processTemplate(item, match, input));
+    }
+    return Object.keys(template).reduce((tmp, key) => {
+        tmp[key] = processTemplate(template[key], match, input);
+        return tmp;
+    }, {});
+};
 const stringTemplateAgent = async ({ params, inputs }) => {
-    const content = inputs.reduce((template, input, index) => {
-        return template.replace("${" + index + "}", input);
+    return inputs.reduce((template, input, index) => {
+        return processTemplate(template, "${" + index + "}", input);
     }, params.template);
-    return content;
 };
 exports.stringTemplateAgent = stringTemplateAgent;
 const sampleInput = ["hello", "test"];
-const sampleParams = { template: "${0}: ${1}" };
-const sampleResult = "hello: test";
 // for test and document
 const stringTemplateAgentInfo = {
     name: "stringTemplateAgent",
@@ -21,8 +28,28 @@ const stringTemplateAgentInfo = {
     samples: [
         {
             inputs: sampleInput,
-            params: sampleParams,
-            result: sampleResult,
+            params: { template: "${0}: ${1}" },
+            result: "hello: test",
+        },
+        {
+            inputs: sampleInput,
+            params: { template: ["${0}: ${1}", "${1}: ${0}"] },
+            result: ["hello: test", "test: hello"],
+        },
+        {
+            inputs: sampleInput,
+            params: { template: { apple: "${0}", lemon: "${1}" } },
+            result: { apple: "hello", lemon: "test" },
+        },
+        {
+            inputs: sampleInput,
+            params: { template: [{ apple: "${0}", lemon: "${1}" }] },
+            result: [{ apple: "hello", lemon: "test" }],
+        },
+        {
+            inputs: sampleInput,
+            params: { template: { apple: "${0}", lemon: ["${1}"] } },
+            result: { apple: "hello", lemon: ["test"] },
         },
     ],
     description: "Template agent",

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { AgentFunctionInfo } from "../../type";
 export declare const bypassAgent: AgentFunction<{
     flat?: number;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { AgentFunctionInfo } from "../../type";
 export declare const copy2ArrayAgent: AgentFunction<{
     count: number;

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const copyMessageAgent: AgentFunction<{
     count: number;
     message: string;

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const countingAgent: AgentFunction<{
     count: number;
 }, {

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const echoAgent: AgentFunction;
 declare const echoAgentInfo: AgentFunctionInfo;
 export default echoAgentInfo;

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const mergeNodeIdAgent: AgentFunction;
 declare const mergeNodeIdAgentInfo: AgentFunctionInfo;
 export default mergeNodeIdAgentInfo;

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const streamMockAgent: AgentFunction;
 declare const streamMockAgentInfo: AgentFunctionInfo;
 export default streamMockAgentInfo;

package/lib/experimental_agents/token_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../graphai";
+import { AgentFunction } from "../index";
 export declare const tokenBoundStringsAgent: AgentFunction<{
     limit?: number;
 }, {

package/lib/graphai.d.ts

@@ -1,4 +1,3 @@
-export { AgentFunction, GraphData, AgentFunctionInfoDictonary } from "./type";
 import { AgentFunctionInfoDictonary, AgentFilterInfo, GraphData, DataSource, ResultDataDictonary, ResultData, DefaultResultData } from "./type";
 import { TransactionLog } from "./transaction_log";
 import { ComputedNode, StaticNode } from "./node";
@@ -45,3 +44,4 @@ export declare class GraphAI {
     injectValue(nodeId: string, value: ResultData, injectFrom?: string): void;
     resultsOf(sources: Array<DataSource>): ResultData[];
 }
+export {};

package/lib/index.d.ts

@@ -1,4 +1,3 @@
-import { GraphAI } from "./graphai";
-export { GraphAI };
-export { AgentFunction, AgentFunctionInfoDictonary, GraphData, ResultDataDictonary, ResultData, NodeState } from "./type";
+export { GraphAI } from "./graphai";
+export { AgentFunction, AgentFunctionInfo, AgentFunctionInfoDictonary, GraphData, ResultDataDictonary, ResultData, NodeState, AgentFilterFunction, } from "./type";
 export type { TransactionLog } from "./transaction_log";

package/lib/index.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.NodeState = exports.GraphAI = void 0;
-const graphai_1 = require("./graphai");
+var graphai_1 = require("./graphai");
 Object.defineProperty(exports, "GraphAI", { enumerable: true, get: function () { return graphai_1.GraphAI; } });
 var type_1 = require("./type");
 Object.defineProperty(exports, "NodeState", { enumerable: true, get: function () { return type_1.NodeState; } });

package/lib/node.d.ts

@@ -1,4 +1,4 @@
-import type { GraphAI, GraphData } from "./graphai";
+import type { GraphAI, GraphData } from "./index";
 import { NodeDataParams, ResultData, DataSource, ComputedNodeData, StaticNodeData, NodeState } from "./type";
 import { TransactionLog } from "./transaction_log";
 export declare class Node {
@@ -31,6 +31,7 @@ export declare class ComputedNode extends Node {
     dataSources: Array<DataSource>;
     pendings: Set<string>;
     private ifSource?;
+    private console;
     readonly isStaticNode = false;
     readonly isComputedNode = true;
     constructor(graphId: string, nodeId: string, data: ComputedNodeData, graph: GraphAI);

package/lib/node.js

@@ -38,6 +38,7 @@ class ComputedNode extends Node {
         this.isComputedNode = true;
         this.graphId = graphId;
         this.params = data.params ?? {};
+        this.console = data.console ?? {};
         this.filterParams = data.filterParams ?? {};
         this.nestedGraph = data.graph;
         if (typeof data.agent === "string") {
@@ -229,7 +230,13 @@ class ComputedNode extends Node {
                 context.graphData = this.nestedGraph;
                 context.agents = this.graph.agentFunctionInfoDictionary;
             }
+            if (this.console.before) {
+                console.log(this.console.before === true ? JSON.stringify(context.inputs, null, 2) : this.console.before);
+            }
             const result = await this.agentFilterHandler(context, agentFunction);
+            if (this.console.after) {
+                console.log(this.console.after === true ? (typeof result === "string" ? result : JSON.stringify(result, null, 2)) : this.console.after);
+            }
             if (this.nestedGraph) {
                 this.graph.taskManager.restoreAfterNesting();
             }

package/lib/type.d.ts

@@ -40,6 +40,7 @@ export type ComputedNodeData = {
     graph?: GraphData;
     isResult?: boolean;
     priority?: number;
+    console?: Record<string, string | boolean>;
 };
 export type NodeData = StaticNodeData | ComputedNodeData;
 export type LoopData = {

package/lib/utils/test_agents.d.ts

@@ -1,2 +1,2 @@
-import { AgentFunctionInfoDictonary } from "../graphai";
+import { AgentFunctionInfoDictonary } from "../index";
 export declare const defaultTestAgents: AgentFunctionInfoDictonary;

package/lib/validators/common.js

@@ -2,5 +2,18 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.staticNodeAttributeKeys = exports.computedNodeAttributeKeys = exports.graphDataAttributeKeys = void 0;
 exports.graphDataAttributeKeys = ["nodes", "concurrency", "agentId", "loop", "verbose", "version"];
-exports.computedNodeAttributeKeys = ["inputs", "anyInput", "params", "retry", "timeout", "agent", "graph", "isResult", "priority", "if", "filterParams"];
+exports.computedNodeAttributeKeys = [
+    "inputs",
+    "anyInput",
+    "params",
+    "retry",
+    "timeout",
+    "agent",
+    "graph",
+    "isResult",
+    "priority",
+    "if",
+    "filterParams",
+    "console",
+];
 exports.staticNodeAttributeKeys = ["value", "update", "isResult"];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.4.1",
+  "version": "0.4.2",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [
@@ -9,6 +9,7 @@
   "scripts": {
     "build": "rm -r lib/* && tsc && tsc-alias",
     "eslint": "eslint",
+    "apiDoc": " npx typedoc src/index.ts --out apiDoc",
     "format": "prettier --write '{src,tests,samples}/**/*.ts'",
     "test": "node --test  -r tsconfig-paths/register --require ts-node/register ./tests/**/test_*.ts",
     "http_test": "node --test  -r tsconfig-paths/register --require ts-node/register ./tests/**/http_*.ts",
@@ -57,5 +58,8 @@
   "directories": {
     "lib": "lib",
     "test": "tests"
+  },
+  "dependencies": {
+    "typedoc": "^0.25.13"
   }
 }