Rerank - Go SDK

1
package main
2
3
import(
4
	"context"
5
	"os"
6
	openrouter "github.com/OpenRouterTeam/go-sdk"
7
	"github.com/OpenRouterTeam/go-sdk/models/operations"
8
	"log"
9
)
10
11
func main() {
12
    ctx := context.Background()
13
14
    s := openrouter.New(
15
        openrouter.WithSecurity(os.Getenv("OPENROUTER_API_KEY")),
16
    )
17
18
    res, err := s.Rerank.Rerank(ctx, operations.CreateRerankRequest{
19
        Model: "cohere/rerank-v3.5",
20
        Query: "What is the capital of France?",
21
        Documents: []string{
22
            "Paris is the capital of France.",
23
            "Berlin is the capital of Germany.",
24
            "Madrid is the capital of Spain.",
25
        },
26
    })
27
    if err != nil {
28
        log.Fatal(err)
29
    }
30
    if res != nil {
31
        // handle response
32
    }
33
}