JetBrains-Research/commit-message-editing-visualization

1import os2from pathlib import Path3 4RANDOM_STATE = 425 6GRAZIE_API_JWT_TOKEN = os.environ.get("GRAZIE_JWT_TOKEN")7GRAZIE_TIMEOUT_SEC = 1.08 9HF_TOKEN = os.environ.get("HF_TOKEN")10 11HF_RAW_DATASET_NAME = "JetBrains-Research/commit-msg-rewriting"12HF_RAW_DATASET_SPLIT = "train"13 14HF_FULL_COMMITS_DATASET_NAME = "JetBrains-Research/lca-commit-message-generation"15HF_FULL_COMMITS_DATASET_SUBNAME = "commitchronicle-py-long"16HF_FULL_COMMITS_DATASET_SPLIT = "test"17 18HF_PREDICTIONS_DATASET_NAME = "JetBrains-Research/lca-results"19HF_PREDICTIONS_MODEL = "gpt_4_0613"20 21HF_SYNTHETIC_DATASET_NAME = "JetBrains-Research/synthetic-commit-msg-rewriting"22HF_SYNTHETIC_DATASET_SPLIT = "train"23 24LLM_MODEL = "gpt-4-1106-preview"25 26CACHE_DIR = Path("cache")27CACHE_DIR.mkdir(exist_ok=True)28 29OUTPUT_DIR = Path("output")30OUTPUT_DIR.mkdir(exist_ok=True)31 32END_TO_START_ARTIFACT = OUTPUT_DIR / "end_to_start.csv"33START_TO_END_ARTIFACT = OUTPUT_DIR / "start_to_end.csv"34SYNTHETIC_DATASET_ARTIFACT = OUTPUT_DIR / "synthetic.csv"35METRICS_CORRELATIONS_ARTIFACT = OUTPUT_DIR / "metrics_correlations.csv"36DATA_FOR_LABELING_ARTIFACT = OUTPUT_DIR / "data_for_labeling.csv"37 38OUTPUT_CHARTS_DIR = OUTPUT_DIR / "charts"39OUTPUT_CHARTS_DIR.mkdir(exist_ok=True)40