wony617 commited on
Commit
5d333b9
Β·
1 Parent(s): ce8a422

Fix log function

Browse files
Files changed (1) hide show
  1. agent/workflow.py +16 -4
agent/workflow.py CHANGED
@@ -217,23 +217,35 @@ def generate_github_pr(
217
  if result["status"] == "success":
218
  # Log successful PR to file
219
  import datetime
220
- log_entry = f"[{datetime.datetime.now().isoformat()}] {result['file_path']} -> {result['pr_url']}\n"
 
221
  try:
222
  with open("pr_success.log", "a", encoding="utf-8") as f:
223
  f.write(log_entry)
224
- print(f"βœ… Logged successful PR: {log_entry.strip()}")
225
  except Exception as e:
226
- print(f"❌ Failed to log PR success: {e}")
227
 
228
  return f"""βœ… **GitHub PR Creation Successful!**
229
 
230
- πŸ”— **PR URL:** {result["pr_url"]}
231
  🌿 **Branch:** {result["branch"]}
232
  πŸ“ **File:** {result["file_path"]}{toctree_status}
233
 
234
  {result["message"]}"""
235
 
236
  elif result["status"] == "partial_success":
 
 
 
 
 
 
 
 
 
 
 
237
  return f"""⚠️ **Partial Success**
238
 
239
  🌿 **Branch:** {result["branch"]}
 
217
  if result["status"] == "success":
218
  # Log successful PR to file
219
  import datetime
220
+ pr_url = result.get('pr_url', 'NO_PR_URL')
221
+ log_entry = f"[{datetime.datetime.now().isoformat()}] {result['file_path']} -> {pr_url} ({result['status']})\n"
222
  try:
223
  with open("pr_success.log", "a", encoding="utf-8") as f:
224
  f.write(log_entry)
225
+ print(f"βœ… Logged PR result: {log_entry.strip()}")
226
  except Exception as e:
227
+ print(f"❌ Failed to log PR result: {e}")
228
 
229
  return f"""βœ… **GitHub PR Creation Successful!**
230
 
231
+ πŸ”— **PR URL:** {result.get('pr_url', 'NO_PR_URL')}
232
  🌿 **Branch:** {result["branch"]}
233
  πŸ“ **File:** {result["file_path"]}{toctree_status}
234
 
235
  {result["message"]}"""
236
 
237
  elif result["status"] == "partial_success":
238
+ # Log partial success to file
239
+ import datetime
240
+ pr_url = result.get('pr_url', 'NO_PR_URL')
241
+ log_entry = f"[{datetime.datetime.now().isoformat()}] {result['file_path']} -> {pr_url} ({result['status']})\n"
242
+ try:
243
+ with open("pr_success.log", "a", encoding="utf-8") as f:
244
+ f.write(log_entry)
245
+ print(f"βœ… Logged PR result: {log_entry.strip()}")
246
+ except Exception as e:
247
+ print(f"❌ Failed to log PR result: {e}")
248
+
249
  return f"""⚠️ **Partial Success**
250
 
251
  🌿 **Branch:** {result["branch"]}