Amazon Bedrock + Lambda로 문서 요약해 S3에 저장하기

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": [
                "bedrock:InvokeModel",
                "bedrock:InvokeModelWithResponseStream"
            ],
            "Resource": "*"
        },
        {
            "Effect": "Allow",
            "Action": [
                "s3:ListBucket"
            ],
            "Resource": "arn:aws:s3:::S3 버킷명"
        },
        {
            "Effect": "Allow",
            "Action": [
                "s3:GetObject",
                "s3:PutObject"
            ],
            "Resource": "arn:aws:s3:::S3 버킷명/*"
        }
    ]
}

# ABC Tech 사내 운영 매뉴얼 (Internal Policy v1.0)

## 1. 회사 기본 정보

* 회사명: ABC Tech
* 설립연도: 2021년
* 주요 서비스: SaaS 기반 업무 관리 플랫폼

---

## 2. 근무 시간 정책

* 기본 근무 시간: 오전 9시 ~ 오후 6시
* 점심 시간: 12시 ~ 1시
* 유연 근무제:

  * 출근: 오전 8시 ~ 10시 사이 선택 가능
  * 퇴근: 출근 시간 기준 9시간 후

---

## 3. 휴가 정책

* 연차 휴가: 연 15일
* 반차: 0.5일 단위 사용 가능
* 병가:

  * 연 최대 10일
  * 3일 이상 시 진단서 필요

---

## 4. 재택근무 정책

* 주 2회까지 가능
* 사전 승인 필수
* 긴급 상황 시 추가 허용 가능

---

## 5. 환불 정책 (고객 대상)

* 구매 후 7일 이내: 100% 환불 가능
* 7일 이후: 환불 불가
* 단, 서비스 장애 발생 시 예외적으로 환불 가능

---

## 6. 보안 정책

* 모든 직원은 2단계 인증(2FA)을 사용해야 함
* 외부 저장장치 사용 금지
* 회사 데이터는 승인된 클라우드에만 저장

---

## 7. IT 자원 사용 정책

* 개인 용도의 과도한 사용 금지
* 불법 소프트웨어 설치 금지
* 회사 계정 공유 금지

---

## 8. 문의 및 지원

* IT 지원: [it-support@abctech.com](mailto:it-support@abctech.com)
* 인사팀: [hr@abctech.com](mailto:hr@abctech.com)

---

## 9. 징계 규정

* 보안 위반: 1차 경고, 2차 정직
* 무단 결근: 3회 시 해고 가능

---

## 10. 기타

* 본 정책은 2026년 1월 1일부터 적용
* 회사 정책은 필요 시 변경될 수 있음

import boto3
import json
import uuid
from typing import List

bedrock = boto3.client("bedrock-runtime", region_name="ap-northeast-1")
s3 = boto3.client("s3")

BUCKET_NAME = "YOUR_BUCKET_NAME"
CLAUDE_MODEL_ID = "jp.anthropic.claude-sonnet-4-6"

MAX_CHARS_PER_DOC = 2000
TOP_K = 3

def embed_text(text: str) -> List[float]:
    return [float(ord(c)) for c in text[:128]]

def cosine_similarity(v1: List[float], v2: List[float]) -> float:
    dot = sum(a*b for a,b in zip(v1,v2))
    norm1 = sum(a*a for a in v1) ** 0.5
    norm2 = sum(b*b for b in v2) ** 0.5
    return dot / (norm1*norm2 + 1e-10)

def read_s3_object(bucket: str, key: str) -> str:
    resp = s3.get_object(Bucket=bucket, Key=key)
    body = resp["Body"].read()
    for enc in ["utf-8", "cp949", "latin1"]:
        try:
            return body.decode(enc)
        except UnicodeDecodeError:
            continue
    return body.decode("utf-8", errors="ignore")

def load_documents_from_s3() -> List[dict]:
    prefixes = ["documents/", ""]
    docs = []
    for prefix in prefixes:
        objects = s3.list_objects_v2(Bucket=BUCKET_NAME, Prefix=prefix)
        for obj in objects.get("Contents", []):
            key = obj["Key"]
            if key.endswith("/"):
                continue
            content = read_s3_object(BUCKET_NAME, key)
            docs.append({"key": key, "text": content, "embedding": embed_text(content)})
    return docs

