release

vector/app/models/vector_models.py

@@ -20,6 +20,26 @@ class VectorBuildRequest(BaseModel):
         example=False
     )
 
+class StoredDataInfo(BaseModel):
+    """Vector DB에 저장된 데이터 정보"""
+    store_info: Dict[str, Any] = Field(description="가게 정보")
+    reviews: List[Dict[str, Any]] = Field(description="리뷰 목록") 
+    review_summary: Dict[str, Any] = Field(description="리뷰 요약 정보")
+    combined_at: str = Field(description="데이터 결합 시간")
+
+class FindReviewsResponse(BaseModel):
+    """리뷰 검색 응답 모델 - Vector DB 저장 데이터 포함"""
+    success: bool = Field(description="검색 성공 여부")
+    message: str = Field(description="응답 메시지")
+    target_store: Optional[Dict[str, Any]] = Field(None, description="본인 가게 정보")
+    food_category: Optional[str] = Field(None, description="추출된 음식 카테고리")
+    total_reviews: int = Field(default=0, description="총 수집된 리뷰 수")
+    total_stores: int = Field(default=0, description="분석된 동종 업체 수")
+    execution_time: Optional[float] = Field(None, description="실행 시간(초)")
+    # Vector DB 저장 데이터
+    stored_data: Optional[Dict[str, StoredDataInfo]] = Field(None, description="Vector DB에 저장된 데이터")
+    sample_similar_stores: List[str] = Field(default=[], description="동종 업체 샘플 목록")
+
 class VectorBuildResponse(BaseModel):
     """Vector DB 구축 응답 모델"""
     success: bool = Field(description="구축 성공 여부")