graphai 0.3.2 → 0.4.1

package/lib/experimental_agents/matrix_agents/index.js

@@ -1,7 +1,17 @@
 "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 });
-exports.sortByValuesAgent = exports.dotProductAgent = void 0;
-var dot_product_agent_1 = require("../../experimental_agents/matrix_agents/dot_product_agent");
-Object.defineProperty(exports, "dotProductAgent", { enumerable: true, get: function () { return dot_product_agent_1.dotProductAgent; } });
-var sort_by_values_agent_1 = require("../../experimental_agents/matrix_agents/sort_by_values_agent");
-Object.defineProperty(exports, "sortByValuesAgent", { enumerable: true, get: function () { return sort_by_values_agent_1.sortByValuesAgent; } });
+__exportStar(require("./vanilla"), exports);

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

@@ -0,0 +1,3 @@
+import dotProductAgent from "../../experimental_agents/matrix_agents/dot_product_agent";
+import sortByValuesAgent from "../../experimental_agents/matrix_agents/sort_by_values_agent";
+export { dotProductAgent, sortByValuesAgent };

package/lib/experimental_agents/matrix_agents/vanilla.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.sortByValuesAgent = exports.dotProductAgent = void 0;
+const dot_product_agent_1 = __importDefault(require("../../experimental_agents/matrix_agents/dot_product_agent"));
+exports.dotProductAgent = dot_product_agent_1.default;
+const sort_by_values_agent_1 = __importDefault(require("../../experimental_agents/matrix_agents/sort_by_values_agent"));
+exports.sortByValuesAgent = sort_by_values_agent_1.default;

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

@@ -1,2 +1,3 @@
-export { wikipediaAgent } from "../../experimental_agents/service_agents/wikipedia";
-export { fetchAgent } from "../../experimental_agents/service_agents/fetch_agent";
+import wikipediaAgent from "../../experimental_agents/service_agents/wikipedia";
+import fetchAgent from "../../experimental_agents/service_agents/fetch_agent";
+export { fetchAgent, wikipediaAgent };

package/lib/experimental_agents/service_agents/index.js

@@ -1,7 +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.fetchAgent = exports.wikipediaAgent = void 0;
-var wikipedia_1 = require("../../experimental_agents/service_agents/wikipedia");
-Object.defineProperty(exports, "wikipediaAgent", { enumerable: true, get: function () { return wikipedia_1.wikipediaAgent; } });
-var fetch_agent_1 = require("../../experimental_agents/service_agents/fetch_agent");
-Object.defineProperty(exports, "fetchAgent", { enumerable: true, get: function () { return fetch_agent_1.fetchAgent; } });
+exports.wikipediaAgent = exports.fetchAgent = void 0;
+const wikipedia_1 = __importDefault(require("../../experimental_agents/service_agents/wikipedia"));
+exports.wikipediaAgent = wikipedia_1.default;
+const fetch_agent_1 = __importDefault(require("../../experimental_agents/service_agents/fetch_agent"));
+exports.fetchAgent = fetch_agent_1.default;

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

@@ -1,2 +1,3 @@
-export { sleeperAgent } from "../../experimental_agents/sleeper_agents/sleeper_agent";
-export { sleeperAgentDebug } from "../../experimental_agents/sleeper_agents/sleeper_agent_debug";
+import sleeperAgent from "../../experimental_agents/sleeper_agents/sleeper_agent";
+import sleeperAgentDebug from "../../experimental_agents/sleeper_agents/sleeper_agent_debug";
+export { sleeperAgent, sleeperAgentDebug };

package/lib/experimental_agents/sleeper_agents/index.js

@@ -1,7 +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.sleeperAgentDebug = exports.sleeperAgent = void 0;
-var sleeper_agent_1 = require("../../experimental_agents/sleeper_agents/sleeper_agent");
-Object.defineProperty(exports, "sleeperAgent", { enumerable: true, get: function () { return sleeper_agent_1.sleeperAgent; } });
-var sleeper_agent_debug_1 = require("../../experimental_agents/sleeper_agents/sleeper_agent_debug");
-Object.defineProperty(exports, "sleeperAgentDebug", { enumerable: true, get: function () { return sleeper_agent_debug_1.sleeperAgentDebug; } });
+const sleeper_agent_1 = __importDefault(require("../../experimental_agents/sleeper_agents/sleeper_agent"));
+exports.sleeperAgent = sleeper_agent_1.default;
+const sleeper_agent_debug_1 = __importDefault(require("../../experimental_agents/sleeper_agents/sleeper_agent_debug"));
+exports.sleeperAgentDebug = sleeper_agent_debug_1.default;

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

