✨ Feature: Add feature: Support Gemini custom base URL

💰 Sponsors: Thanks to

@ioi
for the ¥50 sponsorship, sponsorship information has been added to the README.

README.md

@@ -378,6 +378,7 @@ pex -r requirements.txt \
 We thank the following sponsors for their support:
 <!-- ¥1000 -->
 - @PowerHunter: ¥1800
+- @ioi：¥50
 
 ## How to sponsor us
 

README_CN.md

@@ -367,6 +367,7 @@ pex -r requirements.txt \
 我们感谢以下赞助商的支持：
 <!-- ¥1000 -->
 - @PowerHunter：¥1800
+- @ioi：¥50
 
 ## 如何赞助我们
 

main.py

@@ -749,7 +749,7 @@ async def process_request(request: Union[RequestModel, ImageGenerationRequest, A
     parsed_url = urlparse(url)
     # print("parsed_url", parsed_url)
     engine = None
-    if parsed_url.netloc == 'generativelanguage.googleapis.com':
+    if parsed_url.path.startswith("/v1beta") or parsed_url.path.startswith("/v1"):
         engine = "gemini"
     elif parsed_url.netloc == 'aiplatform.googleapis.com':
         engine = "vertex"

request.py

@@ -124,10 +124,10 @@ async def get_gemini_payload(request, engine, provider):
     model = model_dict[request.model]
     gemini_stream = "streamGenerateContent"
     url = provider['base_url']
-    if url.endswith("v1beta"):
-        url = "https://generativelanguage.googleapis.com/v1beta/models/{model}:{stream}?key={api_key}".format(model=model, stream=gemini_stream, api_key=await provider_api_circular_list[provider['provider']].next(model))
-    if url.endswith("v1"):
-        url = "https://generativelanguage.googleapis.com/v1/models/{model}:{stream}?key={api_key}".format(model=model, stream=gemini_stream, api_key=await provider_api_circular_list[provider['provider']].next(model))
+    parsed_url = urllib.parse.urlparse(url)
+    api_version = parsed_url.path.split('/')[-1]  # 获取 v1 或 v1beta
+    # https://generativelanguage.googleapis.com/v1beta/models/
+    url = f"https://{parsed_url.netloc}/{api_version}/models/{model}:{gemini_stream}?key={await provider_api_circular_list[provider['provider']].next(model)}"
 
     messages = []
     systemInstruction = None