mirror of
https://github.com/d0zingcat/solana-agent-kit.git
synced 2026-05-19 07:36:46 +00:00
29 lines
998 B
TypeScript
29 lines
998 B
TypeScript
import { StateGraph } from "@langchain/langgraph";
|
|
import { solanaAgentState } from "./utils/state";
|
|
import { generalistNode } from "./agents/generalAgent";
|
|
import { transferSwapNode } from "./agents/transferOrSwap";
|
|
import { managerNode } from "./agents/manager";
|
|
import { readNode } from "./agents/readAgent";
|
|
import { START, END } from "@langchain/langgraph";
|
|
import { managerRouter } from "./utils/route";
|
|
import { HumanMessage } from "@langchain/core/messages";
|
|
|
|
const workflow = new StateGraph(solanaAgentState)
|
|
.addNode("generalist", generalistNode)
|
|
.addNode("manager", managerNode)
|
|
.addNode("transferSwap", transferSwapNode)
|
|
.addNode("read", readNode)
|
|
.addEdge(START, "manager")
|
|
.addConditionalEdges("manager", managerRouter)
|
|
.addEdge("generalist", END)
|
|
.addEdge("transferSwap", END)
|
|
.addEdge("read", END);
|
|
|
|
export const graph = workflow.compile();
|
|
|
|
const result = await graph.invoke({
|
|
messages: [new HumanMessage("what is the price of SOL")],
|
|
});
|
|
|
|
console.log(result);
|