GraphAIをNode.jsで試す

GraphAI記事の一覧はこちら

GraphAIのNode.jsで試す

GraphAIを動作させるのに必要な最小限のnpmはgraphaiとvanilla agents です。
最も簡単に使えるAgentである@graphai/vanilla (他にnpmの依存がないagentをvanillaと呼んでいます)を使って簡単なGraphDataを作り、動かしてみます。

最初にnpmの初期化をして、必要なnpmを入れます。typescriptを実行するためにts-nodeも入れます。

npm init
yarn add graphai @graphai/vanilla ts-node

以下が最小限のスクリプトです。graphai.tsで保存します。

このGraphDataには２つのnodeが含まれます。
Static nodeのnode1とComputed nodeのnode2です。

Static nodeのnode1で、固定のメッセージを定義します。
そのデータを受け取ったComputed nodeのnode2でcopyAgentを実行します。
copyAgentはinputsで受け取ったデータをそのまま返却値とて返すagentです。
node2はisResult: trueがセットされているので、GraphAIの実行結果としてnode2の結果を返します。

TypeScriptのresult変数にその結果がセットされます。

import { GraphAI } from "graphai";
import * as agents from "@graphai/vanilla";

const graph_data = {
  version: 0.5,
  nodes: {
    node1: {
      value: "hello, GraphAI",
    },
    node2: {
      agent: "copyAgent",
      inputs: {text: ":node1"},
      isResult: true,
    },
  },
};

export const main = async () => {
  const graph = new GraphAI(graph_data, agents);
  const result = await graph.run();
  console.log(JSON.stringify(result));

};

if (process.argv[1] === __filename) {
  main();
}

実行します

$ npx ts-node graphai.ts
{"node2": {text: "hello, GraphAI"}}

node1で指定して文字列がnode2に渡され、結果として表示されました。