Configure and enable final report generation from journal.

aide/journal2report.py

@@ -1,8 +1,8 @@
-from backend import query
-from journal import Journal
+from .backend import query
+from .journal import Journal
+from .utils.config import StageConfig
 
-
-def journal2report(journal: Journal, task_desc: dict):
+def journal2report(journal: Journal, task_desc: dict, rcfg: StageConfig):
     """
     Generate a report from a journal, the report will be in markdown format.
     """
@@ -24,6 +24,7 @@ def journal2report(journal: Journal, task_desc: dict):
     return query(
         system_message=system_prompt_dict,
         user_message=context_prompt,
-        model="gpt-4-turbo-preview",
+        model=rcfg.model,
+        temperature=rcfg.temp,
         max_tokens=4096,
     )

aide/run.py

@@ -8,6 +8,7 @@ from .utils import tree_export
 from .agent import Agent
 from .interpreter import Interpreter
 from .journal import Journal, Node
+from .journal2report import journal2report
 from omegaconf import OmegaConf
 from rich.columns import Columns
 from rich.console import Group
@@ -130,8 +131,15 @@ def run():
             live.update(generate_live())
     interpreter.cleanup_session()
 
-    # report = journal2report(journal, task_desc)
+    if cfg.generate_report:
+        print("Generating final report from journal...")
+        report = journal2report(journal, task_desc, cfg.report)
+        print(report)
+        report_file_path = cfg.log_dir / 'report.md'
+        with open(report_file_path, "w") as f:
+            f.write(report)
+        print('Report written to file:', report_file_path)
 
 
 if __name__ == "__main__":
-    run()
+    run()

aide/utils/config.py

@@ -58,7 +58,6 @@ class ExecConfig:
     agent_file_name: str
     format_tb_ipython: bool
 
-
 @dataclass
 class Config(Hashable):
     data_dir: Path
@@ -76,6 +75,8 @@ class Config(Hashable):
     exp_name: str
 
     exec: ExecConfig
+    generate_report: bool
+    report: StageConfig
     agent: AgentConfig
 
 

aide/utils/config.yaml

@@ -24,6 +24,12 @@ exec:
   agent_file_name: runfile.py
   format_tb_ipython: False
 
+generate_report: True
+# LLM settings for final report from journal
+report:
+  model: gpt-4-turbo
+  temp: 1.0
+
 # agent hyperparams
 agent:
   # how many improvement iterations to run