completely done

Author: kenxu95

action.yml

@@ -85,6 +85,7 @@ runs:
       shell: bash
 
     - name: Create Pull Request
+      id: create_pr
       uses: peter-evans/create-pull-request@v3
       with:
         token: ${{ inputs.github_token }}
@@ -94,4 +95,13 @@ runs:
         body: ${{ steps.get-pr-body.outputs.body}}
         base: ${{ inputs.base_branch }}
 
-    # TODO: finish the autodoc workflow
+    - name: Mark the Autodoc Run as completed
+      env:
+        PYTHONPATH: ${{ github.action_path }}
+      run: |
+        python ${{ github.action_path }}/scripts/mark_run_completed.py \
+        --server-address ${{ inputs.runllm_server_address }} \
+        --api-key ${{ inputs.runllm_api_key }} \ 
+        --run-id ${{ github.run_id }} \
+        --gh-action-url ${{ steps.create_pr.outputs.pull-request-url }}
+      shell: bash

scripts/mark_run_completed.py

@@ -0,0 +1,14 @@
+import argparse
+
+from scripts.client import RunLLMClient
+
+parser = argparse.ArgumentParser(description="Mark an autodoc run as completed.")
+parser.add_argument("--server-address", type=str, required=True, help="The address of the RunLLM server")
+parser.add_argument("--api-key", type=str, required=True, help="The API key to use for the RunLLM server")
+parser.add_argument("--run-id", type=int, required=True, help="The GH action run id")
+parser.add_argument("--pr-url", type=str, required=True, help="The URL of the pull request")
+args = parser.parse_args()
+
+if __name__ == "__main__":
+    client = RunLLMClient(args.server_address, args.api_key)
+    client.mark_completed(args.run_id, args.pr_url)