@@ -1,2 +1 @@
-export { stringSplitterAgent } from "../../experimental_agents/string_agents/string_splitter_agent";
-export { stringTemplateAgent } from "../../experimental_agents/string_agents/string_template_agent";
+export * from "./vanilla";

package/lib/experimental_agents/string_agents/index.js

@@ -1,7 +1,17 @@
 "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 });
-exports.stringTemplateAgent = exports.stringSplitterAgent = void 0;
-var string_splitter_agent_1 = require("../../experimental_agents/string_agents/string_splitter_agent");
-Object.defineProperty(exports, "stringSplitterAgent", { enumerable: true, get: function () { return string_splitter_agent_1.stringSplitterAgent; } });
-var string_template_agent_1 = require("../../experimental_agents/string_agents/string_template_agent");
-Object.defineProperty(exports, "stringTemplateAgent", { enumerable: true, get: function () { return string_template_agent_1.stringTemplateAgent; } });
+__exportStar(require("./vanilla"), exports);

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

@@ -0,0 +1,3 @@
+import stringSplitterAgent from "../../experimental_agents/string_agents/string_splitter_agent";
+import stringTemplateAgent from "../../experimental_agents/string_agents/string_template_agent";
+export { stringSplitterAgent, stringTemplateAgent };

package/lib/experimental_agents/string_agents/vanilla.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.stringTemplateAgent = exports.stringSplitterAgent = void 0;
+const string_splitter_agent_1 = __importDefault(require("../../experimental_agents/string_agents/string_splitter_agent"));
+exports.stringSplitterAgent = string_splitter_agent_1.default;
+const string_template_agent_1 = __importDefault(require("../../experimental_agents/string_agents/string_template_agent"));
+exports.stringTemplateAgent = string_template_agent_1.default;

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

@@ -1,7 +1 @@
-export { bypassAgent } from "../../experimental_agents/test_agents/bypass_agent";
-export { echoAgent } from "../../experimental_agents/test_agents/echo_agent";
-export { copyMessageAgent } from "../../experimental_agents/test_agents/copy_message_agent";
-export { mergeNodeIdAgent } from "../../experimental_agents/test_agents/merge_node_id_agent";
-export { countingAgent } from "../../experimental_agents/test_agents/counting_agent";
-export { copy2ArrayAgent } from "../../experimental_agents/test_agents/copy2array_agent";
-export { streamMockAgent } from "../../experimental_agents/test_agents/stream_mock_agent";
+export * from "./vanilla";

package/lib/experimental_agents/test_agents/index.js

@@ -1,17 +1,17 @@
 "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 });
-exports.streamMockAgent = exports.copy2ArrayAgent = exports.countingAgent = exports.mergeNodeIdAgent = exports.copyMessageAgent = exports.echoAgent = exports.bypassAgent = void 0;
-var bypass_agent_1 = require("../../experimental_agents/test_agents/bypass_agent");
-Object.defineProperty(exports, "bypassAgent", { enumerable: true, get: function () { return bypass_agent_1.bypassAgent; } });
-var echo_agent_1 = require("../../experimental_agents/test_agents/echo_agent");
-Object.defineProperty(exports, "echoAgent", { enumerable: true, get: function () { return echo_agent_1.echoAgent; } });
-var copy_message_agent_1 = require("../../experimental_agents/test_agents/copy_message_agent");
-Object.defineProperty(exports, "copyMessageAgent", { enumerable: true, get: function () { return copy_message_agent_1.copyMessageAgent; } });
-var merge_node_id_agent_1 = require("../../experimental_agents/test_agents/merge_node_id_agent");
-Object.defineProperty(exports, "mergeNodeIdAgent", { enumerable: true, get: function () { return merge_node_id_agent_1.mergeNodeIdAgent; } });
-var counting_agent_1 = require("../../experimental_agents/test_agents/counting_agent");
-Object.defineProperty(exports, "countingAgent", { enumerable: true, get: function () { return counting_agent_1.countingAgent; } });
-var copy2array_agent_1 = require("../../experimental_agents/test_agents/copy2array_agent");
-Object.defineProperty(exports, "copy2ArrayAgent", { enumerable: true, get: function () { return copy2array_agent_1.copy2ArrayAgent; } });
-var stream_mock_agent_1 = require("../../experimental_agents/test_agents/stream_mock_agent");
-Object.defineProperty(exports, "streamMockAgent", { enumerable: true, get: function () { return stream_mock_agent_1.streamMockAgent; } });
+__exportStar(require("./vanilla"), exports);

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

