app\prompts\graph_extraction.py

"""

Graph Extraction 프롬프트

엔티티(인물/장소)와 관계(사건)를 추출하는 GraphRAG 기반 프롬프트

"""

from typing import Optional


def get_graph_extraction_prompt(

    episode_content: str,

    episode_title: str,

    full_content: Optional[str] = None,

    parent_chunk_info: Optional[str] = None,

    max_length: int = 10000

) -> str:
    """

    Graph Extraction을 위한 프롬프트 생성

    

    Args:

        episode_content: 분석할 회차 내용

        episode_title: 회차 제목 (예: '1화', '2화')

        full_content: 원본 웹소설 전체 내용 (참고용)

        parent_chunk_info: Parent Chunk 정보 (선택사항)

        max_length: 프롬프트에 포함할 최대 텍스트 길이

    

    Returns:

        프롬프트 문자열

    """
    # 회차 내용 길이 제한
    content_preview = episode_content[:max_length]
    is_truncated = len(episode_content) > max_length
    
    truncation_note = "\n(참고: 회차 내용이 길어 일부만 사용되었습니다.)" if is_truncated else ""
    
    # 전체 내용 참고용 (선택사항)
    full_content_preview = ""
    if full_content:
        # 전체 내용이 너무 길면 앞부분과 뒷부분 일부만 사용
        if len(full_content) > 30000:
            full_content_preview = full_content[:15000] + "\n... (중간 생략) ...\n" + full_content[-15000:]
        else:
            full_content_preview = full_content
    
    prompt = f"""다음 웹소설의 {episode_title} 회차에서 엔티티(인물/장소)와 관계(사건)를 추출해주세요.



{parent_chunk_info if parent_chunk_info else ""}



원본 웹소설 전체 내용 (참고용):

{full_content_preview[:50000] if full_content_preview else "없음"}



분석할 회차 내용 ({episode_title}):

{content_preview}{truncation_note}



다음 형식으로 JSON 형식으로만 응답하세요:



{{

    "entities": {{

        "characters": [

            {{

                "name": "인물 이름",

                "type": "인물",

                "description": "인물에 대한 간단한 설명",

                "role": "이 회차에서의 역할 (예: 주인공, 조연, 악역 등)"

            }}

        ],

        "locations": [

            {{

                "name": "장소 이름",

                "type": "장소",

                "description": "장소에 대한 간단한 설명",

                "category": "장소 유형 (예: 도시, 건물, 차원 등)"

            }}

        ]

    }},

    "relationships": [

        {{

            "source": "관계의 주체 (인물 이름)",

            "target": "관계의 대상 (인물 이름 또는 장소 이름)",

            "type": "관계 유형 (예: 친구, 적, 연인, 거주지, 방문지 등)",

            "description": "관계에 대한 상세 설명",

            "event": "이 관계를 형성하거나 변화시킨 사건 (있는 경우)"

        }}

    ],

    "events": [

        {{

            "name": "사건 이름",

            "description": "사건에 대한 상세 설명",

            "participants": ["관련 인물1", "관련 인물2"],

            "location": "사건이 발생한 장소",

            "significance": "사건의 중요도 (예: 주요 사건, 부수 사건 등)"

        }}

    ]

}}



중요 사항:

1. 엔티티는 이 회차에서 실제로 등장하거나 언급된 인물과 장소만 추출하세요.

2. 관계는 이 회차에서 새로 형성되거나 변화한 관계를 중심으로 추출하세요.

3. 사건은 이 회차에서 일어난 구체적인 사건들을 추출하세요.

4. 응답은 오직 JSON 형식만 사용하고, 다른 설명이나 마크다운은 포함하지 마세요.

5. JSON 형식이 올바른지 반드시 확인하세요 (따옴표 이스케이프 등).

6. 배열이 비어있을 경우 빈 배열 []로 표시하세요.

7. 필드 값이 없는 경우 null 대신 빈 문자열 "" 또는 빈 배열 []을 사용하세요."""
    
    return prompt