Brave Search Tool

import "github.com/model-box/agent-kit/tool/brave_search"

export BRAVE_SEARCH_API_KEY="your-brave-search-api-key"

package main

import (
    "context"
    "os"
    
    "github.com/model-box/agent-kit/agent"
    "github.com/model-box/agent-kit/model"
    "github.com/model-box/agent-kit/session"
    "github.com/model-box/agent-kit/tool/brave_search"
)

func main() {
    // Create Brave Search tools
    braveTools := brave_search.NewBraveSearchTools()
    
    // Create model
    model := model.Model("gpt-4o").
        SetAPIKey(os.Getenv("OPENAI_API_KEY"))
    
    // Create agent with Brave Search tool
    agent := agent.New().
        SetModel(model).
        SetSystemPrompt("You are a helpful assistant with web search capabilities.").
        AddTool(braveTools.WebSearch())
    
    // Create session and run
    session := session.New(agent)
    ctx := context.Background()
    
    response, err := session.Run(ctx, []agent.ChatMessage{
        agent.NewUserMessage("Search for the latest developments in quantum computing"),
    }, nil)
    
    if err != nil {
        panic(err)
    }
    
    println(response.GetLastMessage().GetContent())
}