package main

import (
    "fmt"
    "log"

    tavilygo "github.com/diverged/tavily-go"
    "github.com/diverged/tavily-go/models"
)

func main() {
    // Create a new Tavily client
    client := tavilygo.NewClient("your-api-key-here")

    // Perform a search
    searchReq := models.SearchRequest{
        Query:       "What is the capital of France?",
        SearchDepth: "basic",
    }
    
    response, err := tavilygo.Search(client, searchReq)
    if err != nil {
        log.Fatalf("Error performing search: %v", err)
    }

    // Print the search results
    for _, result := range response.Results {
        fmt.Printf("Title: %s\nURL: %s\n\n", result.Title, result.URL)
    }

    // Perform a Q&A search
    answer, err := tavilygo.QnASearch(client, "What is the capital of France?", "basic")
    if err != nil {
        log.Fatalf("Error performing Q&A search: %v", err)
    }

    fmt.Printf("Answer: %s\n", answer)
}