first commit

tests_original_develop/verify_prompts.py

@@ -0,0 +1,53 @@
+import os
+from pathlib import Path
+
+from src.core.prompt_manager import get_prompt, _prompt_cache
+from src.core.config import settings
+import time
+
+def test_prompt_manager():
+    print("Starting Prompt Manager Verification...")
+    
+    local_p = "LOCAL PROMPT"
+    prompt_name = "test_prompt"
+    
+    # 1. Test Local Fallback (Langfuse Disabled)
+    settings.USE_LANGFUSE_PROMPTS = False
+    p = get_prompt(prompt_name, local_p)
+    assert p == local_p
+    print("PASS: Local fallback when USE_LANGFUSE_PROMPTS=False")
+    
+    # 2. Test Fetching (Langfuse Enabled) - This will likely fail or return fallback if not configured
+    settings.USE_LANGFUSE_PROMPTS = True
+    print("\nTesting with USE_LANGFUSE_PROMPTS=True...")
+    # We expect this to either work (if keys are valid and prompt exists) 
+    # or fall back gracefully (if keys are invalid or prompt doesn't exist)
+    p = get_prompt(prompt_name, local_p)
+    print(f"Result for '{prompt_name}': {p[:50]}...")
+    
+    # 3. Test Cache
+    print("\nTesting TTL Cache...")
+    # Manually inject into cache
+    _prompt_cache[prompt_name] = ("CACHED CONTENT", time.time())
+    
+    # TTL is 600 by default. This should return cached content.
+    p = get_prompt(prompt_name, local_p)
+    assert p == "CACHED CONTENT"
+    print("PASS: Cache hit before TTL expiry")
+    
+    # Test TTL Expiry
+    print("\nTesting Cache Expiry...")
+    _prompt_cache[prompt_name] = ("EXPIRED CONTENT", time.time() - 1000)
+    # Should attempt to fetch (and likely fall back to local if test_prompt doesn't exist in Langfuse)
+    p = get_prompt(prompt_name, local_p)
+    # If fetch fails, it returns local_p. If it hits Langfuse, it returns that.
+    # The important part is that it didn't return "EXPIRED CONTENT".
+    assert p != "EXPIRED CONTENT"
+    print(f"PASS: Cache expiry handled (Returned: {p})")
+
+if __name__ == "__main__":
+    try:
+        test_prompt_manager()
+        print("\nAll logical checks passed!")
+    except Exception as e:
+        print(f"\nVerification FAILED: {e}")