(rerankRequest dto.RerankRequest)
| 15 | ) |
| 16 | |
| 17 | func getRerankPromptToken(rerankRequest dto.RerankRequest) int { |
| 18 | token := service.CountTokenInput(rerankRequest.Query, rerankRequest.Model) |
| 19 | for _, document := range rerankRequest.Documents { |
| 20 | tkm := service.CountTokenInput(document, rerankRequest.Model) |
| 21 | token += tkm |
| 22 | } |
| 23 | return token |
| 24 | } |
| 25 | |
| 26 | func RerankHelper(c *gin.Context, relayMode int) (newAPIError *types.NewAPIError) { |
| 27 |
no test coverage detected