graphai 0.0.9 → 0.0.11

package/tests/graphai/test_dispatch.ts

@@ -1,41 +0,0 @@
-import { AgentFunction } from "@/graphai";
-import { fileTestRunner } from "~/utils/runner";
-
-import { sleep } from "~/utils/utils";
-
-import { testAgent } from "~/agents/agents";
-
-import test from "node:test";
-import assert from "node:assert";
-
-const dispatchAgent: AgentFunction<{ delay: number; fail: boolean }, Record<string, any>, Record<string, any>> = async (context) => {
-  const { nodeId, retry, params, inputs } = context;
-  console.log("executing", nodeId);
-  await sleep(params.delay / (retry + 1));
-
-  if (params.fail && retry < 2) {
-    const result = { [nodeId]: "failed" };
-    console.log("failed (intentional)", nodeId, retry);
-    throw new Error("Intentional Failure");
-  } else {
-    const result = inputs.reduce(
-      (result: Record<string, any>, input: Record<string, any>) => {
-        return { ...result, ...input };
-      },
-      { [nodeId]: "dispatch" },
-    );
-    console.log("completing", nodeId);
-    return { output1: result };
-  }
-};
-
-test("test dispatch", async () => {
-  const result = await fileTestRunner("/graphs/test_dispatch.yml", { test: testAgent, alt: dispatchAgent });
-  assert.deepStrictEqual(result, {
-    node1: { node1: "output" },
-    node20: { node2: "dispatch" },
-    node3: { node3: "output", node1: "output", node2: "dispatch" },
-    node4: { node4: "output", node3: "output", node1: "output", node2: "dispatch" },
-    node5: { node5: "output", node4: "output", node3: "output", node1: "output", node2: "dispatch" },
-  });
-});

package/tests/graphai/test_fork.ts

@@ -1,106 +0,0 @@
-import { GraphAI, AgentFunction } from "@/graphai";
-import { testAgent } from "~/agents/agents";
-import { graphDataTestRunner } from "~/utils/runner";
-
-import test from "node:test";
-import assert from "node:assert";
-
-const testAgent1: AgentFunction = async (context) => {
-  const { nodeId, retry, params, inputs } = context;
-  console.log("executing", nodeId, params, inputs);
-
-  const result = {
-    [nodeId]: [nodeId, inputs.map((a) => Object.values(a).flat())]
-      .flat()
-      .filter((a) => !!a)
-      .join(":"),
-  };
-  console.log("completing", nodeId, result);
-  return result;
-};
-
-test("test base", async () => {
-  const forkGraph = {
-    nodes: {
-      node1: {
-        params: {},
-      },
-      node2: {
-        params: {},
-        fork: 10,
-        inputs: ["node1"],
-      },
-      node3: {
-        params: {},
-        // fork: 10,
-        inputs: ["node2"],
-      },
-    },
-  };
-
-  const result = await graphDataTestRunner("fork.yml", forkGraph, testAgent1);
-  // console.log(result);
-  assert.deepStrictEqual(result, {
-    node1: { node1: "node1" },
-    node2_0: { node2_0: "node2_0:node1" },
-    node2_1: { node2_1: "node2_1:node1" },
-    node2_2: { node2_2: "node2_2:node1" },
-    node2_3: { node2_3: "node2_3:node1" },
-    node2_4: { node2_4: "node2_4:node1" },
-    node2_5: { node2_5: "node2_5:node1" },
-    node2_6: { node2_6: "node2_6:node1" },
-    node2_7: { node2_7: "node2_7:node1" },
-    node2_8: { node2_8: "node2_8:node1" },
-    node2_9: { node2_9: "node2_9:node1" },
-    node3: {
-      node3:
-        "node3:node2_0:node1:node2_1:node1:node2_2:node1:node2_3:node1:node2_4:node1:node2_5:node1:node2_6:node1:node2_7:node1:node2_8:node1:node2_9:node1",
-    },
-  });
-});
-
-test("test base", async () => {
-  const forkGraph = {
-    nodes: {
-      node1: {
-        params: {},
-      },
-      node2: {
-        params: {},
-        fork: 10,
-        inputs: ["node1"],
-      },
-      node3: {
-        params: {},
-        fork: 10,
-        inputs: ["node2"],
-      },
-    },
-  };
-
-  const result = await graphDataTestRunner("fork.yml", forkGraph, testAgent1);
-  // console.log(result);
-  assert.deepStrictEqual(result, {
-    node1: { node1: "node1" },
-    node2_0: { node2_0: "node2_0:node1" },
-    node2_1: { node2_1: "node2_1:node1" },
-    node2_2: { node2_2: "node2_2:node1" },
-    node2_3: { node2_3: "node2_3:node1" },
-    node2_4: { node2_4: "node2_4:node1" },
-    node2_5: { node2_5: "node2_5:node1" },
-    node2_6: { node2_6: "node2_6:node1" },
-    node2_7: { node2_7: "node2_7:node1" },
-    node2_8: { node2_8: "node2_8:node1" },
-    node2_9: { node2_9: "node2_9:node1" },
-    node3_0: { node3_0: "node3_0:node2_0:node1" },
-    node3_1: { node3_1: "node3_1:node2_1:node1" },
-    node3_2: { node3_2: "node3_2:node2_2:node1" },
-    node3_3: { node3_3: "node3_3:node2_3:node1" },
-    node3_4: { node3_4: "node3_4:node2_4:node1" },
-    node3_5: { node3_5: "node3_5:node2_5:node1" },
-    node3_6: { node3_6: "node3_6:node2_6:node1" },
-    node3_7: { node3_7: "node3_7:node2_7:node1" },
-    node3_8: { node3_8: "node3_8:node2_8:node1" },
-    node3_9: { node3_9: "node3_9:node2_9:node1" },
-  });
-});

