added formatting options

Author: TLSDC

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()