diff --git a/.github/workflows/interpret_load_test.py b/.github/workflows/interpret_load_test.py index cf289c10e..dc14cdb5a 100644 --- a/.github/workflows/interpret_load_test.py +++ b/.github/workflows/interpret_load_test.py @@ -1,4 +1,6 @@ import csv +import os +from github import Github def csv_to_markdown(csv_file): @@ -52,6 +54,18 @@ def interpret_results(csv_file): if __name__ == "__main__": csv_file = "load_test_stats.csv" # Change this to the path of your CSV file - interpret_results(csv_file) + interpreted_results_str = interpret_results(csv_file) markdown_table = csv_to_markdown(csv_file) print(markdown_table) + + # Update release body with interpreted results + github_token = os.getenv("GITHUB_TOKEN") + g = Github(github_token) + repo = g.get_repo( + "BerriAI/litellm" + ) # Replace with your repository's username and name + latest_release = repo.get_latest_release() + print("got latest release: ", latest_release) + latest_release.update_release( + body=latest_release.body + "\n\n" + interpreted_results_str + ) diff --git a/.github/workflows/load_test.yml b/.github/workflows/load_test.yml index 050f74d51..877f8c39b 100644 --- a/.github/workflows/load_test.yml +++ b/.github/workflows/load_test.yml @@ -8,6 +8,15 @@ jobs: steps: - name: Checkout uses: actions/checkout@v1 + - name: Setup Python + uses: actions/setup-python@v2 + with: + python-version: '3.x' + + - name: Install dependencies + run: | + python -m pip install --upgrade pip + pip install PyGithub - name: Run Load Test id: locust_run uses: BerriAI/locust-github-action@master @@ -24,6 +33,8 @@ jobs: python ".github/workflows/interpret_load_test.py" interpreted_results=$(cat load_test_interpreted_results.txt) echo "::set-output name=interpreted_results::$interpreted_results" + env: + GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} working-directory: ${{ github.workspace }} - name: Upload CSV as Asset to Latest Release uses: xresloader/upload-to-github-release@v1