ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter3/contextual-retrieval-for-user-memory/test_quick.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python3
+"""Quick test to verify the fixes work"""
+
+from config import Config
+from contextual_evaluator import ContextualMemoryEvaluator
+
+# Test configuration loading
+config = Config.from_env()
+print(f"✓ Config loaded successfully")
+print(f"  - EvaluationConfig.use_llm_judge: {config.evaluation.use_llm_judge}")
+print(f"  - AgentConfig.verbose: {config.agent.verbose}")
+
+# Test evaluator initialization
+evaluator = ContextualMemoryEvaluator(config)
+print(f"✓ Evaluator initialized successfully")
+
+# Load test cases
+test_cases = evaluator.load_test_cases("layer1")
+print(f"✓ Loaded {len(test_cases)} test cases")
+
+if test_cases:
+    test_id = test_cases[0]
+    print(f"\nWill test with: {test_id}")
+    print("Attempting evaluation...")
+    
+    try:
+        result = evaluator.evaluate_test_case(test_id)
+        print(f"✓ Evaluation completed!")
+        print(f"  Success: {result.success}")
+        print(f"  Iterations: {result.iterations}")
+        print(f"  Processing time: {result.processing_time:.2f}s")
+    except Exception as e:
+        print(f"✗ Evaluation failed: {e}")
+        import traceback
+        traceback.print_exc()
+else:
+    print("No test cases found")