graphai 0.5.0 → 0.5.1

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/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

@@ -61,25 +61,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;
@@ -89,8 +91,8 @@ export type AgentFilterInfo = {
 };
 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: {
@@ -99,7 +101,6 @@ export type AgentFunctionInfo = {
         result: any;
         graph?: GraphData;
     }[];
-    skipTest?: boolean;
     description: string;
     category: string[];
     author: string;

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.1",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [