1.0.9

Author: kenxu95

scripts/client.py

@@ -132,7 +132,7 @@ def generate_explanation(self, run_id: int, mode: AutodocOutputMode, diff: str)
         self._check_for_error(resp)
         return ExplainAutodocResponse(**resp.json())
 
-    def mark_completed(self, run_id: int, pull_request_url: str) -> None:
+    def mark_completed(self, run_id: str, pull_request_url: str) -> None:
         resp = requests.put(
             self.server_address + f"/api/autodoc/{run_id}",
             headers=self._get_default_headers(),

scripts/mark_run_completed.py

@@ -5,7 +5,7 @@
 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("--run-id", type=str, 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()