@@ -0,0 +1,8 @@
+import echoAgent from "../../experimental_agents/test_agents/echo_agent";
+import bypassAgent from "../../experimental_agents/test_agents/bypass_agent";
+import countingAgent from "../../experimental_agents/test_agents/counting_agent";
+import copyMessageAgent from "../../experimental_agents/test_agents/copy_message_agent";
+import copy2ArrayAgent from "../../experimental_agents/test_agents/copy2array_agent";
+import mergeNodeIdAgent from "../../experimental_agents/test_agents/merge_node_id_agent";
+import streamMockAgent from "../../experimental_agents/test_agents/stream_mock_agent";
+export { echoAgent, bypassAgent, countingAgent, copyMessageAgent, copy2ArrayAgent, mergeNodeIdAgent, streamMockAgent };

package/lib/experimental_agents/test_agents/vanilla.js

@@ -0,0 +1,20 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.streamMockAgent = exports.mergeNodeIdAgent = exports.copy2ArrayAgent = exports.copyMessageAgent = exports.countingAgent = exports.bypassAgent = exports.echoAgent = void 0;
+const echo_agent_1 = __importDefault(require("../../experimental_agents/test_agents/echo_agent"));
+exports.echoAgent = echo_agent_1.default;
+const bypass_agent_1 = __importDefault(require("../../experimental_agents/test_agents/bypass_agent"));
+exports.bypassAgent = bypass_agent_1.default;
+const counting_agent_1 = __importDefault(require("../../experimental_agents/test_agents/counting_agent"));
+exports.countingAgent = counting_agent_1.default;
+const copy_message_agent_1 = __importDefault(require("../../experimental_agents/test_agents/copy_message_agent"));
+exports.copyMessageAgent = copy_message_agent_1.default;
+const copy2array_agent_1 = __importDefault(require("../../experimental_agents/test_agents/copy2array_agent"));
+exports.copy2ArrayAgent = copy2array_agent_1.default;
+const merge_node_id_agent_1 = __importDefault(require("../../experimental_agents/test_agents/merge_node_id_agent"));
+exports.mergeNodeIdAgent = merge_node_id_agent_1.default;
+const stream_mock_agent_1 = __importDefault(require("../../experimental_agents/test_agents/stream_mock_agent"));
+exports.streamMockAgent = stream_mock_agent_1.default;

package/lib/experimental_agents/vanilla.d.ts

@@ -1,8 +1,9 @@
-export * from "./string_agents";
-export * from "./array_agents";
-export * from "./matrix_agents";
-export * from "./test_agents";
-export * from "./graph_agents";
-export * from "./function_agent";
-export * from "./embedding_agent";
-export * from "./data_agents";
+export * from "./string_agents/vanilla";
+export * from "./array_agents/vanilla";
+export * from "./matrix_agents/vanilla";
+export * from "./test_agents/vanilla";
+export * from "./graph_agents/vanilla";
+export * from "./data_agents/vanilla";
+import functionAgent from "./function_agent";
+import stringEmbeddingsAgent from "./embedding_agent";
+export { functionAgent, stringEmbeddingsAgent };

package/lib/experimental_agents/vanilla.js

