diff --git a/cli/text2embeddings.py b/cli/text2embeddings.py index e1b997b..3d9bd85 100644 --- a/cli/text2embeddings.py +++ b/cli/text2embeddings.py @@ -178,9 +178,11 @@ def run_as_cli( task_output_path = os.path.join(output_dir, task.document_id + ".json") try: - write_json_to_s3(task.model_dump_json(), task_output_path) if s3 else Path( - task_output_path - ).write_text(task.model_dump_json()) + write_json_to_s3( + task.model_dump_json(indent=2), task_output_path + ) if s3 else Path(task_output_path).write_text( + task.model_dump_json(indent=2) + ) except Exception as e: logger.info( "Failed to write embeddings data to s3.",