graphai 0.6.9 → 0.6.11

package/lib/node.d.ts

@@ -20,6 +20,7 @@ export declare class ComputedNode extends Node {
     readonly params: NodeDataParams;
     private readonly filterParams;
     readonly nestedGraph?: GraphData | DataSource;
+    private readonly config?;
     readonly retryLimit: number;
     retryCount: number;
     private readonly agentId?;

package/lib/node.js

@@ -40,7 +40,7 @@ class Node {
         }
         else if (this.console.after) {
             if ((0, utils_2.isObject)(this.console.after)) {
-                console.log(JSON.stringify((0, result_1.resultsOf)(this.console.after, { self: this }, this.graph.propFunctions, true), null, 2));
+                console.log(JSON.stringify((0, result_1.resultsOf)(this.console.after, { self: { result } }, this.graph.propFunctions, true), null, 2));
             }
             else {
                 console.log(this.console.after);
@@ -74,6 +74,7 @@ class ComputedNode extends Node {
             const agent = data.agent;
             this.agentFunction = async ({ namedInputs, params }) => agent(namedInputs, params);
         }
+        this.config = this.agentId ? (data.graph ? this.graph.config : ((this.graph.config ?? {})[this.agentId] ?? {})) : {};
         this.anyInput = data.anyInput ?? false;
         this.inputs = data.inputs;
         this.output = data.output;
@@ -241,7 +242,7 @@ class ComputedNode extends Node {
                         agentFilters: this.graph.agentFilters,
                         taskManager: this.graph.taskManager,
                         bypassAgentIds: this.graph.bypassAgentIds,
-                        config: this.graph.config,
+                        config: this.config,
                         graphLoader: this.graph.graphLoader,
                     },
                     onLogCallback: this.graph.onLogCallback,
@@ -314,7 +315,7 @@ class ComputedNode extends Node {
             cacheType: this.agentFunction ? undefined : this.graph.getAgentFunctionInfo(agentId)?.cacheType,
             filterParams: this.filterParams,
             agentFilters: this.graph.agentFilters,
-            config: this.graph.config,
+            config: this.config,
             log: localLog,
         };
         return context;

package/lib/type.d.ts

@@ -13,8 +13,11 @@ export declare enum NodeState {
 }
 export type DefaultResultData = Record<string, any> | string | number | boolean | Array<DefaultResultData>;
 export type DefaultInputData = Record<string, any>;
+export type DefaultConfigData = Record<string, any>;
 export type ResultData<ResultType = DefaultResultData> = ResultType | undefined;
 export type ResultDataDictionary<ResultType = DefaultResultData> = Record<string, ResultData<ResultType>>;
+export type ConfigData<ConfigType = DefaultConfigData> = ConfigType;
+export type ConfigDataDictionary<ConfigType = DefaultConfigData> = Record<string, ConfigType>;
 export type DefaultParamsType = Record<string, any>;
 export type NodeDataParams<ParamsType = DefaultParamsType> = ParamsType;
 export type PassThrough = Record<string, any>;
@@ -78,11 +81,11 @@ export type GraphOptions = {
     agentFilters?: AgentFilterInfo[] | undefined;
     taskManager?: TaskManager | undefined;
     bypassAgentIds?: string[] | undefined;
-    config?: Record<string, unknown>;
+    config?: ConfigDataDictionary;
     graphLoader?: GraphDataLoader;
 };
 export type CacheTypes = "pureAgent" | "impureAgent";
-export type AgentFunctionContext<ParamsType = DefaultParamsType, NamedInputDataType = DefaultInputData> = {
+export type AgentFunctionContext<ParamsType = DefaultParamsType, NamedInputDataType = DefaultInputData, ConfigType = DefaultConfigData> = {
     params: NodeDataParams<ParamsType>;
     inputSchema?: any;
     namedInputs: NamedInputDataType;
@@ -104,9 +107,9 @@ export type AgentFunctionContext<ParamsType = DefaultParamsType, NamedInputDataT
     filterParams: AgentFilterParams;
     agentFilters?: AgentFilterInfo[];
     log?: TransactionLog[];
-    config?: Record<string, unknown>;
+    config?: ConfigType;
 };
-export type AgentFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, NamedInputDataType = DefaultInputData> = (context: AgentFunctionContext<ParamsType, NamedInputDataType>) => Promise<ResultData<ResultType>>;
+export type AgentFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, NamedInputDataType = DefaultInputData, ConfigType = DefaultConfigData> = (context: AgentFunctionContext<ParamsType, NamedInputDataType, ConfigType>) => Promise<ResultData<ResultType>>;
 export type AgentFilterFunction<ParamsType = DefaultParamsType, ResultType = DefaultResultData, NamedInputDataType = DefaultInputData> = (context: AgentFunctionContext<ParamsType, NamedInputDataType>, agent: AgentFunction) => Promise<ResultData<ResultType>>;
 export type AgentFilterInfo = {
     name: string;
@@ -123,12 +126,13 @@ export type AgentFunctionInfoSample = {
 };
 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;
     outputFormat?: any;
     params?: any;
+    config?: any;
     samples: AgentFunctionInfoSample[];
     description: string;
     category: string[];

package/lib/validators/relation_validator.d.ts

@@ -1,2 +1,2 @@
 import { GraphData } from "../type";
-export declare const relationValidator: (data: GraphData, staticNodeIds: string[], computedNodeIds: string[]) => void;
+export declare const relationValidator: (graphData: GraphData, staticNodeIds: string[], computedNodeIds: string[]) => void;

package/lib/validators/relation_validator.js

@@ -4,13 +4,13 @@ exports.relationValidator = void 0;
 const utils_1 = require("../utils/utils");
 const common_1 = require("../validators/common");
 const nodeUtils_1 = require("../utils/nodeUtils");
-const relationValidator = (data, staticNodeIds, computedNodeIds) => {
-    const nodeIds = new Set(Object.keys(data.nodes));
+const relationValidator = (graphData, staticNodeIds, computedNodeIds) => {
+    const nodeIds = new Set(Object.keys(graphData.nodes));
     const pendings = {};
     const waitlist = {};
     // validate input relation and set pendings and wait list
     computedNodeIds.forEach((computedNodeId) => {
-        const nodeData = data.nodes[computedNodeId];
+        const nodeData = graphData.nodes[computedNodeId];
         pendings[computedNodeId] = new Set();
         const dataSourceValidator = (sourceType, sourceNodeIds) => {
             sourceNodeIds.forEach((sourceNodeId) => {
@@ -29,6 +29,10 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
                 const sourceNodeIds = (0, nodeUtils_1.dataSourceNodeIds)((0, nodeUtils_1.inputs2dataSources)(nodeData.inputs));
                 dataSourceValidator("Inputs", sourceNodeIds);
             }
+            if (nodeData.params) {
+                const sourceNodeIds = (0, nodeUtils_1.dataSourceNodeIds)((0, nodeUtils_1.inputs2dataSources)(nodeData.params));
+                dataSourceValidator("Params", sourceNodeIds);
+            }
             if (nodeData.if) {
                 const sourceNodeIds = (0, nodeUtils_1.dataSourceNodeIds)((0, nodeUtils_1.inputs2dataSources)({ if: nodeData.if }));
                 dataSourceValidator("If", sourceNodeIds);
@@ -41,11 +45,15 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
                 const sourceNodeIds = (0, nodeUtils_1.dataSourceNodeIds)((0, nodeUtils_1.inputs2dataSources)({ graph: nodeData.graph }));
                 dataSourceValidator("Graph", sourceNodeIds);
             }
+            if (typeof nodeData.agent === "string" && nodeData.agent[0] === ":") {
+                const sourceNodeIds = (0, nodeUtils_1.dataSourceNodeIds)((0, nodeUtils_1.inputs2dataSources)({ agent: nodeData.agent }));
+                dataSourceValidator("Agent", sourceNodeIds);
+            }
         }
     });
     // TODO. validate update
     staticNodeIds.forEach((staticNodeId) => {
-        const nodeData = data.nodes[staticNodeId];
+        const nodeData = graphData.nodes[staticNodeId];
         if ("value" in nodeData && nodeData.update) {
             const update = nodeData.update;
             const updateNodeId = (0, utils_1.parseNodeName)(update).nodeId;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.6.9",
+  "version": "0.6.11",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/bundle.cjs.js",
   "module": "lib/bundle.esm.js",
@@ -27,7 +27,7 @@
   },
   "homepage": "https://github.com/receptron/graphai#readme",
   "devDependencies": {
-    "typedoc": "^0.27.4"
+    "typedoc": "^0.27.5"
   },
   "types": "./lib/index.d.ts",
   "directories": {