solagent 0.1.12

connect any ai agents to solana protocols in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

// Copyright 2025 zTgx
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use crate::{
    tools::{
        deploy_collection, deploy_token, fetch_price, get_balance_other, get_wallet_address, gibwork,
        helius::{create_webhook, delete_webhook, get_webhook, transaction_parsing},
        jupiter::get_token_data_by_address,
        launch_token_pumpfun, mint_nft, pyth_fetch_price, rugcheck,
        solana::{close_empty_token_accounts, get_balance, get_tps, request_faucet_funds, transfer},
        solayer::stake_with_solayer,
        stake_with_jup, trade,
    },
    SolanaAgentKit,
};
use rig::tool::ToolSet;
use std::sync::Arc;

/// An function to build a set of tools that can be used with Solana.
///
/// - `DeployCollection`: Tool to deploy a new NFT collection on Solana blockchain.
/// - `DeployToken`: Tool to deploy a new token on Solana blockchain.
/// - `FetchPrice`: Tool to fetch the price of a given token in USDC.
/// - `GetBalance`: Tool to get the balance of a specified wallet.
/// - `GetBalanceOther`: Tool to get the balance of another wallet.
/// - `GetTps`: Tool to get the current TPS of the Solana network.
/// - `GetWalletAddress`: Tool to the wallet address of the agent.
/// - `LaunchPumpfunToken`: Tool to launch a token on Pump.fun.
/// - `MintNFT`: Tool to mint a new NFT in a collection on Solana blockchain.
/// - `FetchPricePyTh`: Tool to fetch the price of a given price feed from Pyth's Hermes service.
/// - `RequestFaucetFunds`: Tool to request funds from a faucet.
/// - `StakeWithJup`: Tool to stake your SOL (Solana), also called as SOL staking or liquid staking.
/// - `Trade`: Tool to swap tokens to another token ( It uses Jupiter Exchange ).
/// - `Transfer`: Tool to transfer tokens or SOL to another address ( also called as wallet address ).
/// - `FetchTokenReportSummary`: Tool to fetch a summary report for a specific token from RugCheck.
/// - `FetchTokenReportDetailed`: Tool to fetch a detailed report for a specific token from RugCheck.
/// - `CreateGibworkTask`: Tool to create a task on Gibwork.
/// - `CloseEmptyTokenAccounts`: Tool to close empty SPL Token accounts associated with your wallet to reclaim rent.
/// - `StakeWithSolayer`: Tool to stake native SOL with Solayer's restaking protocol to receive Solayer SOL (sSOL).
/// - `GetTokenData`: Tool to get the token data for a given token mint address.
/// - `CreateWebHook`: Tool to creates a new webhook in the Helius system to monitor transactions for specified account addresses.
/// - `DeleteWebhook`: Tool to deletes a Helius webhook by its unique ID.
/// - `GetWebHook`: Tool to retrieves details of a Helius webhook by its unique ID.
/// - `TransactionParse`: Tool to parse a Solana transaction to retrieve detailed information using the Helius Enhanced Transactions API.
pub fn create_solana_tools(agent: SolanaAgentKit) -> ToolSet {
    let agent = Arc::new(agent);
    let builder = ToolSet::builder()
        .dynamic_tool(deploy_collection::DeployCollection::new(agent.clone()))
        .dynamic_tool(deploy_token::DeployToken::new(agent.clone()))
        .dynamic_tool(fetch_price::FetchPrice::new())
        .dynamic_tool(get_balance::GetBalance::new(agent.clone()))
        .dynamic_tool(get_balance_other::GetBalanceOther::new(agent.clone()))
        .dynamic_tool(get_tps::GetTps::new(agent.clone()))
        .dynamic_tool(get_wallet_address::GetWalletAddress::new(agent.clone()))
        .dynamic_tool(launch_token_pumpfun::LaunchPumpfunToken::new(agent.clone()))
        .dynamic_tool(mint_nft::MintNFT::new(agent.clone()))
        .dynamic_tool(pyth_fetch_price::FetchPricePyTh::new())
        .dynamic_tool(request_faucet_funds::RequestFaucetFunds::new(agent.clone()))
        .dynamic_tool(stake_with_jup::StakeWithJup::new(agent.clone()))
        .dynamic_tool(trade::Trade::new(agent.clone()))
        .dynamic_tool(transfer::Transfer::new(agent.clone()))
        .dynamic_tool(rugcheck::token_report_summary::FetchTokenReportSummary::new())
        .dynamic_tool(rugcheck::token_report_detailed::FetchTokenReportDetailed::new())
        .dynamic_tool(gibwork::create_gibwork_task::CreateGibworkTask::new(agent.clone()))
        .dynamic_tool(close_empty_token_accounts::CloseEmptyTokenAccounts::new(agent.clone()))
        .dynamic_tool(stake_with_solayer::StakeWithSolayer::new(agent.clone()))
        .dynamic_tool(get_token_data_by_address::GetTokenData::new())
        .dynamic_tool(create_webhook::CreateWebHook::new(agent.clone()))
        .dynamic_tool(delete_webhook::DeleteWebHook::new(agent.clone()))
        .dynamic_tool(get_webhook::GetWebHook::new(agent.clone()))
        .dynamic_tool(transaction_parsing::TransactionParse::new(agent.clone()));

    builder.build()
}