From 97d6f10f15add4e8ad564006d749235fb0c7f99e Mon Sep 17 00:00:00 2001
From: CaIon <i@caion.me>
Date: Sat, 2 Aug 2025 12:53:58 +0800
Subject: [PATCH] feat: enhance ConvertGeminiRequest to set default role and
 handle YouTube video MIME type

---
 relay/channel/gemini/adaptor.go | 16 ++++++++++++++++
 relay/channel/vertex/adaptor.go |  3 ++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/relay/channel/gemini/adaptor.go b/relay/channel/gemini/adaptor.go
index 20d43020..14fd278d 100644
--- a/relay/channel/gemini/adaptor.go
+++ b/relay/channel/gemini/adaptor.go
@@ -21,6 +21,22 @@ type Adaptor struct {
 }
 
 func (a *Adaptor) ConvertGeminiRequest(c *gin.Context, info *relaycommon.RelayInfo, request *dto.GeminiChatRequest) (any, error) {
+	if len(request.Contents) > 0 {
+		for i, content := range request.Contents {
+			if i == 0 {
+				if request.Contents[0].Role == "" {
+					request.Contents[0].Role = "user"
+				}
+			}
+			for _, part := range content.Parts {
+				if part.FileData != nil {
+					if part.FileData.MimeType == "" && strings.Contains(part.FileData.FileUri, "www.youtube.com") {
+						part.FileData.MimeType = "video/webm"
+					}
+				}
+			}
+		}
+	}
 	return request, nil
 }
 
diff --git a/relay/channel/vertex/adaptor.go b/relay/channel/vertex/adaptor.go
index 39be998e..9b62cffc 100644
--- a/relay/channel/vertex/adaptor.go
+++ b/relay/channel/vertex/adaptor.go
@@ -45,7 +45,8 @@ type Adaptor struct {
 }
 
 func (a *Adaptor) ConvertGeminiRequest(c *gin.Context, info *relaycommon.RelayInfo, request *dto.GeminiChatRequest) (any, error) {
-	return request, nil
+	geminiAdaptor := gemini.Adaptor{}
+	return geminiAdaptor.ConvertGeminiRequest(c, info, request)
 }
 
 func (a *Adaptor) ConvertClaudeRequest(c *gin.Context, info *relaycommon.RelayInfo, request *dto.ClaudeRequest) (any, error) {