Rerank

package main

import(
	"context"
	"os"
	openrouter "github.com/OpenRouterTeam/go-sdk"
	"github.com/OpenRouterTeam/go-sdk/models/operations"
	"log"
)

func main() {
    ctx := context.Background()

    s := openrouter.New(
        openrouter.WithSecurity(os.Getenv("OPENROUTER_API_KEY")),
    )

    res, err := s.Rerank.Rerank(ctx, operations.CreateRerankRequest{
        Documents: []operations.Document{
            operations.CreateDocumentStr(
                "Paris is the capital of France.",
            ),
            operations.CreateDocumentStr(
                "Berlin is the capital of Germany.",
            ),
        },
        Model: "cohere/rerank-v3.5",
        Query: "What is the capital of France?",
    })
    if err != nil {
        log.Fatal(err)
    }
    if res != nil {
        // handle response
    }
}