🤖

GraphAIのデータ入力について

Isamu

2024/09/11に公開

GraphAIでは、Agentを実行するときに前のノードの結果を入力値として受け取ることができます。入力値を受け取ることができるのはComputed Nodeです。一方で、Static Nodeは特別な条件のときに他のノードの結果を受け取ることができますが、通常はvalueを受け取れませんので、注意が必要です。

GraphAIのComputed Nodeにおける入力について

Computed Nodeはinputsという要素を使ってデータを受け取ることができます。inputsを指定する方法を説明します。

入力値の定義

inputsという名前の要素で定義する
inputsの値は、objectで記述できます。

nodes: {
  StaticNodeA: {
    value: "hello",
  },
  ComputedNodeB: {
    agent: "copyAgent",
    inputs: {text: ":StaticNodeA"}
  },
  ComputedNodeC: {
    agent: "copyAgent",
    inputs: {"A": ":StaticNodeA", "B": ":ComputedNodeB"}
  },
}

直接値を設定する場合
数値（例: 123）、文字列（例: "ABC"）、オブジェクト（例: {test: "hello"}）などを直接入力することができます。

nodes: {
  ...,
  ComputedNodeB: {
    agent: "copyAgent",
    inputs: {array: [123, "abc", {message: "message"}] }
  },
}

ノードを指定する場合
ノードを指定するには、":node1"のように、:（コロン）で始まる文字列を使用します。ノードを指定した場合、そのノードが結果を返すまで実行を待機します。言い換えると、inputsでノードを指定することで、実行順を決めることができます。
特定のプロパティを指定する場合
例えば、inputs: {text: ":nodeA.property1"}と記述すると、nodeAの結果のproperty1が渡されます。ここで期待するデータ形式は { property1: "value" } のようなオブジェクトです。

nodes: {
  nodeA: {
    value: {
      property1: "value"
    },
  },
  ComputedNodeB: {
    agent: "copyAgent",
    inputs: {text: ":nodeA.property1"}
  },
}

配列の要素を指定する場合
inputs: {text: ":nodeA.result.$0"} のように記述すると、nodeAの結果の中にある配列 result の最初の要素が渡されます。ここで $0 は配列の添字を示しています。例えば、{ results: ["A", "B"] } というデータ形式の場合、"A" が入力として渡されます。

nodes: {
  nodeA: {
    value: {
      result: ["A", "B"],
    },
  },
  ComputedNodeB: {
    agent: "copyAgent",
    inputs: {text: ":nodeA.result.$0"}
  },
}

OpenAIのレスポンス例
OpenAIのレスポンスを処理する場合、例えば inputs: {text: ":openAI.choices.$0.message.content"} のように記述します。これにより、OpenAIのレスポンスから choices 配列の最初の要素の message.content が入力として使用されます。

このように、inputs を使って様々な形式で入力値を定義し、ノードの結果や特定のデータを柔軟に扱うことができます。

GraphAIにおける`inputs`のデータ形式

inputsは、objectで記述できます。

データ形式の定義

nodeに指定するinputsのデータはobjectとして定義します。このデータはAgentにはnamedInputsという名前でobjectが渡されます。
例えば、inputs: {dataA: ":nodeA"}と定義すると、Agent内ではnamedInputsとして受け取ります。

ネストされたデータ

ネストした部分にノードを指定
inputs: {dataA: {DataB: ":nodeA"}}のようにネストされた部分にノードを指定すると、ネストした部分もデータとして展開されます。

テンプレート記法

ネストした部分にノードを指定
inputs: {dataA: "hello ${:nodeA.userName}!!"}のようにテンプレートを使った記述も可能です。

Static nodeのデータ受取

Static Nodeは、特定の条件下で他のノードの値を受け取ることができます。特に、loopやnested graphの場合に利用されます。

Loop時のStatic Node

Loopの際のStatic Node
Loop処理中にStatic Nodeの値を受け取ることができます。例えば、以下のように定義された場合：

  loop: {
    5,
  },
  nodes: {
    staticNode: {
      value: "init value",
      update: ":lastDataOfloop"
    },
    lastDataOfloop: {
      ...
    },
  }

staticNodeの値は、loopの1回目では "init value" となり、2回目以降は lastDataOfloop Nodeの結果で上書きされます。

Nested Graphの場合

暗黙のケース（Static Nodeを定義しない）

親Agentの入力値が子Graph内のStatic Nodeとして自動的に受け取られます。例：

parentNodeResult: {
  agent: "arrayFlatAgent",
  inputs: {
    array: [":NodeA", ":NodeB"],
  },
},
childGraph: {
  agent: "nestedAgent",
  inputs: { history: ":parentNodeResult.array" },
  isResult: true,
  graph: {
    version: 0.5,
    nodes: {
      nodeB: {
        agent: "copyAgent",
        inputs: {history: ":history"}
      }
    }
  }
}

この場合、childGraphのinputsで指定された :parentNodeResult.array が namedInputs で受け取られ、history ノードが暗黙的に定義されます。childGraph内で history ノードを入力値として使うことができます。

明示的にStatic Nodeを定義する場合
明示的にStatic Nodeを定義することで、update を使って値を上書きすることができます。例：

parentNodeResult: {
  agent: "arrayFlatAgent",
  inputs: {
    array: [":NodeA", ":NodeB"],
  },
},
childGraph: {
  agent: "nestedAgent",
  inputs: { history: ":parentNodeResult.array" },
  isResult: true,
  graph: {
    version: 0.5,
    loop: {
      5
    },
    nodes: {
      history: {
        value: "",
        update: ":prevResult",
      },
      nodeZ: {
        agent: "copyAgent",
        inputs: { history: ":history" }
      }
    }
  }
}

この場合、history ノードを明示的に定義し、update を使って値を設定することができます。update はloop時に使われるため、2回目以降のloopで指定された内容で上書きされます。

Map Agentの場合

MapAgentは、入力でinputs: { rows: ":arrayNodeData"}でrowsにarrayを渡します。
すると、arrayNodeData.lengthだけGraphAIインスタンスを作成し、ぞれぞれのデータをインスタンスに渡し並列に実行します。
インスタンスである子GraphAIには、それぞれのarrayの値をrowという名前の入力として渡します

暗黙のケース（Static Nodeを定義しない）