graphai 0.4.1 → 0.4.3

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const fetchAgent: AgentFunction<{
     debug?: boolean;
     type?: string;

package/lib/experimental_agents/service_agents/fetch_agent.js

@@ -87,7 +87,7 @@ const fetchAgentInfo = {
         },
     ],
     description: "Retrieves JSON data from the specified URL",
-    category: ["data"],
+    category: ["service"],
     author: "Receptron",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const wikipediaAgent: AgentFunction<{
     lang?: string;
     summary?: boolean;

package/lib/experimental_agents/service_agents/wikipedia.js

@@ -30,7 +30,7 @@ const wikipediaAgentInfo = {
     agent: exports.wikipediaAgent,
     mock: exports.wikipediaAgent,
     description: "Retrieves data from wikipedia",
-    category: ["data"],
+    category: ["service"],
     samples: [],
     author: "Receptron",
     repository: "https://github.com/receptron/graphai",

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const sleeperAgent: AgentFunction<{
     duration?: number;
     value?: Record<string, any>;
@@ -15,7 +15,7 @@ declare const sleeperAgentInfo: {
     }>;
     samples: never[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/sleeper_agents/sleeper_agent.js

@@ -20,7 +20,7 @@ const sleeperAgentInfo = {
     mock: exports.sleeperAgent,
     samples: [],
     description: "sleeper Agent",
-    category: [],
+    category: ["sleeper"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 export declare const sleeperAgentDebug: AgentFunction<{
     duration: number;
     value?: Record<string, any>;
@@ -18,7 +18,7 @@ declare const sleeperAgentDebugInfo: {
     }>;
     samples: never[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/sleeper_agents/sleeper_agent_debug.js

@@ -23,7 +23,7 @@ const sleeperAgentDebugInfo = {
     mock: exports.sleeperAgentDebug,
     samples: [],
     description: "sleeper debug Agent",
-    category: [],
+    category: ["sleeper"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -0,0 +1,38 @@
+import { AgentFunction } from "../../index";
+export declare const jsonParserAgent: AgentFunction<{
+    stringify: boolean;
+}, any, any>;
+declare const jsonParserAgentInfo: {
+    name: string;
+    agent: AgentFunction<{
+        stringify: boolean;
+    }, any, any>;
+    mock: AgentFunction<{
+        stringify: boolean;
+    }, any, any>;
+    samples: ({
+        inputs: {
+            apple: string;
+            lemon: string;
+        }[];
+        params: {
+            stringify: boolean;
+        };
+        result: string;
+    } | {
+        inputs: string[];
+        params: {
+            stringify?: undefined;
+        };
+        result: {
+            apple: string;
+            lemon: string;
+        };
+    })[];
+    description: string;
+    category: string[];
+    author: string;
+    repository: string;
+    license: string;
+};
+export default jsonParserAgentInfo;

package/lib/experimental_agents/string_agents/json_parser_agent.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.jsonParserAgent = void 0;
+const jsonParserAgent = async ({ params, inputs }) => {
+    if (params.stringify) {
+        return JSON.stringify(inputs[0], null, 2);
+    }
+    const match = ("\n" + inputs[0]).match(/\n```[a-zA-z]*([\s\S]*?)\n```/);
+    if (match) {
+        return JSON.parse(match[1]);
+    }
+    return JSON.parse(inputs[0]);
+};
+exports.jsonParserAgent = jsonParserAgent;
+const sample_object = { apple: "red", lemon: "yellow" };
+// for test and document
+const jsonParserAgentInfo = {
+    name: "jsonParserAgent",
+    agent: exports.jsonParserAgent,
+    mock: exports.jsonParserAgent,
+    samples: [
+        {
+            inputs: [sample_object],
+            params: { stringify: true },
+            result: JSON.stringify(sample_object, null, 2),
+        },
+        {
+            inputs: [JSON.stringify(sample_object, null, 2)],
+            params: {},
+            result: sample_object,
+        },
+    ],
+    description: "Template agent",
+    category: ["string"],
+    author: "Satoshi Nakajima",
+    repository: "https://github.com/receptron/graphai",
+    license: "MIT",
+};
+exports.default = jsonParserAgentInfo;

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

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

package/lib/experimental_agents/string_agents/packages.js

@@ -3,8 +3,10 @@ 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;
+exports.jsonParserAgent = 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;
+const json_parser_agent_1 = __importDefault(require("../../experimental_agents/string_agents/json_parser_agent"));
+exports.jsonParserAgent = json_parser_agent_1.default;

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

@@ -1,40 +1,9 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const stringSplitterAgent: AgentFunction<{
     chunkSize?: number;
     overlap?: number;
 }, {
     contents: Array<string>;
 }, string>;
-declare const stringSplitterAgentInfo: {
-    name: string;
-    agent: AgentFunction<{
-        chunkSize?: number | undefined;
-        overlap?: number | undefined;
-    }, {
-        contents: Array<string>;
-    }, string>;
-    mock: AgentFunction<{
-        chunkSize?: number | undefined;
-        overlap?: number | undefined;
-    }, {
-        contents: Array<string>;
-    }, string>;
-    samples: {
-        inputs: string[];
-        params: {
-            chunkSize: number;
-        };
-        result: {
-            contents: string[];
-            count: number;
-            chunkSize: number;
-            overlap: number;
-        };
-    }[];
-    description: string;
-    category: never[];
-    author: string;
-    repository: string;
-    license: string;
-};
+declare const stringSplitterAgentInfo: AgentFunctionInfo;
 export default stringSplitterAgentInfo;

package/lib/experimental_agents/string_agents/string_splitter_agent.js

@@ -57,7 +57,7 @@ const stringSplitterAgentInfo = {
         },
     ],
     description: "This agent strip one long string into chunks using following parameters",
-    category: [],
+    category: ["string"],
     author: "Satoshi Nakajima",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,24 +1,68 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
+type StringTemplate = string | Record<string, string>;
+type StringTemplateObject = StringTemplate | StringTemplate[] | Record<string, StringTemplate>;
 export declare const stringTemplateAgent: AgentFunction<{
-    template: string;
-}, string, string>;
+    template: StringTemplateObject;
+}, StringTemplateObject, string>;
 declare const stringTemplateAgentInfo: {
     name: string;
     agent: AgentFunction<{
-        template: string;
-    }, string, string>;
+        template: StringTemplateObject;
+    }, StringTemplateObject, string>;
     mock: AgentFunction<{
-        template: string;
-    }, string, string>;
-    samples: {
+        template: StringTemplateObject;
+    }, StringTemplateObject, string>;
+    samples: ({
         inputs: string[];
         params: {
             template: string;
         };
         result: string;
-    }[];
+    } | {
+        inputs: string[];
+        params: {
+            template: string[];
+        };
+        result: string[];
+    } | {
+        inputs: string[];
+        params: {
+            template: {
+                apple: string;
+                lemon: string;
+            };
+        };
+        result: {
+            apple: string;
+            lemon: string;
+        };
+    } | {
+        inputs: string[];
+        params: {
+            template: {
+                apple: string;
+                lemon: string;
+            }[];
+        };
+        result: {
+            apple: string;
+            lemon: string;
+        }[];
+    } | {
+        inputs: string[];
+        params: {
+            template: {
+                apple: string;
+                lemon: string[];
+            };
+        };
+        result: {
+            apple: string;
+            lemon: string[];
+        };
+    })[];
     description: string;
-    category: never[];
+    category: string[];
     author: string;
     repository: string;
     license: string;

package/lib/experimental_agents/string_agents/string_template_agent.js

@@ -1,18 +1,28 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.stringTemplateAgent = void 0;
-// see example
-//  tests/agents/test_string_agent.ts
+const processTemplate = (template, match, input) => {
+    if (typeof template === "string") {
+        return template.replace(match, input);
+    }
+    else if (Array.isArray(template)) {
+        return template.map((item) => processTemplate(item, match, input));
+    }
+    return Object.keys(template).reduce((tmp, key) => {
+        tmp[key] = processTemplate(template[key], match, input);
+        return tmp;
+    }, {});
+};
 const stringTemplateAgent = async ({ params, inputs }) => {
-    const content = inputs.reduce((template, input, index) => {
-        return template.replace("${" + index + "}", input);
+    if (params.template === undefined) {
+        console.warn("warning: stringTemplateAgent no template");
+    }
+    return inputs.reduce((template, input, index) => {
+        return processTemplate(template, "${" + index + "}", input);
     }, params.template);
-    return content;
 };
 exports.stringTemplateAgent = stringTemplateAgent;
 const sampleInput = ["hello", "test"];
-const sampleParams = { template: "${0}: ${1}" };
-const sampleResult = "hello: test";
 // for test and document
 const stringTemplateAgentInfo = {
     name: "stringTemplateAgent",
@@ -21,12 +31,32 @@ const stringTemplateAgentInfo = {
     samples: [
         {
             inputs: sampleInput,
-            params: sampleParams,
-            result: sampleResult,
+            params: { template: "${0}: ${1}" },
+            result: "hello: test",
+        },
+        {
+            inputs: sampleInput,
+            params: { template: ["${0}: ${1}", "${1}: ${0}"] },
+            result: ["hello: test", "test: hello"],
+        },
+        {
+            inputs: sampleInput,
+            params: { template: { apple: "${0}", lemon: "${1}" } },
+            result: { apple: "hello", lemon: "test" },
+        },
+        {
+            inputs: sampleInput,
+            params: { template: [{ apple: "${0}", lemon: "${1}" }] },
+            result: [{ apple: "hello", lemon: "test" }],
+        },
+        {
+            inputs: sampleInput,
+            params: { template: { apple: "${0}", lemon: ["${1}"] } },
+            result: { apple: "hello", lemon: ["test"] },
         },
     ],
     description: "Template agent",
-    category: [],
+    category: ["string"],
     author: "Satoshi Nakajima",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

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

package/lib/experimental_agents/string_agents/vanilla.js

@@ -3,8 +3,10 @@ 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;
+exports.jsonParserAgent = 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;
+const json_parser_agent_1 = __importDefault(require("../../experimental_agents/string_agents/json_parser_agent"));
+exports.jsonParserAgent = json_parser_agent_1.default;

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { AgentFunctionInfo } from "../../type";
 export declare const bypassAgent: AgentFunction<{
     flat?: number;

package/lib/experimental_agents/test_agents/bypass_agent.js

@@ -51,7 +51,7 @@ const bypassAgentInfo = {
         },
     ],
     description: "bypass agent",
-    category: [],
+    category: ["test"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../../graphai";
+import { AgentFunction } from "../../index";
 import { AgentFunctionInfo } from "../../type";
 export declare const copy2ArrayAgent: AgentFunction<{
     count: number;

package/lib/experimental_agents/test_agents/copy2array_agent.js

@@ -31,7 +31,7 @@ const copy2ArrayAgentInfo = {
         },
     ],
     description: "Copy2Array agent",
-    category: [],
+    category: ["test"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const copyMessageAgent: AgentFunction<{
     count: number;
     message: string;

package/lib/experimental_agents/test_agents/copy_message_agent.js

@@ -22,7 +22,7 @@ const copyMessageAgentInfo = {
         },
     ],
     description: "CopyMessage agent",
-    category: [],
+    category: ["test"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const countingAgent: AgentFunction<{
     count: number;
 }, {

package/lib/experimental_agents/test_agents/counting_agent.js

@@ -22,7 +22,7 @@ const countingAgentInfo = {
         },
     ],
     description: "Counting agent",
-    category: [],
+    category: ["test"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const echoAgent: AgentFunction;
 declare const echoAgentInfo: AgentFunctionInfo;
 export default echoAgentInfo;

package/lib/experimental_agents/test_agents/echo_agent.js

@@ -13,9 +13,20 @@ const echoAgentInfo = {
     name: "echoAgent",
     agent: exports.echoAgent,
     mock: exports.echoAgent,
-    samples: [],
+    samples: [
+        {
+            inputs: [],
+            params: { message: "this is test" },
+            result: { message: "this is test" },
+        },
+        {
+            inputs: [],
+            params: { message: "If you add filterParams option, it will respond to filterParams", filterParams: true },
+            result: {},
+        },
+    ],
     description: "Echo agent",
-    category: [],
+    category: ["test"],
     author: "Satoshi Nakajima",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const mergeNodeIdAgent: AgentFunction;
 declare const mergeNodeIdAgentInfo: AgentFunctionInfo;
 export default mergeNodeIdAgentInfo;

package/lib/experimental_agents/test_agents/merge_node_id_agent.js

@@ -24,7 +24,7 @@ const mergeNodeIdAgentInfo = {
         },
     ],
     description: "merge node id agent",
-    category: [],
+    category: ["test"],
     author: "Receptron team",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",

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

@@ -1,5 +1,4 @@
-import { AgentFunction } from "../../graphai";
-import { AgentFunctionInfo } from "../../type";
+import { AgentFunction, AgentFunctionInfo } from "../../index";
 export declare const streamMockAgent: AgentFunction;
 declare const streamMockAgentInfo: AgentFunctionInfo;
 export default streamMockAgentInfo;

package/lib/experimental_agents/test_agents/stream_mock_agent.js

@@ -18,11 +18,18 @@ const streamMockAgentInfo = {
     name: "streamMockAgent",
     agent: exports.streamMockAgent,
     mock: exports.streamMockAgent,
-    samples: [],
+    samples: [
+        {
+            inputs: [],
+            params: { message: "this is test" },
+            result: { message: "this is test" },
+        },
+    ],
     description: "Stream mock agent",
-    category: [],
+    category: ["test"],
     author: "Isamu Arimoto",
     repository: "https://github.com/receptron/graphai",
     license: "MIT",
+    stream: true,
 };
 exports.default = streamMockAgentInfo;

package/lib/experimental_agents/token_agent.d.ts

@@ -1,4 +1,4 @@
-import { AgentFunction } from "../graphai";
+import { AgentFunction } from "../index";
 export declare const tokenBoundStringsAgent: AgentFunction<{
     limit?: number;
 }, {

package/lib/experimental_agents/vanilla.d.ts

@@ -4,6 +4,5 @@ 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 };
+export { stringEmbeddingsAgent };

package/lib/experimental_agents/vanilla.js

@@ -18,7 +18,7 @@ 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;
+exports.stringEmbeddingsAgent = void 0;
 // Please refrain from adding agents that require npm. Those should be added to the index.ts.
 __exportStar(require("./string_agents/vanilla"), exports);
 __exportStar(require("./array_agents/vanilla"), exports);
@@ -26,7 +26,5 @@ __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,4 @@
-export { AgentFunction, GraphData, AgentFunctionInfoDictonary } from "./type";
-import { AgentFunctionInfoDictonary, AgentFilterInfo, GraphData, DataSource, ResultDataDictonary, ResultData, DefaultResultData } from "./type";
+import { AgentFunctionInfoDictionary, AgentFilterInfo, GraphData, DataSource, ResultDataDictionary, ResultData, DefaultResultData } from "./type";
 import { TransactionLog } from "./transaction_log";
 import { ComputedNode, StaticNode } from "./node";
 import { TaskManager } from "./task_manager";
@@ -10,7 +9,7 @@ export declare class GraphAI {
     private readonly data;
     private readonly loop?;
     private readonly logs;
-    readonly agentFunctionInfoDictionary: AgentFunctionInfoDictonary;
+    readonly agentFunctionInfoDictionary: AgentFunctionInfoDictionary;
     readonly taskManager: TaskManager;
     readonly agentFilters: AgentFilterInfo[];
     readonly retryLimit?: number;
@@ -22,19 +21,19 @@ export declare class GraphAI {
     private createNodes;
     private getValueFromResults;
     private initializeNodes;
-    constructor(data: GraphData, agentFunctionInfoDictionary: AgentFunctionInfoDictonary, options?: {
+    constructor(data: GraphData, agentFunctionInfoDictionary: AgentFunctionInfoDictionary, options?: {
         agentFilters?: AgentFilterInfo[] | undefined;
         taskManager?: TaskManager | undefined;
     });
     getAgentFunctionInfo(agentId?: string): import("./type").AgentFunctionInfo;
     asString(): string;
-    results<T = DefaultResultData>(all: boolean): ResultDataDictonary<T>;
+    results<T = DefaultResultData>(all: boolean): ResultDataDictionary<T>;
     errors(): Record<string, Error>;
     private pushReadyNodesIntoQueue;
     private pushQueueIfReady;
     pushQueueIfReadyAndRunning(node: ComputedNode): void;
     pushQueue(node: ComputedNode): void;
-    run<T = DefaultResultData>(all?: boolean): Promise<ResultDataDictonary<T>>;
+    run<T = DefaultResultData>(all?: boolean): Promise<ResultDataDictionary<T>>;
     isRunning(): boolean;
     onExecutionComplete(node: ComputedNode): void;
     private processLoopIfNecessary;
@@ -45,3 +44,4 @@ export declare class GraphAI {
     injectValue(nodeId: string, value: ResultData, injectFrom?: string): void;
     resultsOf(sources: Array<DataSource>): ResultData[];
 }
+export {};