SerpAPI Tool

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

export SERPAPI_API_KEY="your-serpapi-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/serp_api"
)

func main() {
    // Create SerpAPI tools
    serpTools := serp_api.NewSerpAPITools()
    
    // Create model
    model := model.Model("gpt-4o").
        SetAPIKey(os.Getenv("OPENAI_API_KEY"))
    
    // Create agent with SerpAPI tools
    agent := agent.New().
        SetModel(model).
        SetSystemPrompt("You are a research assistant with multi-engine search capabilities.").
        AddTool(serpTools.Search()).
        AddTool(serpTools.ImageSearch())
    
    // Create session and run
    session := session.New(agent)
    ctx := context.Background()
    
    response, err := session.Run(ctx, []agent.ChatMessage{
        agent.NewUserMessage("Search for 'golang best practices' on Google and Bing"),
    }, nil)
    
    if err != nil {
        panic(err)
    }
    
    println(response.GetLastMessage().GetContent())
}