added write functionalities ie, swap and transfer

This commit is contained in:
Deepak
2024-12-27 16:53:12 +05:30
parent ab9259d4e2
commit 3cdd45d623
13 changed files with 572 additions and 17 deletions

View File

@@ -0,0 +1,14 @@
import { solanaAgentState } from "./state";
import { END } from "@langchain/langgraph";
export const managerRouter = (state: typeof solanaAgentState.State) => {
const { isSolanaReadQuery, isSolanaWriteQuery, isGeneralQuery } = state;
if (isGeneralQuery) {
return "generalist";
} else if (isSolanaWriteQuery) {
return "transferSwap";
} else {
return END;
}
};

View File

@@ -1,7 +1,9 @@
import { SolanaAgentKit } from "solana-agent-kit";
import { SolanaAgentKit, createSolanaTools } from "solana-agent-kit";
export const agentKit = new SolanaAgentKit(
process.env.SOLANA_PRIVATE_KEY!,
process.env.RPC_URL!,
process.env.OPENAI_API_KEY!,
);
export const solanaTools = createSolanaTools(agentKit);