diff --git a/src/agentlab/analyze/error_analysis/pipeline.py b/src/agentlab/analyze/error_analysis/pipeline.py index 62c313aa..4330cce2 100644 --- a/src/agentlab/analyze/error_analysis/pipeline.py +++ b/src/agentlab/analyze/error_analysis/pipeline.py @@ -52,6 +52,10 @@ def save_analysis(self, exp_result: ExpResult, error_analysis: dict, exists_ok=T json.dump(error_analysis, f, indent=4) +AXTREE_FORMATTER = lambda x: x.get("axtree_txt", "No AXTREE available") +HTML_FORMATTER = lambda x: x.get("pruned_html", "No HTML available") + + if __name__ == "__main__": import argparse @@ -73,7 +77,7 @@ def save_analysis(self, exp_result: ExpResult, error_analysis: dict, exists_ok=T pipeline = ErrorAnalysisPipeline( exp_dir=exp_dir, filter=filter, - episode_summarizer=EpisodeErrorSummarizer(ChangeSummarizer(llm), llm), + episode_summarizer=EpisodeErrorSummarizer(ChangeSummarizer(llm, HTML_FORMATTER), llm), ) pipeline.run_analysis()