Add query conditions to query all models

Author: wu-clan

api/v1/model.py

@@ -1,6 +1,6 @@
 from typing import Annotated
 
-from fastapi import APIRouter, Depends, Path
+from fastapi import APIRouter, Depends, Path, Query
 
 from backend.common.pagination import DependsPagination, PageData
 from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
@@ -20,8 +20,10 @@
 
 
 @router.get('/all', summary='获取所有模型', dependencies=[DependsJwtAuth])
-async def get_all_ai_models(db: CurrentSession) -> ResponseSchemaModel[list[GetAIModelDetail]]:
-    data = await ai_model_service.get_all(db=db)
+async def get_all_ai_models(
+    db: CurrentSession, provider_id: Annotated[int, Query(description='供应商 ID')]
+) -> ResponseSchemaModel[list[GetAIModelDetail]]:
+    data = await ai_model_service.get_all(db=db, provider_id=provider_id)
     return response_base.success(data=data)
 
 

crud/crud_model.py

@@ -35,14 +35,15 @@ async def get_select(self) -> Select:
         """获取模型列表查询表达式"""
         return await self.select_order('id', 'desc')
 
-    async def get_all(self, db: AsyncSession) -> Sequence[AIModel]:
+    async def get_all(self, db: AsyncSession, provider_id: int) -> Sequence[AIModel]:
         """
         获取所有模型
 
         :param db: 数据库会话
+        :param provider_id: 供应商 ID
         :return:
         """
-        return await self.select_models(db)
+        return await self.select_models(db, provider_id=provider_id)
 
     async def create(self, db: AsyncSession, obj: CreateAIModelParam) -> None:
         """

service/model_service.py

@@ -39,14 +39,15 @@ async def get_list(db: AsyncSession) -> dict[str, Any]:
         return await paging_data(db, ai_model_select)
 
     @staticmethod
-    async def get_all(*, db: AsyncSession) -> Sequence[AIModel]:
+    async def get_all(*, db: AsyncSession, provider_id: int) -> Sequence[AIModel]:
         """
         获取所有 AI 模型
 
         :param db: 数据库会话
+        :param provider_id: 供应商 ID
         :return:
         """
-        ai_models = await ai_model_dao.get_all(db)
+        ai_models = await ai_model_dao.get_all(db, provider_id=provider_id)
         return ai_models
 
     @staticmethod