graphai 0.5.0 → 0.5.2

package/README.md

@@ -190,12 +190,14 @@ nodes:
     inputs: [":source"] // == "sourceNode.query"
   database:
     agent: "nestedAgent"
-    inputs: [":question", ":projectId"]
+    inputs:
+      prompt: ":question"
+      projectId: ":projectId"
     graph:
       nodes:
         schema: // retrieves the database schema for the apecified projectId
           agent: "schemaAgent"
-          inputs: [":$1"]
+          inputs: [":projectId"]
         ... // issue query to the database and build an appropriate prompt with it.
         query: // send the generated prompt to the LLM
           agent: "llama3Agent"
@@ -296,7 +298,7 @@ nodes:
     value: [Steve Jobs, Elon Musk, Nikola Tesla]
   retriever:
     agent: "mapAgent"
-    inputs: [":people"]
+    inputs: { rows: ":people" }
     graph:
       nodes:
         query:
@@ -304,7 +306,7 @@ nodes:
           params:
             manifest:
               prompt: Describe about the person in less than 100 words
-          inputs: [":$0"]
+          inputs: [":row"]
 ```
 
 Here is the conceptual representation of this operation.

package/lib/graphai.d.ts

@@ -25,6 +25,7 @@ export declare class GraphAI {
     constructor(data: GraphData, agentFunctionInfoDictionary: AgentFunctionInfoDictionary, options?: GraphOptions);
     getAgentFunctionInfo(agentId?: string): import("./type").AgentFunctionInfo | {
         agent: () => Promise<null>;
+        inputs: null;
     };
     asString(): string;
     results<T = DefaultResultData>(all: boolean): ResultDataDictionary<T>;

package/lib/graphai.js

@@ -103,6 +103,7 @@ class GraphAI {
                 agent: async () => {
                     return null;
                 },
+                inputs: null,
             };
         }
         // We are not supposed to hit this error because the validator will catch it.

package/lib/index.d.ts

@@ -1,6 +1,5 @@
 export { GraphAI } from "./graphai";
-export { AgentFunction, AgentFunctionInfo, AgentFunctionInfoDictionary, AgentFunctionContext, GraphData, ResultDataDictionary, ResultData, NodeState, AgentFilterFunction, AgentFilterInfo, NodeData, StaticNodeData, ComputedNodeData, DefaultResultData, DefaultInputData, } from "./type";
+export { AgentFunction, AgentFunctionInfo, AgentFunctionInfoDictionary, AgentFunctionInfoSample, AgentFunctionContext, GraphData, ResultDataDictionary, ResultData, NodeState, AgentFilterFunction, AgentFilterInfo, NodeData, StaticNodeData, ComputedNodeData, DefaultResultData, DefaultInputData, } from "./type";
 export type { TransactionLog } from "./transaction_log";
-export { agentFilterRunnerBuilder } from "./utils/runner";
 export { defaultAgentInfo, agentInfoWrapper, defaultTestContext, strIntentionalError, assert, sleep } from "./utils/utils";
 export { ValidationError } from "./validators/common";

package/lib/index.js

@@ -1,12 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ValidationError = exports.sleep = exports.assert = exports.strIntentionalError = exports.defaultTestContext = exports.agentInfoWrapper = exports.defaultAgentInfo = exports.agentFilterRunnerBuilder = exports.NodeState = exports.GraphAI = void 0;
+exports.ValidationError = exports.sleep = exports.assert = exports.strIntentionalError = exports.defaultTestContext = exports.agentInfoWrapper = exports.defaultAgentInfo = exports.NodeState = exports.GraphAI = void 0;
 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; } });
-var runner_1 = require("./utils/runner");
-Object.defineProperty(exports, "agentFilterRunnerBuilder", { enumerable: true, get: function () { return runner_1.agentFilterRunnerBuilder; } });
 var utils_1 = require("./utils/utils");
 Object.defineProperty(exports, "defaultAgentInfo", { enumerable: true, get: function () { return utils_1.defaultAgentInfo; } });
 Object.defineProperty(exports, "agentInfoWrapper", { enumerable: true, get: function () { return utils_1.agentInfoWrapper; } });

package/lib/node.js

@@ -250,12 +250,14 @@ class ComputedNode extends Node {
             const context = {
                 params: params,
                 inputs: previousResults,
+                inputSchema: this.agentFunction ? undefined : this.graph.getAgentFunctionInfo(this.agentId)?.inputs,
                 namedInputs: {},
                 debugInfo: {
                     nodeId: this.nodeId,
                     agentId: this.agentId,
                     retry: this.retryCount,
                     verbose: this.graph.verbose,
+                    version: this.graph.version,
                 },
                 filterParams: this.filterParams,
                 agentFilters: this.graph.agentFilters,
@@ -283,7 +285,7 @@ class ComputedNode extends Node {
                 context.agents = this.graph.agentFunctionInfoDictionary;
             }
             if (this.console.before) {
-                console.log(this.console.before === true ? JSON.stringify(context.inputs, null, 2) : this.console.before);
+                console.log(this.console.before === true ? JSON.stringify(this.inputNames ? context.namedInputs : context.inputs, null, 2) : this.console.before);
             }
             const result = await this.agentFilterHandler(context, agentFunction);
             if (this.console.after) {

package/lib/type.d.ts

@@ -4,6 +4,7 @@ export declare enum NodeState {
     Waiting = "waiting",
     Queued = "queued",
     Executing = "executing",
+    ExecutingServer = "executing-server",
     Failed = "failed",
     TimedOut = "timed-out",
     Completed = "completed",
@@ -61,25 +62,27 @@ export type GraphOptions = {
     taskManager?: TaskManager | undefined;
     bypassAgentIds?: string[] | undefined;
 };
-export type AgentFunctionContext<ParamsType = DefaultParamsType, InputDataType = DefaultInputData> = {
+export type AgentFunctionContext<ParamsType = DefaultParamsType, InputDataType = DefaultInputData, NamedInputDataType = DefaultInputData> = {
     params: NodeDataParams<ParamsType>;
     inputs: Array<InputDataType>;
-    namedInputs: Record<string, any>;
+    inputSchema?: any;
+    namedInputs: NamedInputDataType;
     debugInfo: {
         verbose: boolean;
         nodeId: string;
         retry: number;
         agentId?: string;
+        version?: number;
     };
-    graphData?: GraphData | string;
+    graphData?: GraphData;
     agents?: AgentFunctionInfoDictionary;
     taskManager?: TaskManager;
     filterParams: AgentFilterParams;
     agentFilters?: AgentFilterInfo[];
     log?: TransactionLog[];
 };
-export type AgentFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, InputDataType = DefaultInputData> = (context: AgentFunctionContext<ParamsType, InputDataType>) => Promise<ResultData<ResultType>>;
-export type AgentFilterFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, InputDataType = DefaultInputData> = (context: AgentFunctionContext<ParamsType, InputDataType>, agent: AgentFunction) => Promise<ResultData<ResultType>>;
+export type AgentFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, InputDataType = DefaultInputData, NamedInputDataType = DefaultInputData> = (context: AgentFunctionContext<ParamsType, InputDataType, NamedInputDataType>) => Promise<ResultData<ResultType>>;
+export type AgentFilterFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, InputDataType = DefaultInputData, NamedInputDataType = DefaultInputData> = (context: AgentFunctionContext<ParamsType, InputDataType, NamedInputDataType>, agent: AgentFunction) => Promise<ResultData<ResultType>>;
 export type AgentFilterInfo = {
     name: string;
     agent: AgentFilterFunction;
@@ -87,19 +90,19 @@ export type AgentFilterInfo = {
     nodeIds?: string[];
     filterParams?: AgentFilterParams;
 };
+export type AgentFunctionInfoSample = {
+    inputs: any;
+    params: DefaultParamsType;
+    result: any;
+    graph?: GraphData;
+};
 export type AgentFunctionInfo = {
     name: string;
-    agent: AgentFunction<any, any, any>;
-    mock: AgentFunction<any, any, any>;
+    agent: AgentFunction<any, any, any, any>;
+    mock: AgentFunction<any, any, any, any>;
     inputs?: any;
     output?: any;
-    samples: {
-        inputs: any;
-        params: DefaultParamsType;
-        result: any;
-        graph?: GraphData;
-    }[];
-    skipTest?: boolean;
+    samples: AgentFunctionInfoSample[];
     description: string;
     category: string[];
     author: string;

package/lib/type.js

@@ -6,6 +6,7 @@ var NodeState;
     NodeState["Waiting"] = "waiting";
     NodeState["Queued"] = "queued";
     NodeState["Executing"] = "executing";
+    NodeState["ExecutingServer"] = "executing-server";
     NodeState["Failed"] = "failed";
     NodeState["TimedOut"] = "timed-out";
     NodeState["Completed"] = "completed";

package/lib/validators/relation_validator.js

@@ -14,7 +14,7 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
         if ("inputs" in nodeData && nodeData && nodeData.inputs) {
             if (Array.isArray(nodeData.inputs)) {
                 nodeData.inputs.forEach((inputNodeId) => {
-                    const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId, data.version ?? 0.02).nodeId;
+                    const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId, data.version ?? 0.2).nodeId;
                     if (sourceNodeId) {
                         if (!nodeIds.has(sourceNodeId)) {
                             throw new common_1.ValidationError(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${sourceNodeId}`);
