mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-10 17:20:41 +03:00
1e79c981c2
* remove fastapi users dependency * add user service to chatbot service * add user save on bot info command * add user model to admin * fix tests
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from fastapi import Depends
|
|
from starlette.requests import Request
|
|
from telegram import Update
|
|
|
|
from api.auth.deps import get_user_service
|
|
from api.deps import get_database
|
|
from core.auth.services import UserService
|
|
from core.bot.app import BotApplication, BotQueue
|
|
from core.bot.repository import ChatGPTRepository
|
|
from core.bot.services import ChatGptService
|
|
from infra.database.db_adapter import Database
|
|
from settings.config import AppSettings, get_settings
|
|
|
|
|
|
def get_bot_app(request: Request) -> BotApplication:
|
|
return request.app.state.bot_app
|
|
|
|
|
|
def get_bot_queue(request: Request) -> BotQueue:
|
|
return request.app.state.queue
|
|
|
|
|
|
async def get_update_from_request(request: Request, bot_app: BotApplication = Depends(get_bot_app)) -> Update | None:
|
|
data = await request.json()
|
|
return Update.de_json(data, bot_app.bot)
|
|
|
|
|
|
def get_chatgpt_repository(
|
|
db: Database = Depends(get_database), settings: AppSettings = Depends(get_settings)
|
|
) -> ChatGPTRepository:
|
|
return ChatGPTRepository(settings=settings, db=db)
|
|
|
|
|
|
def new_bot_queue(bot_app: BotApplication = Depends(get_bot_app)) -> BotQueue:
|
|
return BotQueue(bot_app=bot_app)
|
|
|
|
|
|
def get_chatgpt_service(
|
|
chatgpt_repository: ChatGPTRepository = Depends(get_chatgpt_repository),
|
|
user_service: UserService = Depends(get_user_service),
|
|
) -> ChatGptService:
|
|
return ChatGptService(repository=chatgpt_repository, user_service=user_service)
|