diff --git a/dto/rerank.go b/dto/rerank.go index 21f6437c..5ea68cba 100644 --- a/dto/rerank.go +++ b/dto/rerank.go @@ -4,7 +4,7 @@ type RerankRequest struct { Documents []any `json:"documents"` Query string `json:"query"` Model string `json:"model"` - TopN int `json:"top_n"` + TopN int `json:"top_n,omitempty"` ReturnDocuments *bool `json:"return_documents,omitempty"` MaxChunkPerDoc int `json:"max_chunk_per_doc,omitempty"` OverLapTokens int `json:"overlap_tokens,omitempty"` diff --git a/relay/channel/jina/constant.go b/relay/channel/jina/constant.go index 45fc44c9..be290fb6 100644 --- a/relay/channel/jina/constant.go +++ b/relay/channel/jina/constant.go @@ -3,6 +3,7 @@ package jina var ModelList = []string{ "jina-clip-v1", "jina-reranker-v2-base-multilingual", + "jina-reranker-m0", } var ChannelName = "jina" diff --git a/relay/rerank_handler.go b/relay/rerank_handler.go index 319811b8..5cf384a8 100644 --- a/relay/rerank_handler.go +++ b/relay/rerank_handler.go @@ -78,12 +78,15 @@ func RerankHelper(c *gin.Context, relayMode int) (openaiErr *dto.OpenAIErrorWith return service.OpenAIErrorWrapperLocal(err, "json_marshal_failed", http.StatusInternalServerError) } requestBody := bytes.NewBuffer(jsonData) - statusCodeMappingStr := c.GetString("status_code_mapping") + if common.DebugEnabled { + println(fmt.Sprintf("Rerank request body: %s", requestBody.String())) + } resp, err := adaptor.DoRequest(c, relayInfo, requestBody) if err != nil { return service.OpenAIErrorWrapper(err, "do_request_failed", http.StatusInternalServerError) } + statusCodeMappingStr := c.GetString("status_code_mapping") var httpResp *http.Response if resp != nil { httpResp = resp.(*http.Response)