diff --git a/repository.py b/repository.py index b4cdff1..3a9a7c4 100644 --- a/repository.py +++ b/repository.py @@ -38,7 +38,7 @@ class QuestionRepository: await session.flush() await session.commit() - return question.id + return question.id, question.answer @classmethod async def find_all(cls) -> list[SQuestion]: diff --git a/router_questions.py b/router_questions.py index 705f9f3..b49e9b0 100644 --- a/router_questions.py +++ b/router_questions.py @@ -17,8 +17,8 @@ async def add_question( type_mood: TypeMood, # Добавлен параметр type_mood type_model: TypeModel, # Добавлен параметр type_model ) -> SQuestionId: - question_id = await QuestionRepository.add_one(question, type_mood, type_model) # Передача параметров type_mood и type_model - return {"ok": True, "question_id": question_id} + question_id, answer = await QuestionRepository.add_one(question, type_mood, type_model) # Передача параметров type_mood и type_model + return {"question_id": question_id, "answer": answer} @router.get("") async def get_questions() -> list[SQuestion]: diff --git a/schemas.py b/schemas.py index e226ce2..2791edb 100644 --- a/schemas.py +++ b/schemas.py @@ -17,8 +17,8 @@ class SQuestion(SQuestionAdd): model_config = ConfigDict(from_attributes=True) class SQuestionId(BaseModel): - ok: bool = True question_id: int + answer: str class Flight(BaseModel): id: int