o "i@sdZddlmZmZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZGdddee e e fZGd d d ee eefZee Zee Zd S) z5 CRUD operations for ModelProvider and Model models. )OptionalList)Session)CRUDBase) ModelProviderModel)ProviderCreateProviderUpdate ModelCreate ModelUpdatec @seZdZdZdddededeedeefdd Z ded edeefd d Z ddded edeedeefddZ ddddededede efddZ dS)CRUDModelProviderz2 CRUD operations for ModelProvider model. N) tenant_iddbnamer returncC@|ttj|k}|durttdr|tj|k}|S)z Get model provider by name. Args: db: Database session name: Provider name tenant_id: Optional tenant ID for filtering Returns: ModelProvider instance or None Nr )queryrfilterrhasattrr first)selfrrr rr9/lsinfo/ai/hellotax_ai/base_platform/app/crud/provider.py get_by_name zCRUDModelProvider.get_by_nameidcC|j||dSz5Alias for delete() method for backward compatibility.)r)delete)rrrrrrremove$zCRUDModelProvider.removecodecCr)z Get model provider by code. Args: db: Database session code: Provider code tenant_id: Optional tenant ID for filtering Returns: ModelProvider instance or None Nr )rrrr!rr r)rrr!r rrrr get_by_code(rzCRUDModelProvider.get_by_coderdskiplimitr%r&cCs&|ttjdk||S)a Get all active model providers. Args: db: Database session skip: Number of records to skip limit: Maximum number of records to return Returns: List of active provider instances T)rrr is_activeoffsetr&all)rrr%r&rrrget_active_providers;s z&CRUDModelProvider.get_active_providers)__name__ __module__ __qualname____doc__rstrrintrrrr"rr*rrrrr s ((r c @seZdZdZdddededeedeefdd Z ded edee fd d Z ded edeefddZ ddddedededede ef ddZddddedededede ef ddZdddddedeededede ef ddZdS) CRUDModelz* CRUD operations for Model model. N) provider_idrrr2rcCsdS)z Get model by name. Args: db: Database session name: Model name provider_id: Optional provider ID to filter by Returns: Model instance or None Nr)rrrr2rrrrWszCRUDModel.get_by_namercCrr)rrrrrr2r)rrrrrrrrir zCRUDModel.remover!cCs|ttj|kS)z Get model by code. Args: db: Database session code: Model code Returns: Model instance or None )rrrr!r)rrr!rrrr"ts zCRUDModel.get_by_coderr#r$r%r&cC&|ttj|k||S)a Get models by provider. Args: db: Database session provider_id: Provider ID skip: Number of records to skip limit: Maximum number of records to return Returns: List of model instances )rrrr2r(r&r))rrr2r%r&rrrget_by_provider zCRUDModel.get_by_provider model_typecCr3)a- Get models by type. Args: db: Database session model_type: Model type (e.g., 'chat', 'embedding') skip: Number of records to skip limit: Maximum number of records to return Returns: List of model instances )rrrtyper(r&r))rrr6r%r&rrr get_by_typer5zCRUDModel.get_by_type)r2r%r&cCs>|ttjdk}|r|tj|k}|||S)a/ Get active models. Args: db: Database session provider_id: Optional provider ID to filter by skip: Number of records to skip limit: Maximum number of records to return Returns: List of active model instances T)rrrr'r2r(r&r))rrr2r%r&rrrrget_active_modelsszCRUDModel.get_active_models)r+r,r-r.rr/rr0rrrrr"rr4r8r9rrrrr1Rsl     r1N)r.typingrrsqlalchemy.ormr app.crud.baserapp.models.providerrrapp.schemas.providerrr r r r r1providermodelrrrrs  F|