graphai 0.4.2 → 0.4.3

package/README.md

@@ -6,7 +6,7 @@ GraphAI is an asynchronous data flow execution engine, which allows developers t
 
 As Andrew Ng has described in his article, "[The batch: Issue 242](https://www.deeplearning.ai/the-batch/issue-242/)", better results can often be achieved by making multiple calls to a Large Language Model (LLM) and allowing it to incrementally build towards a higher-quality output. Dr. Ng refers to this approach as 'agentic workflows.' 
 
-Such *agentic applications* need to make multiple asynchronous API calls (such as OpenAI's chat-completion API, database query, web search, and etc.) and manage data dependencies among them, such as giving the answer from one LLM call to another -- which will become quite difficult to manage in a traditional programing style as the complexity of the application increases, because of the asynchronous nature of those APIs.
+Such *agentic applications* require making multiple asynchronous API calls (e.g., OpenAI's chat-completion API, database queries, web searches) and managing data dependencies among them. As the complexity of the application increases, managing these dependencies in a traditional programming style becomes challenging due to the asynchronous nature of the APIs.
 
 GraphAI allows developers to describe dependencies among those agents (asynchronous API calls) in a data flow graph in YAML or JSON, which is called *declarative data flow programming* . The GraphAI engine will take care of all the complexity of concurrent asynchronous calls, data dependency management, task priority management, map-reduce processing, error handling, retries and logging. 
 
@@ -68,9 +68,9 @@ flowchart TD
  chunkEmbeddings --> similarities(similarities)
  topicEmbedding --> similarities
  similarities --> sortedChunks(sortedChunks)
- sortedChunks --> resourceText(resourceText)
+ sortedChunks --> referenceText(resourceText)
  source -- query --> prompt(prompt)
- resourceText --> prompt
+ referenceText --> prompt
  prompt --> query(query)
 ```
 
@@ -120,7 +120,7 @@ An *agent function* is a TypeScript function, which implements a particular *age
 There are additional optional parameters for developers of nested agents and agent filters.
 
 - *graphData*: an optional GraphData (for nested agents)
-- *agents*: AgentFunctionInfoDictonary (for nested agents)
+- *agents*: AgentFunctionInfoDictionary (for nested agents)
 - *taskManager*: TaskManager (for nested agents)
 - *log*: TransactionLog[] (for nested agents)
 - *filterParams*: agent filter parameters (for agent filters)
@@ -377,77 +377,7 @@ By default, tasks are executed in a first-in, first-out (FIFO) order with a neut
 
 Negative priority values are allowed, enabling you to fine-tune the execution order based on your application's requirements.
 
-## GraphAI class
+## References
 
-### ```constructor(data: GraphData, callbackDictonary: AgentFunctionInfoDictonary)```
-Initializes a new instance of the GraphAI object with the specified graph data and a dictionary of callback functions.
-
-- ```data: GraphData```: The graph data including nodes and optional concurrency limit.
-- ```callbackDictonary: AgentFunctionInfoDictonary```: A dictionary mapping agent IDs to their respective callback functions info to be used for all nodes.
-
-### ```async run(): Promise<ResultDataDictonary<ResultData>>```
-Executes the graph asynchronously, starting with nodes that have no dependencies or whose dependencies have been met. The method continues to execute nodes as their dependencies are satisfied until all nodes have been executed or an error occurs.
-
-Returns: A promise that resolves with the results of all executed nodes or rejects with the first encountered error.
-
-### ```results(): ResultDataDictonary<ResultData>```
-Compiles and returns the results of all executed nodes in the graph.
-
-Returns: A dictionary mapping node IDs to their results. Only includes nodes that have completed execution and produced a result.
-
-### ```errors(): Record<string, Error>```
-Compiles and returns the errors from all nodes that encountered an error during execution.
-
-Returns: A dictionary mapping node IDs to the errors they encountered. Only includes nodes that have executed and encountered an error. It does not include any errors which have been retried.
-
-### ```transactionLogs(): Array<TransactionLog>```
-Retrieves all transaction logs recorded during the execution of the graph.
-
-Returns: An array of transaction logs detailing the execution states and outcomes of the nodes within the graph.
-
-### ```injectValue(nodeId: string, result: ResultData): void```
-Injects a result into a specified node. This is used to manually set the result of a static node, allowing dependent nodes to proceed with execution.
-
-- ```nodeId: string```: The ID of the static node into which the result is to be injected.
-- ```result: ResultData```: The result to be injected into the specified node.
-
-## Collaboration
-
-Step 1. Install git, node and yarn
-
-Step 2. Clone the project and install necessary node modules
-
-```
-git clone git@github.com:receptron/graphai.git
-cd graphai
-yarn install
-```
-
-Step 3. Set the environment variable OPENAI_API_KEY to your own key (=sk-...)
-
-You need to set ANTHROPIC_API_KEY as well, if you want to use Claude.
-
-Step 4. Run the test script
-
-Run the test
-
-```
-npm run test
-```
-
-Step 5. Run one of sample scripts
-
-```
-npm run sample ./samples/home.ts
-```
-
-Step 6. Write some code and send pull requests
-
-- Please run "yarn run format" before sending your pull request.
-- Please do not include any build files (files under /lib) to your pull reuquest.
-
-Key principles:
-
-1. Keep the core (Node and GraphAI classes) small and simple.
-2. Enhance the platform by adding 'agents' (plug ins).
-3. Simple but effective test scripts make it easy to maintain.
+- [Collaboration](./Collaboration.md)
+- [API Document](./APIDocument.md)

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

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

package/lib/experimental_agents/array_agents/push_agent.js

@@ -33,7 +33,7 @@ const pushAgentInfo = {
         },
     ],
     description: "push Agent",
-    category: [],
+    category: ["array"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

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

package/lib/experimental_agents/array_agents/shift_agent.js

@@ -31,7 +31,7 @@ const shiftAgentInfo = {
         },
     ],
     description: "shift Agent",
-    category: [],
+    category: ["array"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -4,6 +4,7 @@ export declare const propertyFilterAgent: AgentFunction<{
     exclude?: Array<string>;
     alter?: Record<string, Record<string, string>>;
     inject?: Array<Record<string, any>>;
+    inspect?: Array<Record<string, any>>;
     swap?: Record<string, string>;
 }>;
 declare const propertyFilterAgentInfo: {
@@ -13,6 +14,7 @@ declare const propertyFilterAgentInfo: {
         exclude?: string[] | undefined;
         alter?: Record<string, Record<string, string>> | undefined;
         inject?: Record<string, any>[] | undefined;
+        inspect?: Record<string, any>[] | undefined;
         swap?: Record<string, string> | undefined;
     }>;
     mock: AgentFunction<{
@@ -20,6 +22,7 @@ declare const propertyFilterAgentInfo: {
         exclude?: string[] | undefined;
         alter?: Record<string, Record<string, string>> | undefined;
         inject?: Record<string, any>[] | undefined;
+        inspect?: Record<string, any>[] | undefined;
         swap?: Record<string, string> | undefined;
     }>;
     samples: ({
@@ -36,6 +39,7 @@ declare const propertyFilterAgentInfo: {
             alter?: undefined;
             inject?: undefined;
             swap?: undefined;
+            inspect?: undefined;
         };
         result: {
             color: string;
@@ -55,6 +59,7 @@ declare const propertyFilterAgentInfo: {
             alter?: undefined;
             inject?: undefined;
             swap?: undefined;
+            inspect?: undefined;
         };
         result: {
             color: string;
@@ -74,6 +79,7 @@ declare const propertyFilterAgentInfo: {
             alter?: undefined;
             inject?: undefined;
             swap?: undefined;
+            inspect?: undefined;
         };
         result: {
             type: string;
@@ -99,6 +105,7 @@ declare const propertyFilterAgentInfo: {
             exclude?: undefined;
             inject?: undefined;
             swap?: undefined;
+            inspect?: undefined;
         };
         result: {
             color: string;
@@ -124,6 +131,7 @@ declare const propertyFilterAgentInfo: {
             exclude?: undefined;
             alter?: undefined;
             swap?: undefined;
+            inspect?: undefined;
         };
         result: {
             color: string;
@@ -150,6 +158,7 @@ declare const propertyFilterAgentInfo: {
             exclude?: undefined;
             alter?: undefined;
             swap?: undefined;
+            inspect?: undefined;
         };
         result: {
             color: string;
@@ -174,6 +183,7 @@ declare const propertyFilterAgentInfo: {
             exclude?: undefined;
             alter?: undefined;
             inject?: undefined;
+            inspect?: undefined;
         };
         result: {
             color: string;
@@ -182,6 +192,41 @@ declare const propertyFilterAgentInfo: {
             maker: string;
             range: number;
         }[];
+    } | {
+        inputs: (string | {
+            color: string;
+            model: string;
+            type: string;
+            maker: string;
+            range: number;
+        }[])[];
+        params: {
+            inspect: ({
+                propId: string;
+                equal: string;
+                notEqual?: undefined;
+                from?: undefined;
+            } | {
+                propId: string;
+                notEqual: string;
+                from: number;
+                equal?: undefined;
+            })[];
+            include?: undefined;
+            exclude?: undefined;
+            alter?: undefined;
+            inject?: undefined;
+            swap?: undefined;
+        };
+        result: {
+            color: string;
+            model: string;
+            type: string;
+            maker: string;
+            range: number;
+            isTesla: boolean;
+            isGM: boolean;
+        }[];
     })[];
     description: string;
     category: string[];

package/lib/experimental_agents/data_agents/property_filter_agent.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.propertyFilterAgent = void 0;
-const applyFilter = (input, index, inputs, include, exclude, alter, inject, swap) => {
+const applyFilter = (input, index, inputs, include, exclude, alter, inject, swap, inspect) => {
     const propIds = include ? include : Object.keys(input);
     const excludeSet = new Set(exclude ?? []);
     const result = propIds.reduce((tmp, propId) => {
@@ -23,6 +23,17 @@ const applyFilter = (input, index, inputs, include, exclude, alter, inject, swap
             }
         });
     }
+    if (inspect) {
+        inspect.forEach((item) => {
+            const value = inputs[item.from ?? 1]; // default is inputs[1]
+            if (item.equal) {
+                result[item.propId] = item.equal === value;
+            }
+            else if (item.notEqual) {
+                result[item.propId] = item.notEqual !== value;
+            }
+        });
+    }
     if (swap) {
         Object.keys(swap).forEach((key) => {
             const tmp = result[key];
@@ -34,11 +45,11 @@ const applyFilter = (input, index, inputs, include, exclude, alter, inject, swap
 };
 const propertyFilterAgent = async ({ inputs, params }) => {
     const [input] = inputs;
-    const { include, exclude, alter, inject, swap } = params;
+    const { include, exclude, alter, inject, swap, inspect } = params;
     if (Array.isArray(input)) {
-        return input.map((item, index) => applyFilter(item, index, inputs, include, exclude, alter, inject, swap));
+        return input.map((item, index) => applyFilter(item, index, inputs, include, exclude, alter, inject, swap, inspect));
     }
-    return applyFilter(input, 0, inputs, include, exclude, alter, inject, swap);
+    return applyFilter(input, 0, inputs, include, exclude, alter, inject, swap, inspect);
 };
 exports.propertyFilterAgent = propertyFilterAgent;
 const testInputs = [
@@ -106,6 +117,19 @@ const propertyFilterAgentInfo = {
                 { color: "blue", model: "Tesla", type: "EV", maker: "Model Y", range: 400 },
             ],
         },
+        {
+            inputs: testInputs,
+            params: {
+                inspect: [
+                    { propId: "isTesla", equal: "Tesla Motors" }, // from: 1 is implied
+                    { propId: "isGM", notEqual: "Tesla Motors", from: 1 },
+                ],
+            },
+            result: [
+                { color: "red", model: "Model 3", type: "EV", maker: "Tesla", range: 300, isTesla: true, isGM: false },
+                { color: "blue", model: "Model Y", type: "EV", maker: "Tesla", range: 400, isTesla: true, isGM: false },
+            ],
+        },
     ],
     description: "Filter properties based on property name either with 'include' or 'exclude'",
     category: ["data"],

package/lib/experimental_agents/embedding_agent.d.ts

@@ -12,7 +12,7 @@ declare const stringEmbeddingsAgentInfo: {
     }, any, string | string[]>;
     samples: never[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/embedding_agent.js

@@ -47,7 +47,7 @@ const stringEmbeddingsAgentInfo = {
     mock: exports.stringEmbeddingsAgent,
     samples: [],
     description: "Embeddings Agent",
-    category: [],
+    category: ["embedding"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -15,7 +15,7 @@ declare const mapAgentInfo: {
     }, Record<string, any[]>, any>;
     samples: never[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/graph_agents/map_agent.js

@@ -62,7 +62,7 @@ const mapAgentInfo = {
     mock: exports.mapAgent,
     samples: [],
     description: "Map Agent",
-    category: [],
+    category: ["graph"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -14,7 +14,7 @@ declare const nestedAgentInfo: {
     }>;
     samples: never[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/graph_agents/nested_agent.js

@@ -50,7 +50,7 @@ const nestedAgentInfo = {
     mock: exports.nestedAgent,
     samples: [],
     description: "nested Agent",
-    category: [],
+    category: ["graph"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,3 +1,4 @@
 import nestedAgent from "../../experimental_agents/graph_agents/nested_agent";
 import mapAgent from "../../experimental_agents/graph_agents/map_agent";
-export { nestedAgent, mapAgent };
+import workerAgent from "../../experimental_agents/graph_agents/worker_agent";
+export { nestedAgent, mapAgent, workerAgent };

package/lib/experimental_agents/graph_agents/packages.js

@@ -3,8 +3,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.mapAgent = exports.nestedAgent = void 0;
+exports.workerAgent = exports.mapAgent = exports.nestedAgent = void 0;
 const nested_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/nested_agent"));
 exports.nestedAgent = nested_agent_1.default;
 const map_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/map_agent"));
 exports.mapAgent = map_agent_1.default;
+const worker_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/worker_agent"));
+exports.workerAgent = worker_agent_1.default;

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

@@ -1,3 +1,4 @@
 import nestedAgent from "../../experimental_agents/graph_agents/nested_agent";
 import mapAgent from "../../experimental_agents/graph_agents/map_agent";
-export { nestedAgent, mapAgent };
+import workerAgent from "../../experimental_agents/graph_agents/worker_agent";
+export { nestedAgent, mapAgent, workerAgent };

package/lib/experimental_agents/graph_agents/vanilla.js

@@ -3,8 +3,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.mapAgent = exports.nestedAgent = void 0;
+exports.workerAgent = exports.mapAgent = exports.nestedAgent = void 0;
 const nested_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/nested_agent"));
 exports.nestedAgent = nested_agent_1.default;
 const map_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/map_agent"));
 exports.mapAgent = map_agent_1.default;
+const worker_agent_1 = __importDefault(require("../../experimental_agents/graph_agents/worker_agent"));
+exports.workerAgent = worker_agent_1.default;

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

@@ -0,0 +1,39 @@
+import { AgentFunction } from "../../index";
+export declare const workerAgent: AgentFunction<{
+    namedInputs?: Array<string>;
+}, any, any>;
+declare const workerAgentInfo: {
+    name: string;
+    agent: AgentFunction<{
+        namedInputs?: string[] | undefined;
+    }, any, any>;
+    mock: AgentFunction<{
+        namedInputs?: string[] | undefined;
+    }, any, any>;
+    samples: {
+        inputs: string[];
+        params: {};
+        result: {
+            message: string;
+        };
+        graph: {
+            version: number;
+            nodes: {
+                source: {
+                    value: string;
+                };
+                message: {
+                    agent: string;
+                    inputs: string[];
+                    isResult: boolean;
+                };
+            };
+        };
+    }[];
+    description: string;
+    category: string[];
+    author: string;
+    repository: string;
+    license: string;
+};
+export default workerAgentInfo;

package/lib/experimental_agents/graph_agents/worker_agent.js

@@ -0,0 +1,111 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.workerAgent = void 0;
+const index_1 = require("../../index");
+const worker_threads_1 = require("worker_threads");
+const experimental_agents_1 = require("../../experimental_agents");
+const utils_1 = require("../../utils/utils");
+const vanillaAgents = {
+    totalAgent: experimental_agents_1.totalAgent,
+    dataSumTemplateAgent: experimental_agents_1.dataSumTemplateAgent,
+    propertyFilterAgent: experimental_agents_1.propertyFilterAgent,
+    copyAgent: experimental_agents_1.copyAgent,
+    pushAgent: experimental_agents_1.pushAgent,
+    popAgent: experimental_agents_1.popAgent,
+    shiftAgent: experimental_agents_1.shiftAgent,
+    nestedAgent: experimental_agents_1.nestedAgent,
+    mapAgent: experimental_agents_1.mapAgent,
+    dotProductAgent: experimental_agents_1.dotProductAgent,
+    sortByValuesAgent: experimental_agents_1.sortByValuesAgent,
+    stringSplitterAgent: experimental_agents_1.stringSplitterAgent,
+    stringTemplateAgent: experimental_agents_1.stringTemplateAgent,
+    jsonParserAgent: experimental_agents_1.jsonParserAgent,
+};
+if (!worker_threads_1.isMainThread && worker_threads_1.parentPort) {
+    const port = worker_threads_1.parentPort;
+    port.on("message", async (data) => {
+        const { graphData } = data;
+        const graphAI = new index_1.GraphAI(graphData, vanillaAgents);
+        const result = await graphAI.run();
+        port.postMessage(result);
+    });
+}
+const workerAgent = async ({ inputs, params, /* agents, log, */ graphData }) => {
+    const namedInputs = params.namedInputs ?? inputs.map((__input, index) => `$${index}`);
+    (0, utils_1.assert)(!!graphData, "required");
+    (0, utils_1.assert)(typeof graphData === "object", "required");
+    namedInputs.forEach((nodeId, index) => {
+        if (graphData.nodes[nodeId] === undefined) {
+            // If the input node does not exist, automatically create a static node
+            graphData.nodes[nodeId] = { value: inputs[index] };
+        }
+        else {
+            // Otherwise, inject the proper data here (instead of calling injectTo method later)
+            graphData.nodes[nodeId]["value"] = inputs[index];
+        }
+    });
+    return new Promise((resolve, reject) => {
+        const worker = new worker_threads_1.Worker("./lib/experimental_agents/graph_agents/worker_agent.js");
+        worker.on("message", (result) => {
+            worker.terminate();
+            resolve(result);
+        });
+        worker.on("error", reject);
+        worker.on("exit", (code) => {
+            if (code !== 0)
+                reject(new Error(`Worker stopped with exit code ${code}`));
+        });
+        // copyAgent is required for test case
+        worker.postMessage({ graphData });
+    });
+};
+exports.workerAgent = workerAgent;
+const workerAgentInfo = {
+    name: "workerAgent",
+    agent: exports.workerAgent,
+    mock: exports.workerAgent,
+    samples: [
+        {
+            inputs: [],
+            params: {},
+            result: { message: "May the force be with you" },
+            graph: {
+                version: 0.3,
+                nodes: {
+                    source: {
+                        value: "May the force be with you",
+                    },
+                    message: {
+                        agent: "copyAgent",
+                        inputs: [":source"],
+                        isResult: true,
+                    },
+                },
+            },
+        },
+        {
+            inputs: ["May the force be with you"],
+            params: {},
+            result: { message: "May the force be with you" },
+            graph: {
+                version: 0.3,
+                nodes: {
+                    source: {
+                        value: "TypeScript compiler fails without this node for some reason.",
+                    },
+                    message: {
+                        agent: "copyAgent",
+                        inputs: [":$0"],
+                        isResult: true,
+                    },
+                },
+            },
+        },
+    ],
+    description: "Map Agent",
+    category: ["graph"],
+    author: "Receptron team",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = workerAgentInfo;

package/lib/experimental_agents/index.d.ts

@@ -8,7 +8,6 @@ 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";
-export { functionAgent, stringEmbeddingsAgent, tokenBoundStringsAgent };
+export { stringEmbeddingsAgent, tokenBoundStringsAgent };

package/lib/experimental_agents/index.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.functionAgent = void 0;
+exports.tokenBoundStringsAgent = exports.stringEmbeddingsAgent = void 0;
 __exportStar(require("./string_agents"), exports);
 __exportStar(require("./array_agents"), exports);
 __exportStar(require("./matrix_agents"), exports);
@@ -30,8 +30,6 @@ __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;
 const embedding_agent_1 = __importDefault(require("./embedding_agent"));
 exports.stringEmbeddingsAgent = embedding_agent_1.default;
 const token_agent_1 = __importDefault(require("./token_agent"));

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

@@ -16,7 +16,14 @@ declare const textInputAgentInfo: {
     }, string | {
         [x: string]: string;
     }>;
-    samples: never[];
+    samples: {
+        inputs: never[];
+        params: {
+            message: string;
+        };
+        result: string;
+    }[];
+    skipTest: boolean;
     description: string;
     category: string[];
     author: string;

package/lib/experimental_agents/input_agents/text_input_agent.js

@@ -13,9 +13,16 @@ const textInputAgentInfo = {
     name: "textInputAgent",
     agent: exports.textInputAgent,
     mock: exports.textInputAgent,
-    samples: [],
+    samples: [
+        {
+            inputs: [],
+            params: { message: "Enter your message to AI." },
+            result: "message from the user",
+        },
+    ],
+    skipTest: true,
     description: "Text Input Agent",
-    category: ["array"],
+    category: ["input"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",