@@ -154,8 +154,10 @@ def optimize_retrieval(db_path: Path, model: str, output_path: Path):
154154 # Evaluate
155155 print ("\n --- Evaluation ---" )
156156 evaluate = dspy .Evaluate (devset = devset , metric = retrieval_metric , num_threads = 2 )
157- baseline_score = evaluate (retrieval_module )
158- optimized_score = evaluate (optimized_retrieval )
157+ baseline_result = evaluate (retrieval_module )
158+ optimized_result = evaluate (optimized_retrieval )
159+ baseline_score = float (getattr (baseline_result , 'score' , baseline_result ) or 0 )
160+ optimized_score = float (getattr (optimized_result , 'score' , optimized_result ) or 0 )
159161 print (f"Baseline: { baseline_score :.3f} " )
160162 print (f"Optimized: { optimized_score :.3f} " )
161163 print (f"Delta: { optimized_score - baseline_score :+.3f} " )
@@ -177,8 +179,10 @@ def optimize_retrieval(db_path: Path, model: str, output_path: Path):
177179 complexity_eval = dspy .Evaluate (
178180 devset = complexity_examples [6 :], metric = complexity_metric , num_threads = 2
179181 )
180- complexity_baseline = complexity_eval (complexity_module )
181- complexity_optimized = complexity_eval (optimized_complexity )
182+ complexity_baseline_result = complexity_eval (complexity_module )
183+ complexity_optimized_result = complexity_eval (optimized_complexity )
184+ complexity_baseline = float (getattr (complexity_baseline_result , 'score' , complexity_baseline_result ) or 0 )
185+ complexity_optimized = float (getattr (complexity_optimized_result , 'score' , complexity_optimized_result ) or 0 )
182186 print (f"Baseline: { complexity_baseline :.3f} " )
183187 print (f"Optimized: { complexity_optimized :.3f} " )
184188
0 commit comments