feat: 添加重试机制和消息转换器，并支持Gemini v1beta API

2026-09-09 09:56:37 +08:00 · 2024-12-27 20:07:43 +08:00
parent 6e90463251
commit 870b1ecc17
12 changed files with 755 additions and 597 deletions
@@ -0,0 +1,50 @@
+# app/services/chat/message_converter.py
+
+from abc import ABC, abstractmethod
+from typing import List, Dict, Any
+
+class MessageConverter(ABC):
+    """消息转换器基类"""
+    
+    @abstractmethod
+    def convert(self, messages: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
+        pass
+
+class OpenAIMessageConverter(MessageConverter):
+    """OpenAI消息格式转换器"""
+    
+    def convert(self, messages: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
+        converted_messages = []
+        for msg in messages:
+            role = "user" if msg["role"] == "user" else "model"
+            parts = []
+            
+            if isinstance(msg["content"], str):
+                parts.append({"text": msg["content"]})
+            elif isinstance(msg["content"], list):
+                for content in msg["content"]:
+                    if isinstance(content, str):
+                        parts.append({"text": content})
+                    elif isinstance(content, dict):
+                        if content["type"] == "text":
+                            parts.append({"text": content["text"]})
+                        elif content["type"] == "image_url":
+                            parts.append(self._convert_image(content["image_url"]["url"]))
+                            
+            converted_messages.append({"role": role, "parts": parts})
+            
+        return converted_messages
+        
+    def _convert_image(self, image_url: str) -> Dict[str, Any]:
+        if image_url.startswith("data:image"):
+            return {
+                "inline_data": {
+                    "mime_type": "image/jpeg",
+                    "data": image_url.split(",")[1]
+                }
+            }
+        return {
+            "image_url": {
+                "url": image_url
+            }
+        }