merge1

utils/deepseek_client.py

@@ -5,6 +5,7 @@ DeepSeek API 클라이언트 모듈
 import os
 import json
 import logging
+import traceback
 from typing import List, Dict, Any, Optional, Union
 from dotenv import load_dotenv
 import requests
@@ -82,22 +83,42 @@ class DeepSeekLLM:
             
             # API 요청 보내기
             endpoint = f"{self.api_base}/v1/chat/completions"
+            
+            # 디버깅: API 요청 데이터 로깅 (민감 정보 제외)
+            debug_data = data.copy()
+            debug_data["messages"] = f"[{len(data['messages'])}개 메시지]"
+            logger.debug(f"DeepSeek API 요청 데이터: {json.dumps(debug_data)}")
+            
             response = requests.post(
                 endpoint,
                 headers=headers,
-                json=data
+                json=data,
+                timeout=30  # 30초 타임아웃 설정
             )
             
-            # 응답 검증
-            response.raise_for_status()
-            return response.json()
+            # 응답 상태 코드 확인
+            if not response.ok:
+                logger.error(f"DeepSeek API 오류: 상태 코드 {response.status_code}")
+                logger.error(f"응답 내용: {response.text}")
+                return {"error": f"API 오류: 상태 코드 {response.status_code}", "detail": response.text}
+            
+            # 응답 파싱
+            try:
+                result = response.json()
+                logger.debug(f"API 응답 구조: {list(result.keys())}")
+                return result
+            except json.JSONDecodeError as e:
+                logger.error(f"DeepSeek API JSON 파싱 실패: {e}")
+                logger.error(f"원본 응답: {response.text[:500]}...")
+                return {"error": "API 응답을 파싱할 수 없습니다", "detail": str(e)}
                 
         except requests.exceptions.RequestException as e:
             logger.error(f"DeepSeek API 요청 실패: {e}")
-            raise Exception(f"DeepSeek API 요청 실패: {e}")
-        except json.JSONDecodeError as e:
-            logger.error(f"DeepSeek API 응답 파싱 실패: {e}")
-            raise Exception(f"DeepSeek API 응답 파싱 실패: {e}")
+            return {"error": f"API 요청 실패: {str(e)}"}
+        except Exception as e:
+            logger.error(f"DeepSeek API 호출 중 예상치 못한 오류: {e}")
+            logger.error(traceback.format_exc())
+            return {"error": f"예상치 못한 오류: {str(e)}"}
     
     def generate(
         self, 
@@ -135,16 +156,31 @@ class DeepSeekLLM:
                 **kwargs
             )
             
-            # 응답 검증 및 처리
-            if not response or 'choices' not in response or not response['choices']:
-                logger.error("DeepSeek API 응답에서 생성된 텍스트를 찾을 수 없습니다.")
-                return ""
+            # 오류 응답 확인
+            if "error" in response:
+                logger.error(f"텍스트 생성 중 API 오류: {response['error']}")
+                error_detail = response.get("detail", "")
+                return f"API 오류: {response['error']} {error_detail}"
+            
+            # 응답 형식 검증
+            if 'choices' not in response or not response['choices']:
+                logger.error(f"API 응답에 'choices' 필드가 없습니다: {response}")
+                return "응답 형식 오류: 생성된 텍스트를 찾을 수 없습니다."
             
-            return response['choices'][0]['message']['content'].strip()
+            # 메시지 컨텐츠 확인
+            choice = response['choices'][0]
+            if 'message' not in choice or 'content' not in choice['message']:
+                logger.error(f"API 응답에 예상 필드가 없습니다: {choice}")
+                return "응답 형식 오류: 메시지 내용을 찾을 수 없습니다."
+            
+            generated_text = choice['message']['content'].strip()
+            logger.info(f"텍스트 생성 완료 (길이: {len(generated_text)})")
+            return generated_text
             
         except Exception as e:
-            logger.error(f"텍스트 생성 중 오류 발생: {e}")
-            return f"오류: {str(e)}"
+            logger.error(f"텍스트 생성 중 예외 발생: {e}")
+            logger.error(traceback.format_exc())
+            return f"오류 발생: {str(e)}"
     
     def rag_generate(
         self, 
@@ -177,39 +213,51 @@ class DeepSeekLLM:
 - 검색 내용을 그대로 복사하지 말고, 자연스러운 한국어로 답변을 작성하세요.
 - 답변은 간결하고 정확하게 제공하세요."""
         
-        # 중요: 컨텍스트 길이 제한
-        max_context = 10
-        if len(context) > max_context:
-            logger.warning(f"컨텍스트가 너무 길어 처음 {max_context}개만 사용합니다.")
-            context = context[:max_context]
-        
-        # 각 컨텍스트 액세스
-        limited_context = []
-        for i, doc in enumerate(context):
-            # 각 문서를 1000자로 제한
-            if len(doc) > 1000:
-                logger.warning(f"문서 {i+1}의 길이가 제한되었습니다 ({len(doc)} -> 1000)")
-                doc = doc[:1000] + "...(생략)"
-            limited_context.append(doc)
-        
-        context_text = "\n\n".join([f"문서 {i+1}: {doc}" for i, doc in enumerate(limited_context)])
-        
-        prompt = f"""질문: {query}
+        try:
+            # 중요: 컨텍스트 길이 제한
+            max_context = 10
+            if len(context) > max_context:
+                logger.warning(f"컨텍스트가 너무 길어 처음 {max_context}개만 사용합니다.")
+                context = context[:max_context]
+            
+            # 각 컨텍스트 액세스
+            limited_context = []
+            for i, doc in enumerate(context):
+                # 각 문서를 1000자로 제한
+                if len(doc) > 1000:
+                    logger.warning(f"문서 {i+1}의 길이가 제한되었습니다 ({len(doc)} -> 1000)")
+                    doc = doc[:1000] + "...(생략)"
+                limited_context.append(doc)
+            
+            context_text = "\n\n".join([f"문서 {i+1}: {doc}" for i, doc in enumerate(limited_context)])
+            
+            prompt = f"""질문: {query}
 
 <context>
 {context_text}
 </context>
 
 위 검색 결과를 참고하여 질문에 답변해 주세요."""
-        
-        try:
-            return self.generate(
+            
+            logger.info(f"RAG 프롬프트 생성 완료 (길이: {len(prompt)})")
+            
+            result = self.generate(
                 prompt=prompt,
                 system_prompt=system_prompt,
                 temperature=temperature,
                 max_tokens=max_tokens,
                 **kwargs
             )
+            
+            # 결과가 오류 메시지인지 확인
+            if result.startswith("오류") or result.startswith("API 오류") or result.startswith("응답 형식 오류"):
+                logger.error(f"RAG 생성 결과가 오류를 포함합니다: {result}")
+                # 좀 더 사용자 친화적인 오류 메시지 반환
+                return "죄송합니다. 현재 응답을 생성하는데 문제가 발생했습니다. 잠시 후 다시 시도해주세요."
+            
+            return result
+            
         except Exception as e:
-            logger.error(f"RAG 텍스트 생성 중 오류 발생: {e}")
-            return f"오류: {str(e)}"
+            logger.error(f"RAG 텍스트 생성 중 예외 발생: {str(e)}")
+            logger.error(traceback.format_exc())
+            return "죄송합니다. 응답 생성 중 오류가 발생했습니다. 잠시 후 다시 시도해주세요."