package/tests/graphai/test_http_client.ts

@@ -1,40 +0,0 @@
-import { GraphAI, AgentFunction } from "@/graphai";
-import { graphDataTestRunner } from "~/utils/runner";
-
-import test from "node:test";
-import assert from "node:assert";
-
-const httpClientAgent: AgentFunction<Record<string, string>> = async (context) => {
-  const { nodeId, retry, params, inputs } = context;
-  console.log("executing", nodeId, params, inputs);
-
-  const response = await fetch(params.url);
-  const result = await response.json();
-
-  console.log("completing", nodeId, result);
-  return result;
-};
-
-const graph_data = {
-  nodes: {
-    node1: {
-      params: {
-        url: "http://127.0.0.1:8080/llm.json",
-      },
-    },
-    node2: {
-      params: {
-        url: "http://127.0.0.1:8080/llm2.json",
-      },
-      inputs: ["node1"],
-    },
-  },
-};
-
-test("test http client", async () => {
-  const result = await graphDataTestRunner("http.log", graph_data, httpClientAgent);
-  assert.deepStrictEqual(result, {
-    node1: { result: true, messages: ["hello"] },
-    node2: { result: true, messages: ["hello2"] },
-  });
-});

package/tests/graphai/test_multiple_functions.ts

@@ -1,46 +0,0 @@
-import path from "path";
-import { GraphAI, AgentFunction } from "@/graphai";
-import { readGraphaiData } from "~/utils/file_utils";
-import { fileTestRunner } from "~/utils/runner";
-
-import test from "node:test";
-import assert from "node:assert";
-
-const testAgent1: AgentFunction = async (context) => {
-  const { nodeId, retry, params } = context;
-  console.log("executing", nodeId, params);
-
-  const result = { [nodeId]: "output 1" };
-  console.log("completing", nodeId, result);
-  return result;
-};
-
-const testAgent2: AgentFunction = async (context) => {
-  const { nodeId, retry, params } = context;
-  console.log("executing", nodeId, params);
-
-  const result = { [nodeId]: "output 2" };
-  console.log("completing", nodeId, result);
-  return result;
-};
-
-const numberTestAgent: AgentFunction<{ number: number }, { [key: string]: number }> = async (context) => {
-  const { nodeId, retry, params } = context;
-  console.log("executing", nodeId, params);
-
-  const result = { [nodeId]: params.number };
-  console.log("completing", nodeId, result);
-  return result;
-};
-
-test("test multiple function", async () => {
-  const result = await fileTestRunner("/graphs/test_multiple_functions_1.yml", { test1: testAgent1, test2: testAgent2, numberTestAgent });
-  assert.deepStrictEqual(result, {
-    node1: { node1: "output 1" },
-    node2: { node2: "output 1" },
-    node3: { node3: "output 2" },
-    node4: { node4: "output 1" },
-    node5: { node5: "output 2" },
-    node6: { node6: 10 },
-  });
-});

