Skip to content

Commit a29a422

Browse files
SAIC: fix(WDoc): correct formatting of final answer display
1 parent 1cf6f4b commit a29a422

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

WDoc/WDoc.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1753,7 +1753,10 @@ def source_replace(input: str) -> str:
17531753
md_printer(source_replace(to_print))
17541754

17551755
# print the final answer
1756-
md_printer(indent(f"# Answer:\n{source_replace(output['final_answer'])}\n", "> "))
1756+
fa = thinking_answer_parser(output["final_answer"])
1757+
fa = "### Thinking:\n" + fa["thinking"] + "\n\n" + "### Answer:\n" + fa["answer"]
1758+
fa = source_replace(fa)
1759+
md_printer(indent(f"# Answer:\n{fa}\n", "> "))
17571760

17581761
# print the breakdown of documents used and chain time
17591762
red(

0 commit comments

Comments
 (0)