15 lines
412 B
Python
15 lines
412 B
Python
from typing import Optional
|
|
|
|
from sqlalchemy import select
|
|
|
|
from .base import BaseRepository
|
|
from ..models import LLMConfig
|
|
|
|
|
|
class LLMConfigRepository(BaseRepository[LLMConfig]):
|
|
model = LLMConfig
|
|
|
|
async def get_by_user(self, user_id: int) -> Optional[LLMConfig]:
|
|
result = await self.session.execute(select(LLMConfig).where(LLMConfig.user_id == user_id))
|
|
return result.scalars().first()
|