mirror of
https://github.com/d0zingcat/solana-agent-kit.git
synced 2026-05-22 15:10:43 +00:00
24 lines
666 B
TypeScript
24 lines
666 B
TypeScript
import { prompt, parser } from "../prompts/manager";
|
|
import { RunnableSequence } from "@langchain/core/runnables";
|
|
import { solanaAgentState } from "../utils/state";
|
|
import { gpt4o } from "../utils/model";
|
|
|
|
const chain = RunnableSequence.from([prompt, gpt4o, parser]);
|
|
|
|
export const managerNode = async (state: typeof solanaAgentState.State) => {
|
|
const { messages } = state;
|
|
|
|
const result = await chain.invoke({
|
|
formatInstructions: parser.getFormatInstructions(),
|
|
messages: messages,
|
|
});
|
|
|
|
const { isSolanaReadQuery, isSolanaWriteQuery, isGeneralQuery } = result;
|
|
|
|
return {
|
|
isSolanaReadQuery,
|
|
isSolanaWriteQuery,
|
|
isGeneralQuery,
|
|
};
|
|
};
|