graphai 0.3.1 → 0.4.0

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/token_agent.d.ts

@@ -16,7 +16,17 @@ declare const tokenBoundStringsAgentInfo: {
     }, {
         content: string;
     }, string[]>;
-    samples: never[];
+    samples: {
+        inputs: string[][];
+        params: {
+            limit: number;
+        };
+        result: {
+            content: string;
+            tokenCount: number;
+            endIndex: number;
+        };
+    }[];
     description: string;
     category: never[];
     author: string;

package/lib/experimental_agents/token_agent.js

@@ -35,7 +35,36 @@ const tokenBoundStringsAgentInfo = {
     name: "tokenBoundStringsAgent",
     agent: exports.tokenBoundStringsAgent,
     mock: exports.tokenBoundStringsAgent,
-    samples: [],
+    samples: [
+        {
+            inputs: [
+                [
+                    "Here's to the crazy ones. The misfits. The rebels. The troublemakers.",
+                    "The round pegs in the square holes. The ones who see things differently.",
+                    "They're not fond of rules. And they have no respect for the status quo.",
+                    "You can quote them, disagree with them, glorify or vilify them.",
+                    "About the only thing you can't do is ignore them.",
+                    "Because they change things.",
+                    "They push the human race forward.",
+                    "And while some may see them as the crazy ones, we see genius.",
+                    "Because the people who are crazy enough to think they can change the world, are the ones who do.",
+                ],
+            ],
+            params: {
+                limit: 80,
+            },
+            result: {
+                content: "Here's to the crazy ones. The misfits. The rebels. The troublemakers.\n" +
+                    "The round pegs in the square holes. The ones who see things differently.\n" +
+                    "They're not fond of rules. And they have no respect for the status quo.\n" +
+                    "You can quote them, disagree with them, glorify or vilify them.\n" +
+                    "About the only thing you can't do is ignore them.\n" +
+                    "Because they change things.",
+                tokenCount: 79,
+                endIndex: 6,
+            },
+        },
+    ],
     description: "token bound Agent",
     category: [],
     author: "Receptron team",

package/lib/experimental_agents/vanilla.d.ts

@@ -1,8 +1,8 @@
-export * from "./string_agents";
-export * from "./array_agents";
-export * from "./matrix_agents";
-export * from "./test_agents";
-export * from "./graph_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";
 export * from "./function_agent";
 export * from "./embedding_agent";
-export * from "./data_agents";

package/lib/experimental_agents/vanilla.js

@@ -16,11 +16,11 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 // 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("./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);
 __exportStar(require("./function_agent"), exports);
 __exportStar(require("./embedding_agent"), exports);
-__exportStar(require("./data_agents"), exports);

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 callbackDictonary: 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, callbackDictonary: AgentFunctionDictonary, options?: {
+    constructor(data: GraphData, agentFunctionInfoDictionary: AgentFunctionInfoDictonary, options?: {
         agentFilters?: AgentFilterInfo[] | undefined;
         taskManager?: TaskManager | undefined;
     });
-    getCallback(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, callbackDictonary, 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.callbackDictonary = callbackDictonary;
+        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(callbackDictonary));
+        (0, validator_1.validateGraphData)(data, Object.keys(agentFunctionInfoDictionary));
         this.nodes = this.createNodes(data);
         this.initializeNodes();
     }
-    getCallback(agentId) {
-        if (agentId && this.callbackDictonary[agentId]) {
-            return this.callbackDictonary[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

@@ -71,7 +71,7 @@ class ComputedNode extends Node {
             }
             return tmp;
         }, {});
-        this.log.initForComputedNode(this);
+        this.log.initForComputedNode(this, graph);
     }
     getAgentId() {
         return this.agentId ?? "__custom__function"; // only for display purpose in the log.
@@ -167,20 +167,20 @@ class ComputedNode extends Node {
         }
         return !agentFilter.agentIds && !agentFilter.nodeIds;
     }
-    agentFilterHandler(context, agent) {
+    agentFilterHandler(context, agentFunction) {
         let index = 0;
-        const next = (context) => {
+        const next = (innerContext) => {
             const agentFilter = this.graph.agentFilters[index++];
             if (agentFilter) {
                 if (this.shouldApplyAgentFilter(agentFilter)) {
                     if (agentFilter.filterParams) {
-                        context.filterParams = { ...agentFilter.filterParams, ...context.filterParams };
+                        innerContext.filterParams = { ...agentFilter.filterParams, ...innerContext.filterParams };
                     }
-                    return agentFilter.agent(context, next);
+                    return agentFilter.agent(innerContext, next);
                 }
-                return next(context);
+                return next(innerContext);
             }
-            return agent(context);
+            return agentFunction(innerContext);
         };
         return next(context);
     }
@@ -201,7 +201,7 @@ class ComputedNode extends Node {
             }, this.timeout);
         }
         try {
-            const callback = this.agentFunction ?? this.graph.getCallback(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]]);
@@ -218,6 +218,7 @@ class ComputedNode extends Node {
                     verbose: this.graph.verbose,
                 },
                 filterParams: this.filterParams,
+                agentFilters: this.graph.agentFilters,
                 log: localLog,
             };
             // NOTE: We use the existence of graph object in the agent-specific params to determine
@@ -226,9 +227,9 @@ class ComputedNode extends Node {
                 this.graph.taskManager.prepareForNesting();
                 context.taskManager = this.graph.taskManager;
                 context.graphData = this.nestedGraph;
-                context.agents = this.graph.callbackDictonary;
+                context.agents = this.graph.agentFunctionInfoDictionary;
             }
-            const result = await this.agentFilterHandler(context, callback);
+            const result = await this.agentFilterHandler(context, agentFunction);
             if (this.nestedGraph) {
                 this.graph.taskManager.restoreAfterNesting();
             }

