LangGraph Multi-Agent System

# Clone repository
npm install -g degit
degit sendaifun/solana-agent-kit/tree/docs/examples/agent-kit-langgraph myproject
cd myproject

# Install dependencies
pnpm install

# Configure environment
cp .env.example .env

OPENAI_API_KEY=your_openai_key
TAVILY_API_KEY=your_tavily_key
RPC_URL=your_solana_rpc
SOLANA_PRIVATE_KEY=your_wallet_key

src/
├── agents/
│   ├── manager.ts
│   ├── general.ts
│   ├── transfer.ts
│   └── read.ts
├── prompts/
│   └── templates/
├── tools/
└── utils/

// agents/manager.ts
import { StateGraph } from "@langchain/langgraph";

const graph = new StateGraph({
  channels: {
    query: ChannelType.GENERAL,
    blockchain: ChannelType.BLOCKCHAIN,
    transaction: ChannelType.TRANSACTION
  }
});

// Add nodes
graph.addNode("manager", managerAgent);
graph.addNode("general", generalAgent);
graph.addNode("transfer", transferAgent);
graph.addNode("read", readAgent);

// Define edges
graph.addEdge("manager", "general");
graph.addEdge("manager", "transfer");
graph.addEdge("manager", "read");

// agents/general.ts
class GeneralistAgent extends BaseAgent {
  async process(input: AgentInput): Promise<AgentOutput> {
    // Implementation using Tavily search
    if (this.config.enableSearch) {
      const searchResults = await this.tavily.search(input.query);
      // Process search results
    }
    
    return {
      response: await this.generateResponse(input, context)
    };
  }
}

const workflow = new SolanaAgentWorkflow();

// General query
const response = await workflow.run({
  query: "What is the current SOL price?",
  type: QueryType.GENERAL
});

// Transaction query
const transferResponse = await workflow.run({
  query: "Send 1 SOL to address...",
  type: QueryType.TRANSACTION
});

// Add custom agent
const customAgent = new CustomAgent({
  name: "custom",
  description: "Handles custom operations",
  tools: [/* custom tools */]
});

workflow.addAgent(customAgent, {
  triggers: ["custom", "specific"]
});

interface WorkflowState {
  query: string;
  context: any;
  history: Message[];
  activeAgent: string;
  status: WorkflowStatus;
}

// State updates
graph.onStateChange((state: WorkflowState) => {
  // Handle state changes
});

enum ChannelType {
  GENERAL = "general",
  BLOCKCHAIN = "blockchain",
  TRANSACTION = "transaction",
  CUSTOM = "custom"
}

describe('Manager Agent', () => {
  it('routes general queries correctly', async () => {
    const result = await managerAgent.route({
      query: "What is Solana?",
      type: QueryType.GENERAL
    });
    expect(result.agent).toBe("general");
  });
});

describe('Workflow', () => {
  it('completes full query cycle', async () => {
    const workflow = new SolanaAgentWorkflow();
    const response = await workflow.run({
      query: "Send 1 SOL",
      type: QueryType.TRANSACTION
    });
    expect(response.status).toBe("completed");
  });
});