rerank.go 481 B

12345678910111213141516171819
  1. package dto
  2. type RerankRequest struct {
  3. Documents []any `json:"documents"`
  4. Query string `json:"query"`
  5. Model string `json:"model"`
  6. TopN int `json:"top_n"`
  7. }
  8. type RerankResponseDocument struct {
  9. Document any `json:"document"`
  10. Index int `json:"index"`
  11. RelevanceScore float64 `json:"relevance_score"`
  12. }
  13. type RerankResponse struct {
  14. Results []RerankResponseDocument `json:"results"`
  15. Usage Usage `json:"usage"`
  16. }