fix llm app bug

microsoft · Dec 24, 2024 · a6505d1 · a6505d1
1 parent 39abb25
commit a6505d1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/rdagent/log/logger.py b/rdagent/log/logger.py
@@ -119,10 +119,10 @@ def log_object(self, obj: object, *, tag: str = "") -> None:
             debug_log_path = self.log_trace_path / "debug_llm.pkl"
             debug_data = {"tag": tag, "obj": obj}
             if debug_log_path.exists():
-                with debug_log_path.open("rb+") as f:
+                with debug_log_path.open("rb") as f:
                     existing_data = pickle.load(f)
-                    existing_data.append(debug_data)
-                    f.seek(0)
+                existing_data.append(debug_data)
+                with debug_log_path.open("wb") as f:
                     pickle.dump(existing_data, f)
             else:
                 with debug_log_path.open("wb") as f:

diff --git a/rdagent/log/ui/llm_st.py b/rdagent/log/ui/llm_st.py
@@ -31,7 +31,7 @@
 
 def load_data():
     try:
-        with open(f"{main_log_path}/{session_state.log_path}/debug_llm.pkl", "r") as f:
+        with open(f"{main_log_path}/{session_state.log_path}/debug_llm.pkl", "rb") as f:
             session_state.data = pickle.load(f)
     except Exception as e:
         session_state.data = [{"error": str(e)}]