graphai 0.0.10 → 0.0.11

package/samples/home.json

@@ -1,112 +0,0 @@
-[
-  {
-      "name": "fill_bath",
-      "description": "Fill the bath tub",
-      "parameters": {
-          "type": "object",
-          "properties": {
-              "temperature": {
-                  "type": "number",
-                  "description": "Water temperature in celsius. If omitted, 41 degree"
-              },
-              "at": {
-                  "type": "string",
-                  "description": "Time to fill. If omitted immediately."
-              }
-          },
-          "required": []
-      }
-  },
-  {
-      "name": "set_temperature",
-      "description": "Set the temperature",
-      "parameters": {
-          "type": "object",
-          "properties": {
-              "temperature": {
-                  "type": "number",
-                  "description": "Room temperature in celsius."
-              },
-              "location": {
-                  "type": "string",
-                  "enum": ["kitchen", "living room", "master bedroom", "bedroom 2", "bedroom 3", "all bedrooms"],
-                  "description": "The room to set the temperature."
-              }
-          },
-          "required": ["temperature", "location"]
-      }
-  },
-  {
-      "name": "start_sprinkler",
-      "description": "Start the sprinkler system",
-      "parameters": {
-          "type": "object",
-          "properties": {
-              "location": {
-                  "type": "string",
-                  "enum": ["all", "lawn", "vegetable garden", "front yard"],
-                  "description": "Specify the location to water."
-              }
-          },
-          "required": ["location"]
-      }
-  },
-  {
-      "name": "take_picture",
-      "description": "Take a picture and send it",
-      "parameters": {
-          "type": "object",
-          "properties": {
-              "location": {
-                  "type": "string",
-                  "enum": ["entrance", "backyard", "kitchen", "living room"],
-                  "description": "Specify the camera location to use."
-              }
-          },
-          "required": ["location"]
-      }
-  },
-  {
-      "name": "play_music",
-      "description": "Play music",
-      "parameters": {
-          "type": "object",
-          "properties": {
-              "location": {
-                  "type": "string",
-                  "enum": ["living room", "master bedroom"],
-                  "description": "Location to play a music."
-              },
-              "music": {
-                  "type": "string",
-                  "description": "Music to play, such as artist, title and playlist."
-              }
-          },
-          "required": ["location"]
-      }
-  },
-  {
-      "name": "control_light",
-      "description": "Turn on/off various lights",
-      "parameters": {
-          "type": "object",
-          "properties": {
-              "location": {
-                  "type": "string",
-                  "enum": ["kitchen", "living room", "master bedroom", "bedroom 2", "bedroom 3", "front gate"],
-                  "description": "Specify the location to control the light."
-              },
-              "switch": {
-                  "type": "string",
-                  "enum": ["on", "off"],
-                  "description": "Specify the light switch."
-              },
-              "dim": {
-                  "type": "number",
-                  "description": "Specify the dim level between 0 and 1.0. The 'switch' parameter must be 'on'."
-              }
-          },
-          "required": ["location", "switch"]
-      }
-  }
-]

package/samples/home.ts

@@ -1,51 +0,0 @@
-import { GraphAI, GraphData } from "@/graphai";
-import path from "path";
-import * as fs from "fs";
-import { slashGPTAgent } from "@/experimental_agents";
-
-const home_actions = {
-  fill_bath: { type: "message_template", message: "Success. I started filling the bath tab." },
-  set_temperature: { type: "message_template", message: "Success. I set the temperature to {temperature} for {location}" },
-  start_sprinkler: { type: "message_template", message: "Success. I started the sprinkler for {location}" },
-  take_picture: { type: "message_template", message: "Success. I took a picture of {location}" },
-  play_music: { type: "message_template", message: "Success. I started playing {music} in {location}" },
-  control_light: { type: "message_template", message: "Success. The light switch of {location} is now {switch}." },
-};
-
-const fileName = path.resolve(__dirname) + "/home.json";
-const json_file = fs.readFileSync(fileName, "utf8");
-const home_functions = JSON.parse(json_file);
-
-const graph_data: GraphData = {
-  nodes: {
-    node1: {
-      source: true,
-      result: { content: "Turn on the light in the kitchen" },
-    },
-    node2: {
-      inputs: ["node1"],
-      params: {
-        manifest: {
-          skip_function_result: true,
-          actions: home_actions,
-          functions: home_functions,
-        },
-      },
-    },
-  },
-};
-
-const runAgent = async () => {
-  const graph = new GraphAI(graph_data, slashGPTAgent);
-  const result = await graph.run();
-  const log_path = path.resolve(__dirname) + "/../tests/logs/home.log";
-  fs.writeFileSync(log_path, JSON.stringify(graph.transactionLogs(), null, 2));
-  console.log(result["node2"]!.content);
-};
-
-const main = async () => {
-  await runAgent();
-  console.log("COMPLETE 1");
-};
-
-main();

package/samples/interaction.ts

@@ -1,42 +0,0 @@
-import { GraphAI, GraphData } from "@/graphai";
-import * as readline from "readline";
-import { testAgent } from "~/agents/agents";
-import { graphDataTestRunner } from "~/utils/runner";
-
-const getUserInput = async (question: string): Promise<string> => {
-  return new Promise((resolve, reject) => {
-    const rl = readline.createInterface({
-      input: process.stdin,
-      output: process.stdout,
-    });
-
-    rl.question(question, (answer) => {
-      rl.close();
-      resolve(answer);
-    });
-  });
-};
-
-const graph_data: GraphData = {
-  nodes: {
-    node1: {
-      source: true,
-    },
-    node2: {
-      inputs: ["node1"],
-    },
-  },
-};
-
-const main = async () => {
-  const query = await getUserInput("Please enter your question: ");
-  console.log("query=", query);
-  graph_data.nodes.node1.result = { query };
-
-  const result = await graphDataTestRunner("interaction.yaml", graph_data, testAgent);
-  console.log(result);
-
-  console.log("COMPLETE 1");
-};
-
-main();

