Initial commit: Miku Discord Bot

bot/utils/context_manager.py

@@ -0,0 +1,94 @@
+# 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.
+"""
+
+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:
+        print(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:
+        print(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:
+        print(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()