[feat] Create endpoints to handle reviewing a PR and checking state of the review

src/seer/app.py

@@ -45,12 +45,16 @@
 )
 from seer.automation.codebase.repo_client import RepoClient
 from seer.automation.codegen.models import (
+    CodegenPrReviewRequest,
+    CodegenPrReviewResponse,
+    CodegenPrReviewStateRequest,
+    CodegenPrReviewStateResponse,
     CodegenUnitTestsRequest,
     CodegenUnitTestsResponse,
     CodegenUnitTestsStateRequest,
     CodegenUnitTestsStateResponse,
 )
-from seer.automation.codegen.tasks import codegen_unittest, get_unittest_state
+from seer.automation.codegen.tasks import codegen_pr_review, codegen_unittest, get_unittest_state
 from seer.automation.summarize.issue import run_summarize_issue
 from seer.automation.summarize.models import SummarizeIssueRequest, SummarizeIssueResponse
 from seer.automation.utils import raise_if_no_genai_consent
@@ -260,6 +264,18 @@ def codegen_unit_tests_state_endpoint(
     )
 
 
+@json_api(blueprint, "/v1/automation/codegen/pr-review")
+def codegen_pr_review_endpoint(data: CodegenPrReviewRequest) -> CodegenPrReviewResponse:
+    return codegen_pr_review(data)
+
+
+@json_api(blueprint, "/v1/automation/codegen/pr-review/state")
+def codegen_pr_review_state_endpoint(
+    data: CodegenPrReviewStateRequest,
+) -> CodegenPrReviewStateResponse:
+    raise NotImplementedError("PR Review state is not implemented yet.")
+
+
 @json_api(blueprint, "/v1/automation/summarize/issue")
 def summarize_issue_endpoint(data: SummarizeIssueRequest) -> SummarizeIssueResponse:
     return run_summarize_issue(data)

src/seer/automation/codegen/models.py

@@ -33,6 +33,11 @@ class CodegenUnitTestsRequest(BaseModel):
     pr_id: int  # The PR number
 
 
+class CodegenPrReviewRequest(BaseModel):
+    repo: RepoDefinition
+    pr_id: int  # The PR number
+
+
 class CodegenContinuation(CodegenState):
     request: CodegenUnitTestsRequest
 
@@ -52,6 +57,10 @@ class CodegenUnitTestsResponse(BaseModel):
     run_id: int
 
 
+class CodegenPrReviewResponse(BaseModel):
+    run_id: int
+
+
 class CodegenUnitTestsStateRequest(BaseModel):
     run_id: int
 
@@ -63,3 +72,16 @@ class CodegenUnitTestsStateResponse(BaseModel):
     triggered_at: datetime.datetime
     updated_at: datetime.datetime
     completed_at: datetime.datetime | None = None
+
+
+class CodegenPrReviewStateRequest(BaseModel):
+    run_id: int
+
+
+class CodegenPrReviewStateResponse(BaseModel):
+    run_id: int
+    status: CodegenStatus
+    changes: list[FileChange]
+    triggered_at: datetime.datetime
+    updated_at: datetime.datetime
+    completed_at: datetime.datetime | None = None

src/seer/automation/codegen/tasks.py

@@ -1,6 +1,7 @@
 from celery_app.config import CeleryQueues
 from seer.automation.codegen.models import (
     CodegenContinuation,
+    CodegenPrReviewRequest,
     CodegenStatus,
     CodegenUnitTestsRequest,
     CodegenUnitTestsResponse,
@@ -46,3 +47,7 @@ def codegen_unittest(request: CodegenUnitTestsRequest):
 def get_unittest_state(request: CodegenUnitTestsStateRequest):
     state = CodegenContinuationState.from_id(request.run_id, model=CodegenContinuation)
     return state.get()
+
+
+def codegen_pr_review(request: CodegenPrReviewRequest):
+    raise NotImplementedError("PR Review is not implemented yet.")