@@ -47,7 +47,7 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
         const nodeData = data.nodes[staticNodeId];
         if ("value" in nodeData && nodeData.update) {
             const update = nodeData.update;
-            const updateNodeId = (0, utils_1.parseNodeName)(update, data.version ?? 0.02).nodeId;
+            const updateNodeId = (0, utils_1.parseNodeName)(update, data.version ?? 0.2).nodeId;
             if (!updateNodeId) {
                 throw new common_1.ValidationError("Update it a literal");
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.5.0",
+  "version": "0.5.2",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [

package/lib/utils/runner.d.ts

@@ -1,2 +0,0 @@
-import { AgentFunctionContext, AgentFunction, AgentFilterInfo, ResultData } from "../type";
-export declare const agentFilterRunnerBuilder: (__agentFilters: AgentFilterInfo[]) => (context: AgentFunctionContext, agent: AgentFunction) => Promise<ResultData>;

package/lib/utils/runner.js

@@ -1,20 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.agentFilterRunnerBuilder = void 0;
-// for test and server.
-const agentFilterRunnerBuilder = (__agentFilters) => {
-    const agentFilters = __agentFilters;
-    const agentFilterRunner = (context, agent) => {
-        let index = 0;
-        const next = (context) => {
-            const agentFilter = agentFilters[index++];
-            if (agentFilter) {
-                return agentFilter.agent(context, next);
-            }
-            return agent(context);
-        };
-        return next(context);
-    };
-    return agentFilterRunner;
-};
-exports.agentFilterRunnerBuilder = agentFilterRunnerBuilder;