Просмотр исходного кода

fix: improve reasoning effort model suffix handling

- Remove model name suffixes after extracting reasoning effort
- Update upstream model name to reflect the base model
- Ensure clean model name is passed to the upstream service
1808837298@qq.com 1 год назад
Родитель
Сommit
a29e1e0aa3
1 измененных файлов с 4 добавлено и 0 удалено
  1. 4 0
      relay/channel/openai/adaptor.go

+ 4 - 0
relay/channel/openai/adaptor.go

@@ -119,11 +119,15 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, re
 		}
 		if strings.HasSuffix(request.Model, "-high") {
 			request.ReasoningEffort = "high"
+			request.Model = strings.TrimSuffix(request.Model, "-high")
 		} else if strings.HasSuffix(request.Model, "-low") {
 			request.ReasoningEffort = "low"
+			request.Model = strings.TrimSuffix(request.Model, "-low")
 		} else if strings.HasSuffix(request.Model, "-medium") {
 			request.ReasoningEffort = "medium"
+			request.Model = strings.TrimSuffix(request.Model, "-medium")
 		}
+		info.UpstreamModelName = request.Model
 	}
 	if request.Model == "o1" || request.Model == "o1-2024-12-17" || strings.HasPrefix(request.Model, "o3") {
 		//修改第一个Message的内容,将system改为developer