package/samples/runner.ts

@@ -1,15 +0,0 @@
-import path from "path";
-
-import { GraphAI, AgentFunctionDictonary, AgentFunction } from "@/graphai";
-import { readGraphaiData } from "~/utils/file_utils";
-import { graphDataTestRunner } from "~/utils/runner";
-
-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);
-};

package/samples/sample_co2.ts

@@ -1,80 +0,0 @@
-import { slashGPTFuncitons2TextAgent } from "./agents/slashgpt_agent";
-import { slashGPTAgent } from "@/experimental_agents";
-
-import { graphDataTestRunner } from "~/utils/runner";
-
-const graph_data = {
-  nodes: {
-    slashGPTAgent: {
-      agentId: "slashGPTAgent",
-      params: {
-        function_result: true,
-        query: "世界で協力してco2を減らす方法を教えて下さい",
-        manifest: {
-          prompt: "あなたは世界経済の専門家です。ユーザの問い合わせについて考え、10この結果をfunctionsの結果に返してください。",
-          skip_function_result: true,
-          actions: {
-            your_ideas: {
-              type: "message_template",
-              message: "dummy",
-            },
-          },
-          functions: [
-            {
-              name: "your_ideas",
-              description: "Your answer to a user's inquiry",
-              parameters: {
-                type: "object",
-                properties: {
-                  methods: {
-                    type: "array",
-                    items: {
-                      type: "object",
-                      properties: {
-                        title: {
-                          type: "string",
-                          description: "title",
-                        },
-                        description: {
-                          type: "string",
-                          description: "description",
-                        },
-                      },
-                      required: ["title", "description"],
-                    },
-                  },
-                },
-              },
-            },
-          ],
-        },
-      },
-    },
-    function2prompt0: {
-      fork: 10,
-      params: {
-        function_data_key: "methods",
-        result_key: 0,
-      },
-      inputs: ["slashGPTAgent"],
-      agentId: "slashGPTFuncitons2TextAgent",
-    },
-    slashGPTAgent0: {
-      agentId: "slashGPTAgent",
-      fork: 10,
-      params: {
-        debug: true,
-        manifest: {
-          prompt: "ユーザの問い合わせにある文章の専門家です。専門家として、ユーザのアイデアに対して実現可能なシナリオを800文字で書いてください。",
-        },
-      },
-      inputs: ["function2prompt0"],
-    },
-  },
-};
-
-const main = async () => {
-  const result = await graphDataTestRunner("sample_co2.yaml", graph_data, { slashGPTAgent, slashGPTFuncitons2TextAgent });
-  console.log("COMPLETE 1");
-};
-main();

package/samples/sample_gpt.ts

@@ -1,8 +0,0 @@
-import { fileTestRunner } from "./runner";
-
-import { stringTemplateAgent, slashGPTAgent } from "@/experimental_agents";
-
-const main = async () => {
-  await fileTestRunner("/graphs/slash_gpt.yml", { slashgpt: slashGPTAgent, stringTemplate: stringTemplateAgent });
-};
-main();

package/samples/sample_paper_ai.ts

@@ -1,10 +0,0 @@
-import { fileTestRunner } from "./runner";
-
-import { slashGPTAgent } from "@/experimental_agents";
-import { arxivAgent, arxiv2TextAgent } from "./agents/arxiv_agent";
-
-const main = async () => {
-  await fileTestRunner("/graphs/arxiv.yml", { arxivAgent: arxivAgent, arxiv2TextAgent, slashGPTAgent });
-  console.log("COMPLETE 1");
-};
-main();

package/src/experimental_agents/index.ts

@@ -1,2 +0,0 @@
-export * from "./string_agent";
-export * from "./slashgpt_agent";

package/src/experimental_agents/slashgpt_agent.ts

@@ -1,30 +0,0 @@
-import path from "path";
-import { AgentFunction } from "@/graphai";
-import { ChatSession, ChatConfig, ManifestData } from "slashgpt";
-
-const config = new ChatConfig(path.resolve(__dirname));
-
-export const slashGPTAgent: AgentFunction<{ manifest: ManifestData; query?: string; function_result?: boolean }, { content: string }> = async (context) => {
-  console.log("executing", context.nodeId, context.params);
-  const session = new ChatSession(config, context.params.manifest ?? {});
-
-  const query = context.params?.query ? [context.params.query] : [];
-  const contents = query.concat(
-    context.inputs.map((input) => {
-      return input.content;
-    }),
-  );
-
-  session.append_user_question(contents.join("\n"));
-  await session.call_loop(() => {});
-  const message = (() => {
-    if (context.params?.function_result) {
-      return session.history.messages().find((m) => m.role === "function_result");
-    }
-    return session.history.last_message();
-  })();
-  if (message === undefined) {
-    throw new Error("No message in the history");
-  }
-  return message;
-};

package/src/experimental_agents/string_agent.ts

@@ -1,10 +0,0 @@
-import { AgentFunction } from "@/graphai";
-
-export const stringTemplateAgent: AgentFunction<{ template: string; inputKey?: string }, { content: string }> = async (context) => {
-  console.log("executing", context.nodeId, context.params);
-  const content = context.inputs.reduce((template, input, index) => {
-    return template.replace("${" + index + "}", input[context.params.inputKey ?? "content"]);
-  }, context.params.template);
-
-  return { content };
-};