package/tests/graphai/test_sample_flow.ts

@@ -1,71 +0,0 @@
-import { GraphAI } from "@/graphai";
-import { testAgent } from "~/agents/agents";
-import { fileTestRunner } from "~/utils/runner";
-
-import test from "node:test";
-import assert from "node:assert";
-
-test("test base", async () => {
-  const result = await fileTestRunner("/graphs/test_base.yml", testAgent);
-  assert.deepStrictEqual(result, {
-    node1: { node1: "output" },
-    node2: { node2: "output" },
-    node3: { node3: "output", node1: "output", node2: "output" },
-    node4: { node4: "output", node3: "output", node1: "output", node2: "output" },
-    node5: { node5: "output", node4: "output", node3: "output", node1: "output", node2: "output" },
-  });
-});
-
-test("test retry", async () => {
-  const result = await fileTestRunner("/graphs/test_retry.yml", testAgent);
-  assert.deepStrictEqual(result, {
-    node1: { node1: "output" },
-    node2: { node2: "output" },
-    node3: { node3: "output", node1: "output", node2: "output" },
-    node4: { node4: "output", node3: "output", node1: "output", node2: "output" },
-    node5: { node5: "output", node4: "output", node3: "output", node1: "output", node2: "output" },
-  });
-});
-
-test("test error", async () => {
-  const result = await fileTestRunner("/graphs/test_error.yml", testAgent);
-  assert.deepStrictEqual(result, {
-    node1: { node1: "output" },
-    node2: { node2: "output" },
-  });
-});
-
-test("test timeout", async () => {
-  const result = await fileTestRunner("/graphs/test_timeout.yml", testAgent);
-  assert.deepStrictEqual(result, {
-    node1: { node1: "output" },
-    node2: { node2: "output" },
-    node3: { node3: "output", node1: "output", node2: "output" },
-  });
-});
-
-test("test source", async () => {
-  const result = await fileTestRunner("/graphs/test_source.yml", testAgent, (graph: GraphAI) => {
-    graph.injectResult("node2", { node2: "injected" });
-  });
-  assert.deepStrictEqual(result, {
-    node1: { node1: "output" },
-    node2: { node2: "injected" },
-    node3: { node3: "output", node1: "output", node2: "injected" },
-    node4: { node4: "output", node3: "output", node1: "output", node2: "injected" },
-    node5: { node5: "output", node4: "output", node3: "output", node1: "output", node2: "injected" },
-  });
-});
-
-test("test source2", async () => {
-  const result = await fileTestRunner("/graphs/test_source2.yml", testAgent, (graph: GraphAI) => {
-    graph.injectResult("node1", { node1: "injected" });
-  });
-  assert.deepStrictEqual(result, {
-    node1: { node1: "injected" },
-    node2: { node2: "preset" },
-    node3: { node3: "output", node1: "injected", node2: "preset" },
-    node4: { node4: "output", node3: "output", node1: "injected", node2: "preset" },
-    node5: { node5: "output", node4: "output", node3: "output", node1: "injected", node2: "preset" },
-  });
-});

