p2 role_label복귀

Author: kocory1

ai_server/app/application/dto/question_dto.py

@@ -18,6 +18,7 @@ class GeneratePersonalQuestionInput:
 
     family_id: str  # UUID
     member_id: str  # UUID
+    role_label: str
     base_question: str
     base_answer: str
     answered_at: datetime
@@ -33,7 +34,16 @@ class GeneratePersonalQuestionOutput:
 
 
 # 가족 질문은 동일한 DTO 사용 (입력 구조 동일)
-GenerateFamilyQuestionInput = GeneratePersonalQuestionInput
+@dataclass
+class GenerateFamilyQuestionInput:
+    """가족 질문 생성 Use Case 입력 DTO"""
+
+    family_id: str  # UUID
+    member_id: str  # UUID
+    base_question: str
+    base_answer: str
+    answered_at: datetime
+
 GenerateFamilyQuestionOutput = GeneratePersonalQuestionOutput
 
 

ai_server/app/application/use_cases/base.py

@@ -194,8 +194,12 @@ async def execute(self, input_dto: Any) -> Any:
         log_prefix = self._get_log_prefix()
         logger.info(f"[Use Case] {log_prefix} 시작")
 
-        # 1. role_label 조회 (memberId로 최근 질문에서 추출)
-        role_label = await self._get_role_label(input_dto.member_id)
+        # 1. role_label 결정
+        # - P2: API에서 role_label을 받는 경우 우선 사용
+        # - 그 외: member_id 기준 최근 질문에서 추출
+        role_label = getattr(input_dto, "role_label", None) or await self._get_role_label(
+            input_dto.member_id
+        )
         if not role_label:
             logger.warning(
                 f"[Use Case] role_label을 찾을 수 없음: member_id={input_dto.member_id}, "

ai_server/app/presentation/routers/question_router.py

@@ -73,6 +73,7 @@ async def generate_personal_question(
         use_case_input = GeneratePersonalQuestionInput(
             family_id=request.familyId,
             member_id=request.memberId,
+            role_label=request.roleLabel,
             base_question=request.baseQuestion,
             base_answer=request.baseAnswer,
             answered_at=datetime.fromisoformat(request.answeredAt.replace("Z", "+00:00")),

ai_server/app/presentation/schemas/question_schemas.py

@@ -25,6 +25,7 @@ class PersonalQuestionRequestSchema(BaseModel):
 
     familyId: str = Field(alias="familyId", description="가족 ID (UUID)")
     memberId: str = Field(alias="memberId", description="멤버 ID (UUID)")
+    roleLabel: str = Field(alias="roleLabel", description="역할 레이블 (예: '첫째 딸', '아빠')")
     baseQuestion: str = Field(alias="baseQuestion", description="기준 질문")
     baseAnswer: str = Field(alias="baseAnswer", description="기준 답변")
     answeredAt: str = Field(alias="answeredAt", description="답변 시각 (ISO 8601)")

ai_server/tests/refactoring/test_infrastructure.py

@@ -299,6 +299,7 @@ async def test_full_flow_with_infrastructure(self):
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="테스트",
             base_answer="테스트",
             answered_at=datetime.now(),

ai_server/tests/refactoring/test_use_cases.py

@@ -93,6 +93,7 @@ async def test_generate_personal_question_success(
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="오늘 뭐 했어?",
             base_answer="친구들과 놀았어요",
             answered_at=datetime(2026, 1, 20, 14, 30, 0),
@@ -138,6 +139,7 @@ async def test_generate_personal_question_with_empty_rag(
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="오늘 뭐 했어?",
             base_answer="공부했어요",
             answered_at=datetime(2026, 1, 20, 14, 30, 0),
@@ -175,6 +177,7 @@ async def test_generate_personal_question_vector_store_failure_raises_error(
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="테스트",
             base_answer="테스트",
             answered_at=datetime.now(),
@@ -347,6 +350,7 @@ async def test_regenerate_when_similarity_over_threshold(
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="오늘 뭐 했어?",
             base_answer="친구랑 놀았어요",
             answered_at=datetime(2026, 1, 20, 14, 30, 0),
@@ -361,9 +365,7 @@ async def test_regenerate_when_similarity_over_threshold(
         assert output.metadata["regeneration_count"] == 1
 
     @pytest.mark.asyncio
-    async def test_max_regeneration_limit(
-        self, mock_vector_store, mock_question_generator
-    ):
+    async def test_max_regeneration_limit(self, mock_vector_store, mock_question_generator):
         """[RED] 최대 3회 재생성 후 마지막 질문 반환"""
         from app.application.use_cases.generate_personal_question import (
             GeneratePersonalQuestionInput,
@@ -373,7 +375,8 @@ async def test_max_regeneration_limit(
 
         # Given: 모든 질문이 유사함
         mock_question_generator.generate_question.return_value = (
-            "계속 유사한 질문", QuestionLevel(2)
+            "계속 유사한 질문",
+            QuestionLevel(2),
         )
         mock_vector_store.search_similar_questions.return_value = 0.95  # 항상 유사
 
@@ -385,6 +388,7 @@ async def test_max_regeneration_limit(
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="오늘 뭐 했어?",
             base_answer="친구랑 놀았어요",
             answered_at=datetime(2026, 1, 20, 14, 30, 0),
@@ -400,9 +404,7 @@ async def test_max_regeneration_limit(
         assert output.metadata["similarity_warning"] is True
 
     @pytest.mark.asyncio
-    async def test_no_regeneration_when_unique(
-        self, mock_vector_store, mock_question_generator
-    ):
+    async def test_no_regeneration_when_unique(self, mock_vector_store, mock_question_generator):
         """[RED] 처음부터 고유하면 재생성 없음"""
         from app.application.use_cases.generate_personal_question import (
             GeneratePersonalQuestionInput,
@@ -420,6 +422,7 @@ async def test_no_regeneration_when_unique(
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="오늘 뭐 했어?",
             base_answer="친구랑 놀았어요",
             answered_at=datetime(2026, 1, 20, 14, 30, 0),
@@ -486,9 +489,7 @@ def mock_question_generator(self):
         return mock
 
     @pytest.mark.asyncio
-    async def test_family_recent_question_success(
-        self, mock_vector_store, mock_question_generator
-    ):
+    async def test_family_recent_question_success(self, mock_vector_store, mock_question_generator):
         """[RED] 가족 최근 질문 기반 생성 - 성공 케이스"""
         # Given
         from app.application.dto.question_dto import (
@@ -566,9 +567,7 @@ async def test_family_recent_question_regenerate_on_high_similarity(
         assert output.metadata["regeneration_count"] == 1
 
     @pytest.mark.asyncio
-    async def test_family_recent_question_empty_context(
-        self, mock_question_generator
-    ):
+    async def test_family_recent_question_empty_context(self, mock_question_generator):
         """[RED] 최근 질문이 없어도 생성 가능"""
         from app.application.dto.question_dto import FamilyRecentQuestionInput
         from app.application.use_cases.family_recent_question import (
@@ -609,6 +608,7 @@ def test_generate_personal_question_input_dto(self):
         input_dto = GeneratePersonalQuestionInput(
             family_id="family-1",
             member_id="member-10",
+            role_label="첫째 딸",
             base_question="테스트",
             base_answer="테스트",
             answered_at=datetime.now(),