修复gemini处理图片不正确的问题

fruitbars · Jul 1, 2024 · 450a3cc · 450a3cc
1 parent e572ba8
commit 450a3cc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pkg/adapter/gemini_request_openai.go b/pkg/adapter/gemini_request_openai.go
@@ -27,7 +27,7 @@ func DeepCopyGeminiRequest(original *googlegemini.GeminiRequest) (*googlegemini.
 	for i, _ := range copyReq.Contents {
 		for j, _ := range copyReq.Contents[i].Parts {
 			if copyReq.Contents[i].Parts[j].InlineData != nil {
-				d := []byte("...")
+				d := "..."
 				copyReq.Contents[i].Parts[j].InlineData.Data = d
 			}
 
@@ -131,7 +131,7 @@ func createPartFromMessageContent(mc openai.ChatMessagePart) (*googlegemini.Part
 		}
 		blob := googlegemini.Blob{
 			MimeType: mimeType,
-			Data:     []byte(imgData),
+			Data:     imgData,
 		}
 		return &googlegemini.Part{InlineData: &blob}, nil
 	default:

diff --git a/pkg/llm/google-gemini/gemini_request.go b/pkg/llm/google-gemini/gemini_request.go
@@ -10,7 +10,7 @@ type Part struct {
 // Blob 表示内嵌的媒体字节数据
 type Blob struct {
 	MimeType string `json:"mimeType,omitempty"`
-	Data     []byte `json:"data,omitempty"`
+	Data     string `json:"data,omitempty"`
 }
 
 // Entry represents a single entry in the conversation.