miku-discord/bot/utils/context_manager.py

# utils/context_manager.py
"""
Structured context management for Miku's personality and knowledge.
Replaces the vector search system with organized, complete context.
Preserves original content files in their entirety.
"""

from utils.logger import get_logger

logger = get_logger('core')


def get_original_miku_lore() -> str:
    """Load the complete, unmodified miku_lore.txt file"""
    try:
        with open("miku_lore.txt", "r", encoding="utf-8") as f:
            return f.read()
    except Exception as e:
        logger.error(f"Failed to load miku_lore.txt: {e}")
        return "## MIKU LORE\n[File could not be loaded]"


def get_original_miku_prompt() -> str:
    """Load the complete, unmodified miku_prompt.txt file"""
    try:
        with open("miku_prompt.txt", "r", encoding="utf-8") as f:
            return f.read()
    except Exception as e:
        logger.error(f"Failed to load miku_prompt.txt: {e}")
        return "## MIKU PROMPT\n[File could not be loaded]"


def get_original_miku_lyrics() -> str:
    """Load the complete, unmodified miku_lyrics.txt file"""
    try:
        with open("miku_lyrics.txt", "r", encoding="utf-8") as f:
            return f.read()
    except Exception as e:
        logger.error(f"Failed to load miku_lyrics.txt: {e}")
        return "## MIKU LYRICS\n[File could not be loaded]"


def get_complete_context() -> str:
    """Returns all essential Miku context using original files in their entirety"""
    return f"""## MIKU LORE (Complete Original)
{get_original_miku_lore()}

## MIKU PERSONALITY & GUIDELINES (Complete Original) 
{get_original_miku_prompt()}

## MIKU SONG LYRICS (Complete Original)
{get_original_miku_lyrics()}"""


def get_context_for_response_type(response_type: str) -> str:
    """Returns appropriate context based on the type of response being generated"""
    
    # Core context always includes the complete original files
    core_context = f"""## MIKU LORE (Complete Original)
{get_original_miku_lore()}

## MIKU PERSONALITY & GUIDELINES (Complete Original) 
{get_original_miku_prompt()}"""
    
    if response_type == "autonomous_general":
        # For general autonomous messages, include everything
        return f"""{core_context}

## MIKU SONG LYRICS (Complete Original)
{get_original_miku_lyrics()}"""
        
    elif response_type == "autonomous_tweet":
        # For tweet responses, include lyrics for musical context
        return f"""{core_context}

## MIKU SONG LYRICS (Complete Original)
{get_original_miku_lyrics()}"""
        
    elif response_type == "dm_response" or response_type == "server_response":
        # For conversational responses, include everything
        return f"""{core_context}

## MIKU SONG LYRICS (Complete Original)
{get_original_miku_lyrics()}"""
        
    elif response_type == "conversation_join":
        # For joining conversations, include everything
        return f"""{core_context}

## MIKU SONG LYRICS (Complete Original)
{get_original_miku_lyrics()}"""

    elif response_type == "emoji_selection":
        # For emoji reactions, no context needed - the prompt has everything
        return ""
            
    else:
        # Default: comprehensive context
        return get_complete_context()