def lambda_handler(event, context):
    session_id = str(uuid.uuid4())
    user_question = event.get("message", "이 문서를 3줄로 요약해줘")

    try:
        documents = load_documents_from_s3()
        if not documents:
            return {
                "statusCode": 400,
                "body": json.dumps({
                    "error": "S3에서 문서를 찾을 수 없습니다. documents/ 폴더 또는 루트에 텍스트 파일을 업로드하세요."
                }, ensure_ascii=False)
            }

        question_vec = embed_text(user_question)

        documents_sorted = sorted(documents, key=lambda d: cosine_similarity(d["embedding"], question_vec), reverse=True)
        top_docs_text = "\n\n".join([d["text"][:MAX_CHARS_PER_DOC] for d in documents_sorted[:TOP_K]])
        source_keys = [d["key"] for d in documents_sorted[:TOP_K]]

        prompt = f"다음 문서를 읽고 질문에 답해 주세요:\n\n{top_docs_text}\n\n질문: {user_question}\n답변:"
        request_body = {
            "anthropic_version": "bedrock-2023-05-31",
            "max_tokens": 1024,
            "system": "당신은 한국어로만 답변하는 AI 어시스턴트입니다. 항상 한국어로 답변하세요.",
            "messages": [{"role": "user", "content": prompt}]
        }

        response = bedrock.invoke_model(
            modelId=CLAUDE_MODEL_ID,
            body=json.dumps(request_body)
        )

        response_body = json.loads(response["body"].read())
        answer = response_body["content"][0]["text"]

        object_key = f"bedrock-results/{session_id}.json"
        s3.put_object(
            Bucket=BUCKET_NAME,
            Key=object_key,
            Body=json.dumps({
                "question": user_question,
                "answer": answer,
                "source_docs": source_keys
            }, ensure_ascii=False)
        )

        return {
            "statusCode": 200,
            "body": json.dumps({
                "message": "Claude 기반 RAG 결과 S3에 저장 완료",
                "s3_object_key": object_key,
                "answer": answer,
                "source_docs": source_keys
            }, ensure_ascii=False)
        }

    except Exception as e:
        return {
            "statusCode": 500,
            "body": json.dumps({"error": str(e)})
        }

Response:
{
  "statusCode": 200,
  "body": "{\"message\": \"Claude 기반 RAG 결과 S3에 저장 완료\", \"s3_object_key\": \"bedrock-results/xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxx.json\", \"answer\": \"## 문서 요약\\n\\n1. **ABC Tech**는 2021년 설립된 SaaS 기반 업무 관리 플랫폼 회사로, 유연 근무제·재택근무(주 2회)·연차 15일 등 직원 친화적 근무 환경을 운영합니다.\\n\\n2. **보안 및 IT 정책**으로 2단계 인증(2FA) 의무화, 외부 저장장치 사용 금지, 불법 소프트웨어 설치 금지 등 엄격한 내부 규정을 적용합니다.\\n\\n3. **고객 환불**은 구매 후 7일 이내에만 가능하며, 보안 위반이나 무단 결근 등 규정 위반 시 경고·정직·해고 등의 징계가 적용되고, 본 정책은 **2026년 1월 1일부터 시행**됩니다.\", \"source_docs\": [\"example.txt\"]}"
}

{"question": "이 문서를 3줄로 요약해줘", "answer": "## 문서 요약\n\n1. **ABC Tech**는 2021년 설립된 SaaS 기반 업무 관리 플랫폼 회사로, 유연 근무제·재택근무(주 2회)·연차 15일 등의 근무 정책을 운영합니다.\n\n2. **보안 및 IT 정책**으로 2단계 인증(2FA) 필수, 외부 저장장치 사용 금지, 불법 소프트웨어 설치 금지 등을 규정하고 있으며, 위반 시 경고·정직 등의 징계가 부과됩니다.\n\n3. **고객 환불 정책**은 구매 후 7일 이내 100% 환불 가능하며, 본 정책 전반은 **2026년 1월 1일부터 적용**됩니다.", "source_docs": ["example.txt"]}