mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import argparse
|
|
import os
|
|
|
|
from openhands.resolver.io_utils import load_single_resolver_output
|
|
|
|
|
|
def visualize_resolver_output(
|
|
issue_number: int, output_dir: str, vis_method: str
|
|
) -> None:
|
|
output_jsonl = os.path.join(output_dir, 'output.jsonl')
|
|
resolver_output = load_single_resolver_output(output_jsonl, issue_number)
|
|
if vis_method == 'json':
|
|
print(resolver_output.model_dump_json(indent=4))
|
|
else:
|
|
raise ValueError(f'Invalid visualization method: {vis_method}')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Visualize a patch.')
|
|
parser.add_argument(
|
|
'--issue-number',
|
|
type=int,
|
|
required=True,
|
|
help='Issue number to send the pull request for.',
|
|
)
|
|
parser.add_argument(
|
|
'--output-dir',
|
|
type=str,
|
|
default='output',
|
|
help='Output directory to write the results.',
|
|
)
|
|
parser.add_argument(
|
|
'--vis-method',
|
|
type=str,
|
|
default='json',
|
|
choices=['json'],
|
|
help='Method to visualize the patch [json].',
|
|
)
|
|
my_args = parser.parse_args()
|
|
|
|
visualize_resolver_output(
|
|
issue_number=my_args.issue_number,
|
|
output_dir=my_args.output_dir,
|
|
vis_method=my_args.vis_method,
|
|
)
|