fix

app/app_routes.py

@@ -216,26 +216,53 @@ def register_routes(app, login_required, llm_interface, retriever, stt_client, D
                 # retriever 상태 검증
                 if retriever is None:
                     logger.warning("Retriever가 초기화되지 않았습니다.")
+                    # Retriever가 None인 경우 빈 객체를 생성하고 search 메소드 추가
+                    from types import SimpleNamespace
+                    retriever = SimpleNamespace()
+                    retriever.search = lambda q, **kwargs: []
+                    logger.info("임시 retriever 객체 생성 및 빈 search 메소드 추가")
                     search_warning = "검색 기능이 아직 준비되지 않았습니다."
                 elif hasattr(retriever, 'is_mock') and retriever.is_mock:
                     logger.info("Mock Retriever 사용 중 - 검색 결과 없음.")
+                    # search 메소드가 있는지 확인, 없으면 추가
+                    if not hasattr(retriever, 'search'):
+                        retriever.search = lambda q, **kwargs: []
+                        logger.info("Mock retriever에 search 메소드 추가")
                     search_warning = "검색 인덱스가 아직 구축 중입니다. 기본 응답만 제공됩니다."
                 elif not hasattr(retriever, 'search'):
                     logger.warning("Retriever에 search 메소드가 없습니다.")
+                    # search 메소드 동적 추가
+                    retriever.search = lambda q, **kwargs: []
+                    logger.info("retriever에 빈 search 메소드 동적 추가")
                     search_warning = "검색 기능이 현재 제한되어 있습니다."
                 else:
                     try:
                         logger.info(f"검색 수행: {query[:50]}...")
                         # 오류 시 빈 결과를 반환하는 추가 try-except 블록
-                        search_results = retriever.search(query, top_k=5, first_stage_k=6)
+                        try:
+                            search_results = retriever.search(query, top_k=5, first_stage_k=6)
+                        except AttributeError as attr_err:
+                            logger.error(f"retriever에 search 메소드가 없습니다: {attr_err}", exc_info=True)
+                            # retriever가 MockComponent가 아니거나 search 메소드가 없는 경우
+                            # 동적으로 메소드 추가
+                            if not hasattr(retriever, 'search'):
+                                logger.warning("retriever에 search 메소드를 동적으로 추가합니다.")
+                                retriever.search = lambda q, **kwargs: []
+                            search_results = []
+                            search_warning = "검색 기능이 현재 제한되어 있습니다."
+                        except Exception as search_err:
+                            logger.error(f"retriever.search() 호출 중 오류 발생: {search_err}", exc_info=True)
+                            search_results = []
+                            search_warning = f"검색 중 오류 발생: {str(search_err)}"
+                            
                         if not search_results:
                             logger.info("검색 결과가 없습니다.")
                         else:
                             logger.info(f"검색 결과: {len(search_results)}개 항목")
-                    except Exception as search_err:
-                        logger.error(f"retriever.search() 호출 중 오류 발생: {search_err}", exc_info=True)
+                    except Exception as e:
+                        logger.error(f"검색 처리 중 예상치 못한 오류: {e}", exc_info=True)
                         search_results = []
-                        search_warning = f"검색 중 오류 발생: {str(search_err)}"
+                        search_warning = f"검색 처리 중 오류 발생: {str(e)}"
             except Exception as e:
                 logger.error(f"검색 중 오류 발생: {str(e)}", exc_info=True)
                 search_results = []
@@ -393,29 +420,52 @@ def register_routes(app, login_required, llm_interface, retriever, stt_client, D
                     # retriever 상태 검증
                     if retriever is None:
                         logger.warning("Retriever가 초기화되지 않았습니다.")
+                        # Retriever가 None인 경우 빈 객체를 생성하고 search 메소드 추가
+                        from types import SimpleNamespace
+                        retriever = SimpleNamespace()
+                        retriever.search = lambda q, **kwargs: []
+                        logger.info("임시 retriever 객체 생성 및 빈 search 메소드 추가")
                         search_warning = "검색 기능이 아직 준비되지 않았습니다."
                     elif hasattr(retriever, 'is_mock') and retriever.is_mock:
                         logger.info("Mock Retriever 사용 중 - ���색 결과 없음.")
+                        # search 메소드가 있는지 확인, 없으면 추가
+                        if not hasattr(retriever, 'search'):
+                            retriever.search = lambda q, **kwargs: []
+                            logger.info("Mock retriever에 search 메소드 추가")
                         search_warning = "검색 인덱스가 아직 구축 중입니다. 기본 응답만 제공됩니다."
                     elif not hasattr(retriever, 'search'):
                         logger.warning("Retriever에 search 메소드가 없습니다.")
+                        # search 메소드 동적 추가
+                        retriever.search = lambda q, **kwargs: []
+                        logger.info("retriever에 빈 search 메소드 동적 추가")
                         search_warning = "검색 기능이 현재 제한되어 있습니다."
                     else:
                         try:
                             logger.info(f"검색 수행: {transcription[:50]}...")
-                            search_results = retriever.search(transcription, top_k=5, first_stage_k=6)
+                            try:
+                                search_results = retriever.search(transcription, top_k=5, first_stage_k=6)
+                            except AttributeError as attr_err:
+                                logger.error(f"retriever에 search 메소드가 없습니다: {attr_err}", exc_info=True)
+                                # retriever가 MockComponent가 아니거나 search 메소드가 없는 경우
+                                # 동적으로 메소드 추가
+                                if not hasattr(retriever, 'search'):
+                                    logger.warning("retriever에 search 메소드를 동적으로 추가합니다.")
+                                    retriever.search = lambda q, **kwargs: []
+                                search_results = []
+                                search_warning = "검색 기능이 현재 제한되어 있습니다."
+                            except Exception as search_err:
+                                logger.error(f"retriever.search() 호출 중 오류 발생: {search_err}", exc_info=True)
+                                search_results = []
+                                search_warning = f"검색 중 오류 발생: {str(search_err)}"
+                            
                             if not search_results:
                                 logger.info("검색 결과가 없습니다.")
                             else:
                                 logger.info(f"검색 결과: {len(search_results)}개 항목")
-                        except Exception as search_err:
-                            logger.error(f"retriever.search() 호출 중 오류 발생: {search_err}", exc_info=True)
+                        except Exception as e:
+                            logger.error(f"검색 처리 중 예상치 못한 오류: {e}", exc_info=True)
                             search_results = []
-                            search_warning = f"검색 중 오류 발생: {str(search_err)}"
-                except Exception as e:
-                    logger.error(f"검색 중 오류 발생: {str(e)}", exc_info=True)
-                    search_results = []
-                    search_warning = f"검색 중 오류 발생: {str(e)}"
+                            search_warning = f"검색 처리 중 오류 발생: {str(e)}"
                 except Exception as e:
                     logger.error(f"검색 엔진 접근 중 오류 발생: {str(e)}", exc_info=True)
                     search_results = []