Google Search Tool

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

export GOOGLE_API_KEY="your-google-api-key"
export GOOGLE_SEARCH_ENGINE_ID="your-custom-search-engine-id"

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/google_search"
)

func main() {
    // Create Google Search tools
    googleTools := google_search.NewGoogleSearchTools()
    
    // Create model
    model := model.Model("gpt-4o").
        SetAPIKey(os.Getenv("OPENAI_API_KEY"))
    
    // Create agent with Google Search tool
    agent := agent.New().
        SetModel(model).
        SetSystemPrompt("You are a helpful assistant that can search the web.").
        AddTool(googleTools.Search())
    
    // Create session and run
    session := session.New(agent)
    ctx := context.Background()
    
    response, err := session.Run(ctx, []agent.ChatMessage{
        agent.NewUserMessage("Search for the latest Go programming tutorials"),
    }, nil)
    
    if err != nil {
        panic(err)
    }
    
    println(response.GetLastMessage().GetContent())
}