mirror of
https://github.com/hwanny1128/HGZero.git
synced 2025-12-06 07:56:24 +00:00
9c2d8dc9b2
- ClaudeService에 analyze_suggestions 메서드 추가 - 개선된 제안사항 추출 프롬프트 생성 (구체적이고 실행 가능한 제안사항) - API 경로 수정: /api/v1/ai/suggestions → /api/ai/suggestions - 프론트엔드 HTML API 경로 업데이트 (v1 제거) - RealtimeSuggestionsResponse 모델 export 추가 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
140 lines
4.4 KiB
Python
140 lines
4.4 KiB
Python
"""Claude API Service"""
|
|
import anthropic
|
|
import json
|
|
import logging
|
|
from typing import Dict, Any
|
|
from app.config import get_settings
|
|
|
|
logger = logging.getLogger(__name__)
|
|
settings = get_settings()
|
|
|
|
|
|
class ClaudeService:
|
|
"""Claude API 호출 서비스"""
|
|
|
|
def __init__(self):
|
|
self.client = anthropic.Anthropic(api_key=settings.claude_api_key)
|
|
self.model = settings.claude_model
|
|
self.max_tokens = settings.claude_max_tokens
|
|
self.temperature = settings.claude_temperature
|
|
|
|
async def generate_completion(
|
|
self,
|
|
prompt: str,
|
|
system_prompt: str = None
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Claude API 호출하여 응답 생성
|
|
|
|
Args:
|
|
prompt: 사용자 프롬프트
|
|
system_prompt: 시스템 프롬프트 (선택)
|
|
|
|
Returns:
|
|
Claude API 응답 (JSON 파싱)
|
|
"""
|
|
try:
|
|
# 메시지 구성
|
|
messages = [
|
|
{
|
|
"role": "user",
|
|
"content": prompt
|
|
}
|
|
]
|
|
|
|
# API 호출
|
|
logger.info(f"Claude API 호출 시작 - Model: {self.model}")
|
|
|
|
if system_prompt:
|
|
response = self.client.messages.create(
|
|
model=self.model,
|
|
max_tokens=self.max_tokens,
|
|
temperature=self.temperature,
|
|
system=system_prompt,
|
|
messages=messages
|
|
)
|
|
else:
|
|
response = self.client.messages.create(
|
|
model=self.model,
|
|
max_tokens=self.max_tokens,
|
|
temperature=self.temperature,
|
|
messages=messages
|
|
)
|
|
|
|
# 응답 텍스트 추출
|
|
response_text = response.content[0].text
|
|
logger.info(f"Claude API 응답 수신 완료 - Tokens: {response.usage.input_tokens + response.usage.output_tokens}")
|
|
|
|
# JSON 파싱
|
|
# ```json ... ``` 블록 제거
|
|
if "```json" in response_text:
|
|
response_text = response_text.split("```json")[1].split("```")[0].strip()
|
|
elif "```" in response_text:
|
|
response_text = response_text.split("```")[1].split("```")[0].strip()
|
|
|
|
result = json.loads(response_text)
|
|
|
|
return result
|
|
|
|
except json.JSONDecodeError as e:
|
|
logger.error(f"JSON 파싱 실패: {e}")
|
|
logger.error(f"응답 텍스트: {response_text[:500]}...")
|
|
raise ValueError(f"Claude API 응답을 JSON으로 파싱할 수 없습니다: {str(e)}")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Claude API 호출 실패: {e}")
|
|
raise
|
|
|
|
async def analyze_suggestions(self, transcript_text: str):
|
|
"""
|
|
회의 텍스트에서 AI 제안사항 추출
|
|
|
|
Args:
|
|
transcript_text: 회의 텍스트
|
|
|
|
Returns:
|
|
RealtimeSuggestionsResponse 객체
|
|
"""
|
|
from app.models import RealtimeSuggestionsResponse, SimpleSuggestion
|
|
from app.prompts.suggestions_prompt import get_suggestions_prompt
|
|
from datetime import datetime
|
|
import uuid
|
|
|
|
try:
|
|
# 프롬프트 생성
|
|
system_prompt, user_prompt = get_suggestions_prompt(transcript_text)
|
|
|
|
# Claude API 호출
|
|
result = await self.generate_completion(
|
|
prompt=user_prompt,
|
|
system_prompt=system_prompt
|
|
)
|
|
|
|
# 응답 파싱
|
|
suggestions_data = result.get("suggestions", [])
|
|
|
|
# SimpleSuggestion 객체로 변환
|
|
suggestions = [
|
|
SimpleSuggestion(
|
|
id=str(uuid.uuid4()),
|
|
content=s["content"],
|
|
timestamp=datetime.now().strftime("%H:%M:%S"),
|
|
confidence=s.get("confidence", 0.85)
|
|
)
|
|
for s in suggestions_data
|
|
if s.get("confidence", 0) >= 0.7 # 신뢰도 0.7 이상만
|
|
]
|
|
|
|
logger.info(f"AI 제안사항 {len(suggestions)}개 추출 완료")
|
|
|
|
return RealtimeSuggestionsResponse(suggestions=suggestions)
|
|
|
|
except Exception as e:
|
|
logger.error(f"제안사항 분석 실패: {e}", exc_info=True)
|
|
# 빈 응답 반환
|
|
return RealtimeSuggestionsResponse(suggestions=[])
|
|
|
|
|
|
# 싱글톤 인스턴스
|
|
claude_service = ClaudeService()
|