Fix the issue where Gemini cannot be used.

main.py

@@ -68,7 +68,8 @@ async def process_request(request: RequestModel, provider: Dict):
     #     "headers": headers,
     #     "payload": payload
     # }
-    # print(f"Request details: {json.dumps(request_info, indent=4, ensure_ascii=False)}")
+    # import json
+    # logger.info(f"Request details: {json.dumps(request_info, indent=4, ensure_ascii=False)}")
 
     if request.stream:
         model = provider['model'][request.model]

request.py

@@ -47,7 +47,8 @@ async def get_gemini_payload(request, engine, provider):
 
     messages = []
     for msg in request.messages:
-        name = None
+        if msg.role == "assistant":
+            msg.role = "model"
         if isinstance(msg.content, list):
             content = []
             for item in msg.content:
@@ -59,7 +60,7 @@ async def get_gemini_payload(request, engine, provider):
                     image_message = await get_image_message(item.image_url.url, engine)
                     content.append(image_message)
         else:
-            content = msg.content
+            content = [{"text": msg.content}]
         if msg.role != "system":
             messages.append({"role": msg.role, "parts": content})
 

response.py

@@ -39,6 +39,14 @@ async def generate_sse_response(timestamp, model, content=None, tools_id=None, f
 async def fetch_gemini_response_stream(client, url, headers, payload, model):
     timestamp = datetime.timestamp(datetime.now())
     async with client.stream('POST', url, headers=headers, json=payload) as response:
+        if response.status_code != 200:
+            error_message = await response.aread()
+            error_str = error_message.decode('utf-8', errors='replace')
+            try:
+                error_json = json.loads(error_str)
+            except json.JSONDecodeError:
+                error_json = error_str
+            yield {"error": f"fetch_gpt_response_stream HTTP Error {response.status_code}", "details": error_json}
         buffer = ""
         async for chunk in response.aiter_text():
             buffer += chunk