@@ -14,13 +14,19 @@ 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"), exports);
-__exportStar(require("./array_agents"), exports);
-__exportStar(require("./matrix_agents"), exports);
-__exportStar(require("./test_agents"), exports);
-__exportStar(require("./graph_agents"), exports);
-__exportStar(require("./function_agent"), exports);
-__exportStar(require("./embedding_agent"), exports);
-__exportStar(require("./data_agents"), exports);
+__exportStar(require("./string_agents/vanilla"), exports);
+__exportStar(require("./array_agents/vanilla"), exports);
+__exportStar(require("./matrix_agents/vanilla"), exports);
+__exportStar(require("./test_agents/vanilla"), exports);
+__exportStar(require("./graph_agents/vanilla"), exports);
+__exportStar(require("./data_agents/vanilla"), 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/graphai.d.ts

@@ -1,5 +1,5 @@
-export { AgentFunction, AgentFunctionDictonary, GraphData } from "./type";
-import { AgentFunctionDictonary, AgentFilterInfo, GraphData, DataSource, ResultDataDictonary, ResultData, DefaultResultData } from "./type";
+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";
 import { TaskManager } from "./task_manager";
@@ -10,7 +10,7 @@ export declare class GraphAI {
     private readonly data;
     private readonly loop?;
     private readonly logs;
-    readonly agentFunctionDictionary: AgentFunctionDictonary;
+    readonly agentFunctionInfoDictionary: AgentFunctionInfoDictonary;
     readonly taskManager: TaskManager;
     readonly agentFilters: AgentFilterInfo[];
     readonly retryLimit?: number;
@@ -22,11 +22,11 @@ export declare class GraphAI {
     private createNodes;
     private getValueFromResults;
     private initializeNodes;
-    constructor(data: GraphData, agentFunctionDictionary: AgentFunctionDictonary, options?: {
+    constructor(data: GraphData, agentFunctionInfoDictionary: AgentFunctionInfoDictonary, options?: {
         agentFilters?: AgentFilterInfo[] | undefined;
         taskManager?: TaskManager | undefined;
     });
-    getAgentFunction(agentId?: string): import("./type").AgentFunction<any, any, any>;
+    getAgentFunctionInfo(agentId?: string): import("./type").AgentFunctionInfo;
     asString(): string;
     results<T = DefaultResultData>(all: boolean): ResultDataDictonary<T>;
     errors(): Record<string, Error>;

package/lib/graphai.js

@@ -63,7 +63,7 @@ class GraphAI {
             }
         });
     }
-    constructor(data, agentFunctionDictionary, options = { taskManager: undefined, agentFilters: [] }) {
+    constructor(data, agentFunctionInfoDictionary, options = { taskManager: undefined, agentFilters: [] }) {
         this.logs = [];
         this.onLogCallback = (__log, __isUpdate) => { };
         this.repeatCount = 0;
@@ -77,7 +77,7 @@ class GraphAI {
         this.retryLimit = data.retry; // optional
         this.graphId = URL.createObjectURL(new Blob()).slice(-36);
         this.data = data;
-        this.agentFunctionDictionary = agentFunctionDictionary;
+        this.agentFunctionInfoDictionary = agentFunctionInfoDictionary;
         this.taskManager = options.taskManager ?? new task_manager_1.TaskManager(data.concurrency ?? defaultConcurrency);
         this.agentFilters = options.agentFilters ?? [];
         this.loop = data.loop;
@@ -85,13 +85,13 @@ class GraphAI {
         this.onComplete = () => {
             console.error("-- SOMETHING IS WRONG: onComplete is called without run()");
         };
-        (0, validator_1.validateGraphData)(data, Object.keys(agentFunctionDictionary));
+        (0, validator_1.validateGraphData)(data, Object.keys(agentFunctionInfoDictionary));
         this.nodes = this.createNodes(data);
         this.initializeNodes();
     }
-    getAgentFunction(agentId) {
-        if (agentId && this.agentFunctionDictionary[agentId]) {
-            return this.agentFunctionDictionary[agentId];
+    getAgentFunctionInfo(agentId) {
+        if (agentId && this.agentFunctionInfoDictionary[agentId]) {
+            return this.agentFunctionInfoDictionary[agentId];
         }
         throw new Error("No agent: " + agentId);
     }

package/lib/index.d.ts

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

package/lib/node.js

@@ -201,7 +201,7 @@ class ComputedNode extends Node {
             }, this.timeout);
         }
         try {
-            const agentFunction = this.agentFunction ?? this.graph.getAgentFunction(this.agentId);
+            const agentFunction = this.agentFunction ?? this.graph.getAgentFunctionInfo(this.agentId).agent;
             const localLog = [];
             const params = Object.keys(this.dynamicParams).reduce((tmp, key) => {
                 const [result] = this.graph.resultsOf([this.dynamicParams[key]]);
@@ -227,7 +227,7 @@ class ComputedNode extends Node {
                 this.graph.taskManager.prepareForNesting();
                 context.taskManager = this.graph.taskManager;
                 context.graphData = this.nestedGraph;
-                context.agents = this.graph.agentFunctionDictionary;
+                context.agents = this.graph.agentFunctionInfoDictionary;
             }
             const result = await this.agentFilterHandler(context, agentFunction);
             if (this.nestedGraph) {

package/lib/type.d.ts

@@ -64,7 +64,7 @@ export type AgentFunctionContext<ParamsType = DefaultParamsType, InputDataType =
         agentId?: string;
     };
     graphData?: GraphData | string;
-    agents?: AgentFunctionDictonary;
+    agents?: AgentFunctionInfoDictonary;
     taskManager?: TaskManager;
     filterParams: AgentFilterParams;
     agentFilters?: AgentFilterInfo[];
@@ -79,7 +79,6 @@ export type AgentFilterInfo = {
     nodeIds?: string[];
     filterParams?: AgentFilterParams;
 };
-export type AgentFunctionDictonary = Record<string, AgentFunction<any, any, any>>;
 export type AgentFunctionInfo = {
     name: string;
     agent: AgentFunction<any, any, any>;
@@ -96,3 +95,4 @@ export type AgentFunctionInfo = {
     repository: string;
     license: string;
 };
+export type AgentFunctionInfoDictonary = Record<string, AgentFunctionInfo>;

package/lib/utils/test_agents.d.ts

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

package/lib/utils/test_agents.js

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.defaultTestAgents = void 0;
 const experimental_agents_1 = require("../experimental_agents");
-const openai_agent_1 = require("../experimental_agents/llm_agents/openai_agent");
+// import openAIMockAgent from "../experimental_agents/llm_agents/openai_agent";
 exports.defaultTestAgents = {
     bypassAgent: experimental_agents_1.bypassAgent,
     echoAgent: experimental_agents_1.echoAgent,
@@ -20,5 +20,5 @@ exports.defaultTestAgents = {
     popAgent: experimental_agents_1.popAgent,
     shiftAgent: experimental_agents_1.shiftAgent,
     streamMockAgent: experimental_agents_1.streamMockAgent,
-    openAIMockAgent: openai_agent_1.openAIMockAgent,
+    // openAIMockAgent,
 };

package/lib/utils/test_utils.d.ts

@@ -1,4 +1,19 @@
 import { AgentFunctionInfo, AgentFunctionContext, AgentFunction, AgentFilterInfo, ResultData } from "../type";
+export declare const getAgentInfo: (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>;
+};
 export declare const defaultTestContext: {
     debugInfo: {
         nodeId: string;

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.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",

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.3.2",
+  "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",

package/lib/experimental_agents/packages.d.ts

@@ -1,13 +0,0 @@
-export * from "./string_agents/packages";
-export * from "./sleeper_agents/packages";
-export * from "./data_agents/packages";
-export * from "./array_agents/packages";
-export * from "./matrix_agents/packages";
-export * from "./test_agents/packages";
-export * from "./graph_agents/packages";
-export * from "./llm_agents/packages";
-export * from "./service_agents/packages";
-import stringEmbeddingsAgent from "./embedding_agent";
-import tokenBoundStringsAgent from "./token_agent";
-import functionAgentInfo from "./function_agent";
-export { stringEmbeddingsAgent, tokenBoundStringsAgent, functionAgentInfo };

package/lib/experimental_agents/packages.js

@@ -1,35 +0,0 @@
-"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);
-};
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-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);
-__exportStar(require("./array_agents/packages"), exports);
-__exportStar(require("./matrix_agents/packages"), exports);
-__exportStar(require("./test_agents/packages"), exports);
-__exportStar(require("./graph_agents/packages"), exports);
-__exportStar(require("./llm_agents/packages"), exports);
-__exportStar(require("./service_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;