From ded82cd729ba9c01ee214f343dec8953bab89095 Mon Sep 17 00:00:00 2001 From: 9qeklajc <9qeklajc> Date: Tue, 4 Nov 2025 23:16:08 +0100 Subject: [PATCH] fix model endpoint & filter query --- routstr/proxy.py | 5 +++-- scripts/build-ui.sh | 1 + ui/components/ModelSelector.tsx | 12 ++++-------- 3 files changed, 8 insertions(+), 10 deletions(-) diff --git a/routstr/proxy.py b/routstr/proxy.py index 8eece67b..914d20bd 100644 --- a/routstr/proxy.py +++ b/routstr/proxy.py @@ -3,7 +3,7 @@ from typing import Any from fastapi import APIRouter, Depends, HTTPException, Request from fastapi.responses import Response, StreamingResponse -from sqlmodel import col, select +from sqlmodel import select from .algorithm import create_model_mappings from .auth import pay_for_request, revert_pay_for_request, validate_bearer_key @@ -101,9 +101,10 @@ async def refresh_model_maps() -> None: } disabled_result = await session.exec( - select(ModelRow.id).where(ModelRow.enabled == False) + select(ModelRow.id).where(~ModelRow.enabled) ) disabled_model_ids = {row for row in disabled_result.all()} + print(disabled_model_ids) _model_instances, _provider_map, _unique_models = create_model_mappings( upstreams=_upstreams, diff --git a/scripts/build-ui.sh b/scripts/build-ui.sh index 557c26db..7a235f01 100755 --- a/scripts/build-ui.sh +++ b/scripts/build-ui.sh @@ -49,6 +49,7 @@ else npm run build fi +rm -rf ../ui_out mkdir -p ../ui_out mv out/* ../ui_out diff --git a/ui/components/ModelSelector.tsx b/ui/components/ModelSelector.tsx index 5b03e350..1a9a2814 100644 --- a/ui/components/ModelSelector.tsx +++ b/ui/components/ModelSelector.tsx @@ -387,14 +387,10 @@ export function ModelSelector({ providerIdNum, model.id ); - await AdminService.updateProviderModel( - providerIdNum, - model.full_name, - { - ...existingModel, - enabled: true, - } - ); + await AdminService.updateProviderModel(providerIdNum, model.id, { + ...existingModel, + enabled: true, + }); totalEnabled++; } catch (error) { console.error(`Failed to enable model ${model.full_name}:`, error);