graphai 0.4.1 → 0.4.3

package/lib/validators/relation_validator.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.relationValidator = void 0;
 const utils_1 = require("../utils/utils");
+const common_1 = require("../validators/common");
 const relationValidator = (data, staticNodeIds, computedNodeIds) => {
     const nodeIds = new Set(Object.keys(data.nodes));
     const pendings = {};
@@ -15,7 +16,7 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
                 const sourceNodeId = (0, utils_1.parseNodeName)(inputNodeId, data.version ?? 0.02).nodeId;
                 if (sourceNodeId) {
                     if (!nodeIds.has(sourceNodeId)) {
-                        throw new Error(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${sourceNodeId}`);
+                        throw new common_1.ValidationError(`Inputs not match: NodeId ${computedNodeId}, Inputs: ${sourceNodeId}`);
                     }
                     waitlist[sourceNodeId] === undefined && (waitlist[sourceNodeId] = new Set());
                     pendings[computedNodeId].add(sourceNodeId);
@@ -31,10 +32,10 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
             const update = nodeData.update;
             const updateNodeId = (0, utils_1.parseNodeName)(update, data.version ?? 0.02).nodeId;
             if (!updateNodeId) {
-                throw new Error("Update it a literal");
+                throw new common_1.ValidationError("Update it a literal");
             }
             if (!nodeIds.has(updateNodeId)) {
-                throw new Error(`Update not match: NodeId ${staticNodeId}, update: ${update}`);
+                throw new common_1.ValidationError(`Update not match: NodeId ${staticNodeId}, update: ${update}`);
             }
         }
     });
@@ -55,13 +56,13 @@ const relationValidator = (data, staticNodeIds, computedNodeIds) => {
     };
     let runningQueue = cycle(staticNodeIds);
     if (runningQueue.length === 0) {
-        throw new Error("No Initial Runnning Node");
+        throw new common_1.ValidationError("No Initial Runnning Node");
     }
     do {
         runningQueue = cycle(runningQueue);
     } while (runningQueue.length > 0);
     if (Object.keys(pendings).length > 0) {
-        throw new Error("Some nodes are not executed: " + Object.keys(pendings).join(", "));
+        throw new common_1.ValidationError("Some nodes are not executed: " + Object.keys(pendings).join(", "));
     }
 };
 exports.relationValidator = relationValidator;

package/lib/validators/static_node_validator.js

@@ -5,7 +5,7 @@ const common_1 = require("../validators/common");
 const staticNodeValidator = (nodeData) => {
     Object.keys(nodeData).forEach((key) => {
         if (!common_1.staticNodeAttributeKeys.includes(key)) {
-            throw new Error("Static node does not allow " + key);
+            throw new common_1.ValidationError("Static node does not allow " + key);
         }
     });
     return true;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphai",
-  "version": "0.4.1",
+  "version": "0.4.3",
   "description": "Asynchronous data flow execution engine for agentic AI apps.",
   "main": "lib/index.js",
   "files": [
@@ -9,6 +9,7 @@
   "scripts": {
     "build": "rm -r lib/* && tsc && tsc-alias",
     "eslint": "eslint",
+    "apiDoc": " npx typedoc src/index.ts --out apiDoc",
     "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",
@@ -27,6 +28,7 @@
   },
   "homepage": "https://github.com/receptron/graphai#readme",
   "devDependencies": {
+    "@anthropic-ai/sdk": "^0.21.0",
     "@inquirer/prompts": "^5.0.0",
     "@types/cors": "^2.8.17",
     "@types/express": "^4.17.21",
@@ -47,6 +49,7 @@
     "ts-node": "^10.9.2",
     "tsc-alias": "^1.8.8",
     "tsconfig-paths": "^4.2.0",
+    "typedoc": "^0.25.13",
     "typescript": "^5.4.5",
     "typescript-eslint": "^7.8.0",
     "wikipedia": "^2.1.2",

package/lib/experimental_agents/function_agent.d.ts

@@ -1,42 +0,0 @@
-import { AgentFunction } from "../graphai";
-export declare const functionAgent: AgentFunction<{
-    function: (...args: any[]) => any;
-}>;
-declare const functionAgentInfo: {
-    name: string;
-    agent: AgentFunction<{
-        function: (...args: any[]) => any;
-    }>;
-    mock: AgentFunction<{
-        function: (...args: any[]) => any;
-    }>;
-    samples: ({
-        inputs: {
-            model: string;
-            maker: string;
-            range: number;
-            price: number;
-        }[];
-        params: {
-            function: (info: Record<string, any>) => string;
-        };
-        result: string;
-    } | {
-        inputs: string[];
-        params: {
-            function: (str: string) => any;
-        };
-        result: {
-            model: string;
-            maker: string;
-            range: number;
-            price: number;
-        };
-    })[];
-    description: string;
-    category: string[];
-    author: string;
-    repository: string;
-    license: string;
-};
-export default functionAgentInfo;

package/lib/experimental_agents/function_agent.js

@@ -1,40 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.functionAgent = void 0;
-const functionAgent = async ({ inputs, params }) => {
-    return params.function(...inputs);
-};
-exports.functionAgent = functionAgent;
-const carInfo = { model: "Model 3", maker: "Tesla", range: 300, price: 35000 };
-const functionAgentInfo = {
-    name: "functionAgent",
-    agent: exports.functionAgent,
-    mock: exports.functionAgent,
-    samples: [
-        {
-            inputs: [carInfo],
-            params: {
-                function: (info) => {
-                    const { model, maker, range, price } = info;
-                    return `A ${maker} ${model} has the range of ${range} miles. It costs $${price}.`;
-                },
-            },
-            result: "A Tesla Model 3 has the range of 300 miles. It costs $35000.",
-        },
-        {
-            inputs: [JSON.stringify(carInfo)],
-            params: {
-                function: (str) => {
-                    return JSON.parse(str);
-                },
-            },
-            result: carInfo,
-        },
-    ],
-    description: "It allows developers to implement the agent function within the graph itself.",
-    category: ["data"],
-    author: "Receptron team",
-    repository: "https://github.com/receptron/graphai",
-    license: "MIT",
-};
-exports.default = functionAgentInfo;

package/lib/experimental_agents/llm_agents/groq_stream_agent.d.ts

@@ -1,42 +0,0 @@
-import { AgentFunction } from "../../graphai";
-import { Groq } from "groq-sdk";
-export declare const groqStreamAgent: AgentFunction<{
-    model: string;
-    query?: string;
-    system?: string;
-    verbose?: boolean;
-    tools?: Groq.Chat.CompletionCreateParams.Tool[];
-    temperature?: number;
-    max_tokens?: number;
-    tool_choice?: Groq.Chat.CompletionCreateParams.ToolChoice;
-}, any, string | Array<Groq.Chat.CompletionCreateParams.Message>>;
-declare const groqStreamAgentInfo: {
-    name: string;
-    agent: AgentFunction<{
-        model: string;
-        query?: string | undefined;
-        system?: string | undefined;
-        verbose?: boolean | undefined;
-        tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
-        temperature?: number | undefined;
-        max_tokens?: number | undefined;
-        tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
-    }, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
-    mock: AgentFunction<{
-        model: string;
-        query?: string | undefined;
-        system?: string | undefined;
-        verbose?: boolean | undefined;
-        tools?: Groq.Chat.Completions.CompletionCreateParams.Tool[] | undefined;
-        temperature?: number | undefined;
-        max_tokens?: number | undefined;
-        tool_choice?: Groq.Chat.Completions.CompletionCreateParams.ToolChoice | undefined;
-    }, any, string | Groq.Chat.Completions.CompletionCreateParams.Message[]>;
-    samples: never[];
-    description: string;
-    category: string[];
-    author: string;
-    repository: string;
-    license: string;
-};
-export default groqStreamAgentInfo;

package/lib/experimental_agents/llm_agents/groq_stream_agent.js

@@ -1,84 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.groqStreamAgent = void 0;
-const groq_sdk_1 = require("groq-sdk");
-const utils_1 = require("../../utils/utils");
-const groq = process.env.GROQ_API_KEY ? new groq_sdk_1.Groq({ apiKey: process.env.GROQ_API_KEY }) : undefined;
-//
-// This agent takes two optional inputs, and following parameters.
-// inputs:
-// - [0]: query string (typically from the user), optional
-// - [1]: array of messages from previous conversation, optional
-//
-// params:
-// - model: LLM model (Llama3-8b-8192, Llama3-70b-8192, Mixtral-8x7b-32768), required.
-// - query: Additional query string from the app to prepend the query from the user, optional.
-// - system: System prompt (ignored if inputs[1] is specified), optional
-// - tools: Function definitions, optional
-// - tool_choice: Tool choice parameter, optional (default = "auto")
-// - temperature: Controls randomness of responses, optional (default = 0.7)
-// - max_tokens: The maximum number of tokens that the model can process in a single response, optional.
-// - verbose: dumps the message array to the debug console, before sending it the LLM.
-//
-// https://console.groq.com/docs/quickstart
-//
-const groqStreamAgent = async ({ params, inputs, filterParams }) => {
-    (0, utils_1.assert)(groq !== undefined, "The GROQ_API_KEY environment variable is missing.");
-    const { verbose, query, system, tools, tool_choice, max_tokens, temperature } = params;
-    const [input_query, previous_messages] = inputs;
-    // Notice that we ignore params.system if previous_message exists.
-    const messages = previous_messages && Array.isArray(previous_messages) ? previous_messages : system ? [{ role: "system", content: system }] : [];
-    const content = (query ? [query] : []).concat(input_query && typeof input_query === "string" ? [input_query] : []).join("\n");
-    if (content) {
-        messages.push({
-            role: "user",
-            content,
-        });
-    }
-    if (verbose) {
-        console.log(messages);
-    }
-    const options = {
-        messages,
-        model: params.model,
-        temperature: temperature ?? 0.7,
-        stream: true,
-    };
-    if (max_tokens) {
-        options.max_tokens = max_tokens;
-    }
-    if (tools) {
-        options.tools = tools;
-        options.tool_choice = tool_choice ?? "auto";
-    }
-    const stream = await groq.chat.completions.create(options);
-    let lastMessage = null;
-    const contents = [];
-    for await (const message of stream) {
-        const token = message.choices[0].delta.content;
-        if (token) {
-            if (filterParams && filterParams.streamTokenCallback) {
-                filterParams.streamTokenCallback(token);
-            }
-            contents.push(token);
-        }
-        lastMessage = message;
-    }
-    if (lastMessage) {
-        lastMessage.choices[0]["message"] = [{ role: "assistant", content: contents.join("") }];
-    }
-    return lastMessage;
-};
-exports.groqStreamAgent = groqStreamAgent;
-const groqStreamAgentInfo = {
-    name: "groqStreamAgent",
-    agent: exports.groqStreamAgent,
-    mock: exports.groqStreamAgent,
-    samples: [],
-    description: "Groq Stream Agent",
-    category: ["llm"],
-    author: "Receptron team",
-    repository: "https://github.com/receptron/graphai",
-    license: "MIT",
-};
-exports.default = groqStreamAgentInfo;