package/tests/graphs/test_base.yml

@@ -1,19 +0,0 @@
-nodes:
-  node1:
-    params:
-      delay: 500
-  node2:
-    params:
-      delay: 100
-  node3:
-    params:
-      delay: 500
-    inputs: [node1, node2]
-  node4:
-    params:
-      delay: 100
-    inputs: [node3]
-  node5:
-    params:
-      delay: 500
-    inputs: [node2, node4]

package/tests/graphs/test_cli.yaml

@@ -1,4 +0,0 @@
-nodes:
-  node1:
-    agentId: test
-    

package/tests/graphs/test_dispatch.yml

@@ -1,29 +0,0 @@
-nodes:
-  node1:
-    params:
-      delay: 500
-    agentId: test
-  node2:
-    params:
-      delay: 100
-    agentId: alt
-    outputs:
-      output1: node20
-  node20:
-    source: true
-    agentId: test
-  node3:
-    params:
-      delay: 500
-    inputs: [node1, node20]
-    agentId: test
-  node4:
-    params:
-      delay: 100
-    inputs: [node3]
-    agentId: test
-  node5:
-    params:
-      delay: 500
-    inputs: [node20, node4]
-    agentId: test

package/tests/graphs/test_error.yml

@@ -1,22 +0,0 @@
-nodes:
-  node1:
-    params:
-      delay: 500
-  node2:
-    params:
-      delay: 100
-  node3:
-    params:
-      delay: 500
-      fail: true
-    inputs: [node1, node2]
-  node4:
-    timeout: 200
-    retry: 2
-    params:
-      delay: 300
-    inputs: [node3]
-  node5:
-    params:
-      delay: 100
-    inputs: [node4]

package/tests/graphs/test_multiple_functions_1.yml

@@ -1,30 +0,0 @@
-nodes:
-  node1:
-    params:
-      delay: 500
-    agentId: test1
-  node2:
-    params:
-      delay: 100
-    agentId: test1
-  node3:
-    params:
-      delay: 500
-    inputs: [node1, node2]
-    agentId: test2
-  node4:
-    params:
-      delay: 100
-    inputs: [node3]
-    agentId: test1
-  node5:
-    params:
-      delay: 500
-    inputs: [node2, node4]
-    agentId: test2
-  node6:
-    params:
-      delay: 100
-      number: 10
-    inputs: [node4]
-    agentId: numberTestAgent

package/tests/graphs/test_retry.yml

@@ -1,23 +0,0 @@
-nodes:
-  node1:
-    params:
-      delay: 500
-  node2:
-    params:
-      delay: 100
-  node3:
-    params:
-      delay: 500
-      fail: true
-    inputs: [node1, node2]
-    retry: 2
-  node4:
-    timeout: 200
-    retry: 2
-    params:
-      delay: 300
-    inputs: [node3]
-  node5:
-    params:
-      delay: 100
-    inputs: [node4]

package/tests/graphs/test_source.yml

@@ -1,18 +0,0 @@
-nodes:
-  node1:
-    params:
-      delay: 500
-  node2:
-    source: true
-  node3:
-    params:
-      delay: 500
-    inputs: [node1, node2]
-  node4:
-    params:
-      delay: 100
-    inputs: [node3]
-  node5:
-    params:
-      delay: 500
-    inputs: [node2, node4]

package/tests/graphs/test_source2.yml

@@ -1,19 +0,0 @@
-nodes:
-  node1:
-    source: true
-  node2:
-    source: true
-    result:
-      node2: preset
-  node3:
-    params:
-      delay: 500
-    inputs: [node1, node2]
-  node4:
-    params:
-      delay: 100
-    inputs: [node3]
-  node5:
-    params:
-      delay: 500
-    inputs: [node2, node4]

package/tests/graphs/test_timeout.yml

