gemini ok

routers/cohere.py

@@ -10,8 +10,6 @@ router = APIRouter()
 
 db_module_filename = f"{get('project_root')}/db/cloudflare.py"
 
-print('db_module_filename',db_module_filename)
-
 @router.post("/embed")
 async def embed(request:Request, current_user: UserModel = Depends(get_current_user)):
     data = await request.json()

routers/openai_v1_1.py

@@ -1,6 +1,8 @@
 from fastapi import APIRouter, Depends, Request
 import requests
 from datetime import datetime
+from pydantic import BaseModel, Field
+from typing import List, Dict
 
 from global_state import get
 from db.tbs_db import TbsDb
@@ -9,29 +11,43 @@ from db_model.user import UserModel
 
 router = APIRouter()
 
+class Message(BaseModel):
+    role: str
+    content: str
+
+class GeminiModel(BaseModel):
+    model: str
+    temperature: float = Field(..., gt=0)
+    top_p: float = Field(..., gt=0)
+    # frequency_penalty: float = Field(..., ge=0)
+    # presence_penalty: float = Field(..., ge=0)
+    n: int = Field(..., ge=0)
+    stream: bool
+    messages: List[Message]
+
 db_module_filename = f"{get('project_root')}/db/cloudflare.py"
 
 @router.post("/chat/completions")
 async def chat_completions(request:Request, current_user: UserModel = Depends(get_current_user)):
     data = await request.json()
-    model = data.get('model', '')  
+    model = data.get('model', '')
     if (model=='')or(model is None):
         model = await get_default_model()
     api_key_info = await get_api_key(model)
     api_key = api_key_info.get('api_key', '')
     base_url = api_key_info.get('base_url', '')
+    group_name=api_key_info.get('group_name', '')
+    if group_name=='gemini':
+        data = GeminiModel(**data).model_dump()
 
     headers = {
         'Content-Type': 'application/json',
         'Authorization': f'Bearer {api_key}',
         'User-Agent': 'PostmanRuntime/7.43.0'
     }
-
-    print(f'base_url: {base_url}')
     
     try:
         response = requests.post(url=f"{base_url}/chat/completions", headers=headers, json=data)
-        # print(response.json())
         return response.json()
     except Exception as e:
         print(e)