mirror of
https://github.com/hoshikawa2/compass_backoffice.git
synced 2026-07-09 22:04:20 +00:00
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
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}")
|