added formatting options

ServiceNow · Jan 28, 2025 · 2be23e5 · 2be23e5
1 parent 3fab5b4
commit 2be23e5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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()