klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-09 17:57:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Gemini

Browse Source
This commit is contained in:
fit2cloud-chenyw
2025-08-05 14:48:57 +08:00
parent 5654c732ce
commit 7df8d49d48
5 changed files with 64 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/system/api/aimodel.py
+2 -2
View File
@@ -10,7 +10,7 @@ from sqlmodel import func, select, update
from apps.system.models.system_model import AiModelDetail
from common.core.deps import SessionDep, Trans
from common.utils.time import get_timestamp
from common.utils.utils import SQLBotLogUtil
from common.utils.utils import SQLBotLogUtil, prepare_model_arg
router = APIRouter(tags=["system/aimodel"], prefix="/system/aimodel")
@@ -18,7 +18,7 @@ router = APIRouter(tags=["system/aimodel"], prefix="/system/aimodel")
async def check_llm(info: AiModelCreator, trans: Trans):
async def generate():
try:
additional_params = {item.key: item.val for item in info.config_list}
additional_params = {item.key: prepare_model_arg(item.val) for item in info.config_list if item.key and item.val}
config = LLMConfig(
model_type="openai",
model_name=info.base_model,