bailian

type RerankInput struct {
	Query     string   `json:"query"`
	Documents []string `json:"documents"`
}

type RerankParameter struct {
	ReturnDocuments bool   `json:"return_documents"`
	TopN            int    `json:"top_n,omitempty"`
	Instruct        string `json:"instruct,omitempty"`
}

type RerankRequest struct {
	Model      string           `json:"model"`
	Input      RerankInput      `json:"input"`
	Parameters *RerankParameter `json:"parameters,omitempty"`
}

type Reranker struct {
	Ctx    context.Context
	Config RerankerConfig
}

type RerankerConfig struct {
	APIKey  string
	Model   string
	BaseUrl string
}