From 812c188ab1e35d3265b61acf12c99d6230f280b8 Mon Sep 17 00:00:00 2001
From: "1808837298@qq.com" <1808837298@qq.com>
Date: Tue, 18 Feb 2025 16:00:56 +0800
Subject: [PATCH] fix: Extend temperature handling for OpenAI-like models

- Add support for suppressing temperature for o1 models
- Expand model prefix check to include 'o1' alongside 'o3' models
---
 relay/channel/openai/adaptor.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/relay/channel/openai/adaptor.go b/relay/channel/openai/adaptor.go
index e94399ea..f927fa74 100644
--- a/relay/channel/openai/adaptor.go
+++ b/relay/channel/openai/adaptor.go
@@ -119,7 +119,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, re
 			request.MaxCompletionTokens = request.MaxTokens
 			request.MaxTokens = 0
 		}
-		if strings.HasPrefix(request.Model, "o3") {
+		if strings.HasPrefix(request.Model, "o3") || strings.HasPrefix(request.Model, "o1") {
 			request.Temperature = nil
 		}
 		if strings.HasSuffix(request.Model, "-high") {