@@ -1,22 +0,0 @@
-nodes:
-  node1:
-    params:
-      delay: 500
-  node2:
-    params:
-      delay: 100
-  node3:
-    params:
-      delay: 500
-      fail: true
-    inputs: [node1, node2]
-    retry: 2
-  node4:
-    timeout: 200
-    params:
-      delay: 300
-    inputs: [node3]
-  node5:
-    params:
-      delay: 100
-    inputs: [node4]

package/tests/http-server/README.md

@@ -1,10 +0,0 @@
-This is a simple http server for testing http client.
-
-```
-cd docs
-```
-and
-
-```
-npx http-server
-```

package/tests/http-server/docs/llm.json

@@ -1,4 +0,0 @@
-{
-  "result": true,
-  "messages": ["hello"]
-}

package/tests/http-server/docs/llm2.json

@@ -1,4 +0,0 @@
-{
-  "result": true,
-  "messages": ["hello2"]
-}

package/tests/utils/file_utils.ts

@@ -1,33 +0,0 @@
-import fs from "fs";
-import path from "path";
-import YAML from "yaml";
-
-export const mkdirLogDir = () => {
-  const logsDir = path.join(__dirname, "../logs");
-
-  if (!fs.existsSync(logsDir)) {
-    fs.mkdirSync(logsDir, { recursive: true });
-  }
-};
-
-export const readGraphaiData = (file: string) => {
-  if (file.endsWith(".yaml") || file.endsWith(".yml")) {
-    return readYamlManifest(file);
-  }
-  if (file.endsWith(".json")) {
-    return readJsonManifest(file);
-  }
-  throw new Error("No file exists " + file);
-};
-
-export const readJsonManifest = (fileName: string) => {
-  const manifest_file = fs.readFileSync(fileName, "utf8");
-  const manifest = JSON.parse(manifest_file);
-  return manifest;
-};
-
-export const readYamlManifest = (fileName: string) => {
-  const manifest_file = fs.readFileSync(fileName, "utf8");
-  const manifest = YAML.parse(manifest_file);
-  return manifest;
-};

package/tests/utils/runner.ts

@@ -1,40 +0,0 @@
-import { GraphAI, GraphData, AgentFunctionDictonary, AgentFunction } from "@/graphai";
-import path from "path";
-import * as fs from "fs";
-import { readGraphaiData, mkdirLogDir } from "~/utils/file_utils";
-
-export const fileTestRunner = async (
-  file: string,
-  callbackDictonary: AgentFunctionDictonary | AgentFunction<any, any, any>,
-  callback: (graph: GraphAI) => void = () => {},
-) => {
-  const file_path = path.resolve(__dirname) + "/.." + file;
-  const graph_data = readGraphaiData(file_path);
-  return await graphDataTestRunner(file, graph_data, callbackDictonary, callback);
-};
-
-export const graphDataTestRunner = async (
-  logFileName: string,
-  graph_data: GraphData,
-  callbackDictonary: AgentFunctionDictonary | AgentFunction<any, any, any>,
-  callback: (graph: GraphAI) => void = () => {},
-) => {
-  mkdirLogDir();
-
-  const log_path = path.resolve(__dirname) + "/../logs/" + path.basename(logFileName).replace(/\.yml$/, ".log");
-  const graph = new GraphAI(graph_data, callbackDictonary);
-  callback(graph);
-
-  try {
-    const results = await graph.run();
-    fs.writeFileSync(log_path, JSON.stringify(graph.transactionLogs(), null, 2));
-    // console.log(graph.transactionLogs());
-    return results;
-  } catch (error) {
-    if (error instanceof Error) {
-      console.log("Error:", error.message);
-    }
-    // console.log(graph.transactionLogs());
-    return graph.results();
-  }
-};

package/tests/utils/utils.ts

@@ -1,3 +0,0 @@
-export const sleep = async (milliseconds: number) => {
-  return await new Promise((resolve) => setTimeout(resolve, milliseconds));
-};