package/lib/task_manager.js

@@ -58,12 +58,9 @@ class TaskManager {
         this.concurrency--;
     }
     getStatus(verbose = false) {
-        const nodes = verbose
-            ? {
-                runningNodes: Array.from(this.runningNodes).map((node) => node.nodeId),
-                queuedNodes: this.taskQueue.map((task) => task.node.nodeId),
-            }
-            : {};
+        const runningNodes = Array.from(this.runningNodes).map((node) => node.nodeId);
+        const queuedNodes = this.taskQueue.map((task) => task.node.nodeId);
+        const nodes = verbose ? { runningNodes, queuedNodes } : {};
         return {
             concurrency: this.concurrency,
             queue: this.taskQueue.length,

package/lib/transaction_log.d.ts

@@ -19,7 +19,7 @@ export declare class TransactionLog {
     repeatCount?: number;
     log?: TransactionLog[];
     constructor(nodeId: string);
-    initForComputedNode(node: ComputedNode): void;
+    initForComputedNode(node: ComputedNode, graph: GraphAI): void;
     onInjected(node: StaticNode, graph: GraphAI, injectFrom?: string): void;
     onComplete(node: ComputedNode, graph: GraphAI, localLog: TransactionLog[]): void;
     beforeExecute(node: ComputedNode, graph: GraphAI, transactionId: number, inputs: ResultData[]): void;

package/lib/transaction_log.js

@@ -7,9 +7,10 @@ class TransactionLog {
         this.nodeId = nodeId;
         this.state = type_1.NodeState.Waiting;
     }
-    initForComputedNode(node) {
+    initForComputedNode(node, graph) {
         this.agentId = node.getAgentId();
         this.params = node.params;
+        graph.appendLog(this);
     }
     onInjected(node, graph, injectFrom) {
         const isUpdating = "endTime" in this;

package/lib/type.d.ts

@@ -64,9 +64,10 @@ export type AgentFunctionContext<ParamsType = DefaultParamsType, InputDataType =
         agentId?: string;
     };
     graphData?: GraphData | string;
-    agents?: AgentFunctionDictonary;
+    agents?: AgentFunctionInfoDictonary;
     taskManager?: TaskManager;
     filterParams: AgentFilterParams;
+    agentFilters?: AgentFilterInfo[];
     log?: TransactionLog[];
 };
 export type AgentFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, InputDataType = DefaultInputData> = (context: AgentFunctionContext<ParamsType, InputDataType>) => Promise<ResultData<ResultType>>;
@@ -78,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>;
@@ -88,9 +88,11 @@ export type AgentFunctionInfo = {
         params: DefaultParamsType;
         result: any;
     }[];
+    skipTest?: boolean;
     description: string;
     category: string[];
     author: string;
     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

@@ -10,5 +10,33 @@ export declare const defaultTestContext: {
     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: {
+        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 agentTestRunner: (agentInfo: AgentFunctionInfo) => Promise<void>;
 export declare const agentFilterRunnerBuilder: (__agentFilters: AgentFilterInfo[]) => (context: AgentFunctionContext, agent: AgentFunction) => Promise<ResultData>;

package/lib/utils/test_utils.js

@@ -3,7 +3,7 @@ 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.getAgentInfo = exports.defaultAgentInfo = exports.defaultTestContext = void 0;
 const node_assert_1 = __importDefault(require("node:assert"));
 const node_test_1 = __importDefault(require("node:test"));
 exports.defaultTestContext = {
@@ -17,10 +17,33 @@ 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 } = agentInfo;
-    if (samples.length === 0) {
+    const { agent, samples, skipTest } = agentInfo;
+    if (samples.length === 0 || skipTest) {
         console.log(`test ${agentInfo.name}: No test`);
     }
     else {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.3.1",
+  "version": "0.4.0",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [
@@ -8,8 +8,8 @@
   ],
   "scripts": {
     "build": "rm -r lib/* && tsc && tsc-alias",
-    "eslint": "eslint --fix --ext .ts ./src ./tests ./samples",
-    "format": "prettier --write '{src,tests,samples}/**/*.ts' .eslintrc.js",
+    "eslint": "eslint",
+    "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",
     "b": "yarn run format && yarn run eslint && yarn run build",
@@ -30,26 +30,25 @@
     "@inquirer/prompts": "^5.0.0",
     "@types/cors": "^2.8.17",
     "@types/express": "^4.17.21",
-    "@types/node": "^20.8.7",
+    "@types/node": "^20.12.11",
     "@types/xml2js": "^0.4.14",
-    "@typescript-eslint/eslint-plugin": "^6.8.0",
-    "@typescript-eslint/parser": "^6.8.0",
     "arxiv-api-ts": "^1.0.3",
     "cors": "^2.8.5",
     "deepmerge": "^4.3.1",
     "dotenv": "^16.4.5",
-    "eslint": "^7.32.0 || ^8.2.0",
-    "eslint-plugin-import": "^2.25.2",
+    "eslint": "^9.2.0",
     "express": "^4.19.2",
+    "globals": "^15.2.0",
     "groq-sdk": "^0.3.3",
     "openai": "^4.12.4",
-    "prettier": "^3.0.3",
+    "prettier": "^3.2.5",
     "slashgpt": "^0.0.9",
     "tiktoken": "^1.0.14",
-    "ts-node": "^10.9.1",
+    "ts-node": "^10.9.2",
     "tsc-alias": "^1.8.8",
     "tsconfig-paths": "^4.2.0",
-    "typescript": "^5.2.2",
+    "typescript": "^5.4.5",
+    "typescript-eslint": "^7.8.0",
     "wikipedia": "^2.1.2",
     "xml2js": "^0.6.2",
     "yaml": "^2.4.1"

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;