wony617 commited on
Commit
f319272
Β·
1 Parent(s): 259de55

Fix log error and remove toctree logging

Browse files
Files changed (2) hide show
  1. agent/toctree_handler.py +0 -1
  2. agent/workflow.py +28 -17
agent/toctree_handler.py CHANGED
@@ -265,7 +265,6 @@ Korean title:"""
265
  # Process toctree update with LLM
266
  self.process_pr_commit(filepath)
267
  # Commit toctree as separate commit
268
- print("self.updated_ko_toctree:", self.updated_ko_toctree)
269
  if self.updated_ko_toctree:
270
  return self.commit_and_push_toctree(
271
  pr_agent=pr_agent,
 
265
  # Process toctree update with LLM
266
  self.process_pr_commit(filepath)
267
  # Commit toctree as separate commit
 
268
  if self.updated_ko_toctree:
269
  return self.commit_and_push_toctree(
270
  pr_agent=pr_agent,
agent/workflow.py CHANGED
@@ -12,7 +12,14 @@ from translator.content import (
12
  preprocess_content,
13
  )
14
  from translator.retriever import report, get_github_issue_open_pr
15
- from pr_generator.agent import GitHubPRAgent
 
 
 
 
 
 
 
16
 
17
  import json
18
  from logger.github_logger import GitHubLogger
@@ -175,21 +182,22 @@ def generate_github_pr(
175
  print(f" 🏠 Repository: {github_config['owner']}/{github_config['repo_name']}")
176
 
177
  agent = GitHubPRAgent()
178
- # result = agent.run_translation_pr_workflow(
179
- # reference_pr_url=github_config["reference_pr_url"],
180
- # target_language=target_language,
181
- # filepath=filepath,
182
- # translated_doc=translated_content,
183
- # owner=github_config["owner"],
184
- # repo_name=github_config["repo_name"],
185
- # base_branch=github_config.get("base_branch", "main"),
186
- # )
187
- result = {
188
- 'status': 'partial_success',
189
- 'branch': 'ko-attention_interface',
190
- 'file_path': 'docs/source/ko/attention_interface.md',
191
- 'message': 'File was saved and commit was successful.\nPR creation failed: ERROR: Existing PR found: https://github.com/Jwaminju/transformers/pull/1', 'error_details': 'ERROR: Existing PR found: https://github.com/Jwaminju/transformers/pull/1'
192
- }
 
193
  # Process toctree update after successful translation PR
194
  toctree_result = None
195
  if en_title:
@@ -210,7 +218,10 @@ def generate_github_pr(
210
 
211
  # Append full result JSON to dedicated GitHub logging repository (always)
212
  try:
213
- log_entry = json.dumps(result, ensure_ascii=False) + "\n"
 
 
 
214
  log_res = GitHubLogger().append_jsonl(log_entry)
215
  print(f"πŸ“ Log append result: {log_res}")
216
  except Exception as e:
 
12
  preprocess_content,
13
  )
14
  from translator.retriever import report, get_github_issue_open_pr
15
+ # GitHub PR Agent import
16
+ try:
17
+ from pr_generator.agent import GitHubPRAgent
18
+
19
+ GITHUB_PR_AVAILABLE = True
20
+ except ImportError as e:
21
+ print(f"⚠️ GitHub PR Agent is not available: {e}")
22
+ GITHUB_PR_AVAILABLE = False
23
 
24
  import json
25
  from logger.github_logger import GitHubLogger
 
182
  print(f" 🏠 Repository: {github_config['owner']}/{github_config['repo_name']}")
183
 
184
  agent = GitHubPRAgent()
185
+ result = agent.run_translation_pr_workflow(
186
+ reference_pr_url=github_config["reference_pr_url"],
187
+ target_language=target_language,
188
+ filepath=filepath,
189
+ translated_doc=translated_content,
190
+ owner=github_config["owner"],
191
+ repo_name=github_config["repo_name"],
192
+ base_branch=github_config.get("base_branch", "main"),
193
+ )
194
+ # TEST CODE
195
+ # result = {
196
+ # 'status': 'partial_success',
197
+ # 'branch': 'ko-attention_interface',
198
+ # 'file_path': 'docs/source/ko/attention_interface.md',
199
+ # 'message': 'File was saved and commit was successful.\nPR creation failed: ERROR: Existing PR found: https://github.com/Jwaminju/transformers/pull/1', 'error_details': 'ERROR: Existing PR found: https://github.com/Jwaminju/transformers/pull/1'
200
+ # }
201
  # Process toctree update after successful translation PR
202
  toctree_result = None
203
  if en_title:
 
218
 
219
  # Append full result JSON to dedicated GitHub logging repository (always)
220
  try:
221
+ log_data = result.copy()
222
+ if toctree_result:
223
+ log_data["toctree_result"] = toctree_result
224
+ log_entry = json.dumps(log_data, ensure_ascii=False) + "\n"
225
  log_res = GitHubLogger().append_jsonl(log_entry)
226
  print(f"πŸ“ Log append result: {log_res}")
227
  except Exception as e: