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.

When LANGUAGE_MODE is "japanese", appends a Japanese instruction to ensure
all responses are in Japanese without requiring separate files.
"""

import globals
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_japanese_instruction() -> str:
    """
    Returns the Japanese language instruction to append to context.
    Ensures all responses are in Japanese when in Japanese mode.
    """
    return "\n\n[CRITICAL INSTRUCTION - 重要な指示]\n**YOU MUST RESPOND ENTIRELY IN JAPANESE (日本語). NO ENGLISH ALLOWED.**\nすべての返答は必ず日本語で行ってください。英語での返答は一切禁止されています。\nこれは最優先の指示です。必ず守ってください。"


def get_complete_context() -> str:
    """
    Returns all essential Miku context using original files in their entirety.
    
    If LANGUAGE_MODE is "japanese", appends a Japanese instruction to ensure
    all responses are in Japanese.
    """
    lore = get_original_miku_lore()
    prompt = get_original_miku_prompt()
    lyrics = get_original_miku_lyrics()
    
    combined = f"""## MIKU LORE (Complete Original)
{lore}

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

## MIKU SONG LYRICS (Complete Original)
{lyrics}"""
    
    # Append Japanese instruction if in Japanese mode
    if globals.LANGUAGE_MODE == "japanese":
        combined += _get_japanese_instruction()
    
    logger.info(f"[core] Context loaded in {globals.LANGUAGE_MODE} mode")
    return combined


def get_context_for_response_type(response_type: str) -> str:
    """
    Returns appropriate context based on the type of response being generated.
    
    If LANGUAGE_MODE is "japanese", appends Japanese instruction to all contexts
    to ensure responses are in Japanese.
    """
    
    lore = get_original_miku_lore()
    prompt = get_original_miku_prompt()
    lyrics = get_original_miku_lyrics()
    
    # Build core context (always in English source files)
    core_context = f"""## MIKU LORE (Complete Original)
{lore}

## MIKU PERSONALITY & GUIDELINES (Complete Original) 
{prompt}"""
    
    # Return context based on response type
    if response_type == "autonomous_general":
        context = f"""{core_context}

## MIKU SONG LYRICS (Complete Original)
{lyrics}"""
        
    elif response_type == "autonomous_tweet":
        context = f"""{core_context}

## MIKU SONG LYRICS (Complete Original)
{lyrics}"""
        
    elif response_type == "dm_response" or response_type == "server_response":
        context = f"""{core_context}

## MIKU SONG LYRICS (Complete Original)
{lyrics}"""
        
    elif response_type == "conversation_join":
        context = f"""{core_context}

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

    elif response_type == "emoji_selection":
        # For emoji reactions, minimal context needed
        context = ""
            
    else:
        # Default: comprehensive context
        context = get_complete_context()
    
    # Append Japanese instruction if in Japanese mode
    if globals.LANGUAGE_MODE == "japanese" and context:
        context += _get_japanese_instruction()
    
    return context