🏭 스마트 설비 운영 지원 플랫폼

import json
import boto3
 
# Bedrock 클라이언트 (리전 확인)
bedrock = boto3.client("bedrock-runtime", region_name="ap-northeast-2")
 
# 설비 정비 컨텍스트 (실제로는 S3/DB에서 로드)
MAINTENANCE_CONTEXT = """
당신은 대한설비(주)의 AI 설비 정비 도우미입니다.
아래 정비 매뉴얼을 참고하여 현장 근무자의 질문에 답변하세요.
 
[정비 매뉴얼 요약]
 
1. 컨베이어 벨트 이상 소음:
   - 원인: 롤러 베어링 마모, 벨트 장력 불균형, 이물질 끼임
   - 조치: 베어링 교체 주기 확인(6개월), 벨트 장력 조절, 이물질 제거
   - 긴급도: 소음 크기에 따라 LOW~HIGH
 
2. 프레스기 진동 과다:
   - 원인: 볼트 풀림, 기초 불량, 금형 마모
   - 조치: 볼트 재조임(토크 확인), 기초 재시공, 금형 교체
   - 긴급도: 진동값 2mm/s 이상이면 CRITICAL
 
3. 냉각 시스템 온도 이상:
   - 원인: 냉각수 부족, 펌프 고장, 필터 막힘, 열교환기 스케일
   - 조치: 냉각수 보충, 펌프 점검, 필터 교체(월 1회), 열교환기 세척(분기 1회)
   - 긴급도: 40도 이상이면 CRITICAL, 즉시 가동 중단
 
4. 용접 로봇 정밀도 저하:
   - 원인: 서보 모터 마모, 케이블 손상, 캘리브레이션 틀어짐
   - 조치: 모터 교체, 케이블 교체, 재캘리브레이션(월 1회)
   - 긴급도: 불량률 1% 이상이면 HIGH
 
답변 시 아래 규칙을 따르세요:
- 한국어로 답변합니다
- 200자 이내로 간결하게 답변합니다
- 가능한 원인, 권장 조치, 긴급도를 포함합니다
- 매뉴얼에 없는 내용은 "매뉴얼에서 확인되지 않은 항목입니다. 설비팀에 문의하세요."로 답합니다
"""
 
def lambda_handler(event, context):
    try:
        body = json.loads(event.get("body", "{}"))
        question = body.get("question", "")
 
        if not question:
            return {
                "statusCode": 400,
                "headers": {"Content-Type": "application/json", "Access-Control-Allow-Origin": "*"},
                "body": json.dumps({"error": "질문을 입력해 주세요"})
            }
 
        # Bedrock Claude 호출
        response = bedrock.invoke_model(
            modelId="anthropic.claude-3-haiku-20240307-v1:0",
            contentType="application/json",
            accept="application/json",
            body=json.dumps({
                "anthropic_version": "bedrock-2023-05-31",
                "max_tokens": 500,
                "system": MAINTENANCE_CONTEXT,
                "messages": [
                    {"role": "user", "content": question}
                ]
            })
        )
 
        result = json.loads(response["body"].read())
        answer = result["content"][0]["text"]
 
        return {
            "statusCode": 200,
            "headers": {"Content-Type": "application/json", "Access-Control-Allow-Origin": "*"},
            "body": json.dumps({
                "question": question,
                "answer": answer,
                "model": "Claude 3 Haiku"
            })
        }
 
    except Exception as e:
        return {
            "statusCode": 500,
            "headers": {"Content-Type": "application/json", "Access-Control-Allow-Origin": "*"},
            "body": json.dumps({"error": f"AI 응답 오류: {str(e)}"})
        }

API_URL="https://xxxxxxxxxx.execute-api.ap-northeast-2.amazonaws.com"
 
# 질문 1: 컨베이어 관련
curl -X POST "$API_URL/ai/ask" \
  -H "Content-Type: application/json" \
  -d '{"question": "1라인 컨베이어에서 이상한 소리가 나는데 원인이 뭘까요?"}'
 
# 질문 2: 냉각 시스템 관련
curl -X POST "$API_URL/ai/ask" \
  -H "Content-Type: application/json" \
  -d '{"question": "냉각수 온도가 42도입니다. 어떻게 해야 하나요?"}'
 
# 질문 3: 매뉴얼에 없는 질문
curl -X POST "$API_URL/ai/ask" \
  -H "Content-Type: application/json" \
  -d '{"question": "엘리베이터 점검 주기가 어떻게 되나요?"}'