Merge branch 'develop'

CutTheWire · CutTheWire · commit a0e78809eef7 · 2025-05-30T21:52:47.000+09:00
diff --git a/README.md b/README.md
@@ -36,6 +36,21 @@ ChatBot-AI/
 
 ---
 
+## 📋 UML 클래스 다이어그램 
+### 📑 ChatBot-AI/fastapi/src/utils/ai_models 클래스 다이어그램 
+![image](https://lh3.googleusercontent.com/d/11BO1kgmcn_I0N-gAegB8p36-PrAm4IHn)
+
+### 📑 ChatBot-AI/fastapi/src/utils/handlers 클래스 다이어그램 
+![image](https://lh3.googleusercontent.com/d/10s3xwUFxnmfKb8WBEvU3jqQhJgExNa28)
+
+### 📑 ChatBot-AI/fastapi/src/utils/schemas 클래스 다이어그램
+![image](https://lh3.googleusercontent.com/d/1Az97lKerSOJltMPWEMeAW6G72axCdIii)
+
+## 📋 UML 패키지 다이어그램 
+![image](https://lh3.googleusercontent.com/d/1_fifSzf7YFoEMQd80hUQGgF0rI0vsYtm)
+
+---
+
 ## 🚀 빠른 시작 (Docker 기반)
 
 ### 1. **필수 요구사항**
diff --git a/fastapi/src/server-character/server.py b/fastapi/src/server-character/server.py
@@ -52,6 +52,8 @@ async def lifespan(app: FastAPI):
     """    
     try:
         assert AppState.LlamaCharacter_model is not None, "LlamaCharacter_model is not initialized"
+        if AppState.mongo_handler is not None:
+            await AppState.mongo_handler.init()
     except AssertionError as e:
         print(f"{RED}ERROR{RESET}:    {str(e)}")
     print(f"{GREEN}INFO{RESET}:     LlamaCharacter 모델 로드 완료")
diff --git a/fastapi/src/server-character/utils/app_state.py b/fastapi/src/server-character/utils/app_state.py
@@ -7,11 +7,11 @@
 RESET = "\033[0m"
 
 LlamaCharacter_model: Optional[LlamaCharacterModel] = None
+mongo_handler: Optional[MongoDBHandler] = None
 
 try:
-    # Character 서버는 GPU 1 사용
-    LlamaCharacter_model = LlamaCharacterModel()  # 컨테이너 내에서는 0이 RTX 3060
-    mongo_handler: Optional[MongoDBHandler] = MongoDBHandler()
+    LlamaCharacter_model = LlamaCharacterModel()
+    mongo_handler = MongoDBHandler()  # 비동기 초기화는 lifespan에서!
 except ChatError.InternalServerErrorException as e:
     mongo_handler = None
     print(f"{RED}ERROR{RESET}:    MongoDB 초기화 오류 발생: {str(e)}")
diff --git a/fastapi/src/server-character/utils/handlers/mongodb_handler.py b/fastapi/src/server-character/utils/handlers/mongodb_handler.py
@@ -66,133 +66,44 @@ def __init__(self) -> None:
                 "&socketTimeoutMS=6000"   # 네 번째 옵션
             )
             
-            # 이벤트 루프 가져오기
-            self.loop = asyncio.get_event_loop()
-            
-            # MongoDB 클라이언트 초기화 (이벤트 루프 지정)
-            self.client = AsyncIOMotorClient(
-                self.mongo_uri,
-                io_loop = self.loop,
-                serverSelectionTimeoutMS = 500,
-                connectTimeoutMS = 1000,
-                socketTimeoutMS = 6000
-            )
-            
-            # 연결 테스트는 비동기로 수행
-            async def test_connection():
-                await self.client.admin.command('ping')
-                
-            # 연결 테스트 실행
-            self.loop.run_until_complete(test_connection())
-            self.db = self.client[mongo_db]
-            print(f"{GREEN}INFO{RESET}:     MongoDB 연결 성공: {mongo_host}:{mongo_port}")\
-                
-        except PyMongoError as e:
-            print(f"{RED}ERROR{RESET}:    MongoDB 연결 실패")
-            raise InternalServerErrorException(detail = f"MongoDB 연결 오류 - 호스트: {mongo_host}, 포트: {mongo_port}")
+            self.client = None
+            self.db = None
+            print(f"{GREEN}INFO{RESET}:     MongoDBHandler 초기화 완료")
         except Exception as e:
             raise InternalServerErrorException(detail = f"MongoDBHandler 초기화 오류: {str(e)}")
-    
-    async def get_office_log(self, user_id: str, document_id: str, router: str) -> List[Dict]:
-        try:
-            collection = self.db[f'{router}_log_{user_id}']
-            
-            # 이벤트 루프 확인 및 설정
-            if self.loop !=  asyncio.get_event_loop():
-                self.loop = asyncio.get_event_loop()
-                self.client = AsyncIOMotorClient(
-                    self.mongo_uri,
-                    io_loop = self.loop
-                )
-                self.db = self.client[os.getenv("MONGO_DATABASE")]
-                collection = self.db[f'{router}_log_{user_id}']
-
-            document = await collection.find_one({"id": document_id})
-
-            if document is None or not document.get("value", []):
-                return []
-
-            value_list = document.get("value", [])
-            sorted_value_list = sorted(value_list, key = lambda x: x.get("index", 0))
-            
-            # 최신 8개만 선택
-            latest_messages = sorted_value_list[-8:] if len(sorted_value_list) > 8 else sorted_value_list
-
-            # 대화 기록 형식 수정
-            formatted_chat_list = []
-            for chat in latest_messages:
-                formatted_chat = {
-                    "index": chat.get("index"),
-                    "input_data": chat.get("input_data"),  # input_data 직접 사용
-                    "output_data": chat.get("output_data") # output_data 직접 사용
-                }
-                formatted_chat_list.append(formatted_chat)
-
-            return formatted_chat_list
 
+    async def init(self):
+        try:
+            self.client = AsyncIOMotorClient(self.mongo_uri)
+            await self.client.admin.command('ping')
+            self.db = self.client[os.getenv("MONGO_DATABASE")]
+            print(f"{GREEN}INFO{RESET}:     MongoDB 연결 성공")
         except PyMongoError as e:
-            raise InternalServerErrorException(detail = f"Error retrieving chatlog value: {str(e)}")
+            print(f"{RED}ERROR{RESET}:    MongoDB 연결 실패")
+            raise InternalServerErrorException(detail = f"MongoDB 연결 오류 - 호스트: {os.getenv('MONGO_HOST')}, 포트: {os.getenv('MONGO_PORT')}")
         except Exception as e:
-            raise InternalServerErrorException(detail = f"Unexpected error: {str(e)}")
-    
+            raise InternalServerErrorException(detail = f"MongoDBHandler 초기화 오류: {str(e)}")
+
     async def get_character_log(self, user_id: str, document_id: str, router: str) -> List[Dict]:
         """
         최신 10개의 대화 기록을 가져와서 Llama 프롬프트 형식으로 변환합니다.
-
-        Args:
-            user_id (str): 사용자 ID
-            document_id (str): 문서 ID
-            router (str): 라우터 이름
-
-        Returns:
-            List[Dict]: Llama 프롬프트 형식의 대화 기록 리스트. 대화 기록이 없으면 빈 리스트 반환.
         """
         try:
             collection = self.db[f'{router}_log_{user_id}']
-            
-            # 이벤트 루프 확인 및 설정
-            if self.loop !=  asyncio.get_event_loop():
-                self.loop = asyncio.get_event_loop()
-                self.client = AsyncIOMotorClient(
-                    self.mongo_uri,
-                    io_loop = self.loop
-                )
-                self.db = self.client[os.getenv("MONGO_DATABASE")]
-                collection = self.db[f'{router}_log_{user_id}']
-            
+
             document = await collection.find_one({"id": document_id})
 
-            # 문서가 없거나 value 리스트가 비어있는 경우 빈 리스트 반환
             if document is None or not document.get("value", []):
                 return []
 
             value_list = document.get("value", [])
-            
-            # index 기준으로 정렬
             sorted_value_list = sorted(value_list, key = lambda x: x.get("index", 0))
-            
-            # 최신 10개만 선택 (마지막 10개)
             latest_messages = sorted_value_list[-10:] if len(sorted_value_list) > 10 else sorted_value_list
 
-            # 대화 기록 변환 수행
-            formatted_chat_list = []
-            for chat in latest_messages:
-                formatted_chat = {
-                    "index": chat.get("index"),
-                    "img_url": chat.get("img_url"),
-                    "dialogue": (
-                        f"<|start_header_id|>user<|end_header_id|>\n"
-                        f"{chat.get('input_data')}<|eot_id|>"
-                        f"<|start_header_id|>assistant<|end_header_id|>\n"
-                        f"{chat.get('output_data')}<|eot_id|>"
-                    )
-                }
-                formatted_chat_list.append(formatted_chat)
-
-            return formatted_chat_list
+            # 필요시 포맷 변환
+            return latest_messages
 
         except PyMongoError as e:
             raise InternalServerErrorException(detail = f"Error retrieving chatlog value: {str(e)}")
         except Exception as e:
             raise InternalServerErrorException(detail = f"Unexpected error: {str(e)}")
-        
diff --git a/fastapi/src/server-office/server.py b/fastapi/src/server-office/server.py
@@ -52,13 +52,12 @@ async def lifespan(app: FastAPI):
     """    
     try:
         assert AppState.LlamaOffice_model is not None, "LlamaOffice_model is not initialized"
+        if AppState.mongo_handler is not None:
+            await AppState.mongo_handler.init()
     except AssertionError as e:
         print(f"{RED}ERROR{RESET}:    {str(e)}")
     print(f"{GREEN}INFO{RESET}:     LlamaOffice 모델 로드 완료")
-
     yield
-
-    # 모델 메모리 해제
     AppState.LlamaOffice_model = None
     print(f"{GREEN}INFO{RESET}:     모델 해제 완료")
 
diff --git a/fastapi/src/server-office/utils/app_state.py b/fastapi/src/server-office/utils/app_state.py
@@ -7,11 +7,11 @@
 RESET = "\033[0m"
 
 LlamaOffice_model: Optional[LlamaOfficeModel] = None
+mongo_handler: Optional[MongoDBHandler] = None
 
 try:
-    # Office 서버는 GPU 0 사용
     LlamaOffice_model = LlamaOfficeModel()
-    mongo_handler: Optional[MongoDBHandler] = MongoDBHandler()
+    mongo_handler = MongoDBHandler()
 except ChatError.InternalServerErrorException as e:
     mongo_handler = None
     print(f"{RED}ERROR{RESET}:    MongoDB 초기화 오류 발생: {str(e)}")
diff --git a/fastapi/src/server-office/utils/handlers/mongodb_handler.py b/fastapi/src/server-office/utils/handlers/mongodb_handler.py
@@ -66,27 +66,33 @@ def __init__(self) -> None:
                 "&socketTimeoutMS=6000"   # 네 번째 옵션
             )
             
-            # 이벤트 루프 가져오기
-            self.loop = asyncio.get_event_loop()
+            # 클라이언트와 데이터베이스 초기화
+            self.client = None
+            self.db = None
             
-            # MongoDB 클라이언트 초기화 (이벤트 루프 지정)
-            self.client = AsyncIOMotorClient(
-                self.mongo_uri,
-                io_loop = self.loop,
-                serverSelectionTimeoutMS = 500,
-                connectTimeoutMS = 1000,
-                socketTimeoutMS = 6000
-            )
-            
-            # 연결 테스트는 비동기로 수행
-            async def test_connection():
-                await self.client.admin.command('ping')
-                
-            # 연결 테스트 실행
-            self.loop.run_until_complete(test_connection())
-            self.db = self.client[mongo_db]
-            print(f"{GREEN}INFO{RESET}:     MongoDB 연결 성공: {mongo_host}:{mongo_port}")\
+            print(f"{GREEN}INFO{RESET}:     MongoDBHandler 초기화 완료")
                 
+        except Exception as e:
+            raise InternalServerErrorException(detail = f"MongoDBHandler 초기화 오류: {str(e)}")
+    
+    async def init(self):
+        """
+        MongoDB 클라이언트를 초기화하고 데이터베이스에 연결합니다.
+
+        Raises:
+            InternalServerErrorException: MongoDB 연결 중 오류 발생 시
+        """
+        try:
+            # MongoDB 클라이언트 초기화
+            self.client = AsyncIOMotorClient(self.mongo_uri)
+            
+            # 연결 테스트
+            await self.client.admin.command('ping')
+            
+            # 데이터베이스 선택
+            self.db = self.client[os.getenv("MONGO_DATABASE")]
+            print(f"{GREEN}INFO{RESET}:     MongoDB 연결 성공")
+        
         except PyMongoError as e:
             print(f"{RED}ERROR{RESET}:    MongoDB 연결 실패")
             raise InternalServerErrorException(detail = f"MongoDB 연결 오류 - 호스트: {mongo_host}, 포트: {mongo_port}")
@@ -96,16 +102,6 @@ async def test_connection():
     async def get_office_log(self, user_id: str, document_id: str, router: str) -> List[Dict]:
         try:
             collection = self.db[f'{router}_log_{user_id}']
-            
-            # 이벤트 루프 확인 및 설정
-            if self.loop !=  asyncio.get_event_loop():
-                self.loop = asyncio.get_event_loop()
-                self.client = AsyncIOMotorClient(
-                    self.mongo_uri,
-                    io_loop = self.loop
-                )
-                self.db = self.client[os.getenv("MONGO_DATABASE")]
-                collection = self.db[f'{router}_log_{user_id}']
 
             document = await collection.find_one({"id": document_id})
 
@@ -114,7 +110,7 @@ async def get_office_log(self, user_id: str, document_id: str, router: str) -> L
 
             value_list = document.get("value", [])
             sorted_value_list = sorted(value_list, key = lambda x: x.get("index", 0))
-            
+
             # 최신 8개만 선택
             latest_messages = sorted_value_list[-8:] if len(sorted_value_list) > 8 else sorted_value_list
 
@@ -123,69 +119,8 @@ async def get_office_log(self, user_id: str, document_id: str, router: str) -> L
             for chat in latest_messages:
                 formatted_chat = {
                     "index": chat.get("index"),
-                    "input_data": chat.get("input_data"),  # input_data 직접 사용
-                    "output_data": chat.get("output_data") # output_data 직접 사용
-                }
-                formatted_chat_list.append(formatted_chat)
-
-            return formatted_chat_list
-
-        except PyMongoError as e:
-            raise InternalServerErrorException(detail = f"Error retrieving chatlog value: {str(e)}")
-        except Exception as e:
-            raise InternalServerErrorException(detail = f"Unexpected error: {str(e)}")
-    
-    async def get_character_log(self, user_id: str, document_id: str, router: str) -> List[Dict]:
-        """
-        최신 10개의 대화 기록을 가져와서 Llama 프롬프트 형식으로 변환합니다.
-
-        Args:
-            user_id (str): 사용자 ID
-            document_id (str): 문서 ID
-            router (str): 라우터 이름
-
-        Returns:
-            List[Dict]: Llama 프롬프트 형식의 대화 기록 리스트. 대화 기록이 없으면 빈 리스트 반환.
-        """
-        try:
-            collection = self.db[f'{router}_log_{user_id}']
-            
-            # 이벤트 루프 확인 및 설정
-            if self.loop !=  asyncio.get_event_loop():
-                self.loop = asyncio.get_event_loop()
-                self.client = AsyncIOMotorClient(
-                    self.mongo_uri,
-                    io_loop = self.loop
-                )
-                self.db = self.client[os.getenv("MONGO_DATABASE")]
-                collection = self.db[f'{router}_log_{user_id}']
-            
-            document = await collection.find_one({"id": document_id})
-
-            # 문서가 없거나 value 리스트가 비어있는 경우 빈 리스트 반환
-            if document is None or not document.get("value", []):
-                return []
-
-            value_list = document.get("value", [])
-            
-            # index 기준으로 정렬
-            sorted_value_list = sorted(value_list, key = lambda x: x.get("index", 0))
-            
-            # 최신 10개만 선택 (마지막 10개)
-            latest_messages = sorted_value_list[-10:] if len(sorted_value_list) > 10 else sorted_value_list
-
-            # 대화 기록 변환 수행
-            formatted_chat_list = []
-            for chat in latest_messages:
-                formatted_chat = {
-                    "index": chat.get("index"),
-                    "img_url": chat.get("img_url"),
-                    "dialogue": (
-                        f"<|start_header_id|>user<|end_header_id|>\n"
-                        f"{chat.get('input_data')}<|eot_id|>"
-                        f"<|start_header_id|>assistant<|end_header_id|>\n"
-                        f"{chat.get('output_data')}<|eot_id|>"
-                    )
+                    "input_data": chat.get("input_data"),
+                    "output_data": chat.get("output_data")
                 }
                 formatted_chat_list.append(formatted_chat)
 
