argus-review-code 0.1.0__py3-none-any.whl

argus_review/cli/commands/run_clear_inline_review.py

@@ -0,0 +1,6 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_clear_inline_review():
+    review_service = ReviewService()
+    await review_service.run_clear_inline_review()

argus_review/cli/commands/run_clear_summary_review.py

@@ -0,0 +1,6 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_clear_summary_review():
+    review_service = ReviewService()
+    await review_service.run_clear_summary_review()

argus_review/cli/commands/run_context_review.py

@@ -0,0 +1,7 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_context_review_command():
+    review_service = ReviewService()
+    await review_service.run_context_review()
+    review_service.report_total_cost()

argus_review/cli/commands/run_inline_reply_review.py

@@ -0,0 +1,7 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_inline_reply_review_command():
+    review_service = ReviewService()
+    await review_service.run_inline_reply_review()
+    review_service.report_total_cost()

argus_review/cli/commands/run_inline_review.py

@@ -0,0 +1,7 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_inline_review_command():
+    review_service = ReviewService()
+    await review_service.run_inline_review()
+    review_service.report_total_cost()

argus_review/cli/commands/run_review.py

@@ -0,0 +1,8 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_review_command():
+    review_service = ReviewService()
+    await review_service.run_inline_review()
+    await review_service.run_summary_review()
+    review_service.report_total_cost()

argus_review/cli/commands/run_summary_reply_review.py

@@ -0,0 +1,7 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_summary_reply_review_command():
+    review_service = ReviewService()
+    await review_service.run_summary_reply_review()
+    review_service.report_total_cost()

argus_review/cli/commands/run_summary_review.py

@@ -0,0 +1,7 @@
+from argus_review.services.review.service import ReviewService
+
+
+async def run_summary_review_command():
+    review_service = ReviewService()
+    await review_service.run_summary_review()
+    review_service.report_total_cost()

argus_review/cli/main.py

@@ -0,0 +1,97 @@
+import asyncio
+
+import typer
+
+app = typer.Typer(help="ArgusReview CLI")
+
+
+@app.command("run")
+def run():
+    """Run the full AI review pipeline"""
+    from argus_review.cli.commands.run_review import run_review_command
+
+    typer.secho("Starting full AI review...", fg=typer.colors.CYAN, bold=True)
+    asyncio.run(run_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("run-inline")
+def run_inline():
+    """Run only the inline review"""
+    from argus_review.cli.commands.run_inline_review import run_inline_review_command
+
+    typer.secho("Starting inline AI review...", fg=typer.colors.CYAN)
+    asyncio.run(run_inline_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("run-context")
+def run_context():
+    """Run only the context review"""
+    from argus_review.cli.commands.run_context_review import run_context_review_command
+
+    typer.secho("Starting context AI review...", fg=typer.colors.CYAN)
+    asyncio.run(run_context_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("run-summary")
+def run_summary():
+    """Run only the summary review"""
+    from argus_review.cli.commands.run_summary_review import run_summary_review_command
+
+    typer.secho("Starting summary AI review...", fg=typer.colors.CYAN)
+    asyncio.run(run_summary_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("run-inline-reply")
+def run_inline_reply():
+    """Run only the inline reply review"""
+    from argus_review.cli.commands.run_inline_reply_review import run_inline_reply_review_command
+
+    typer.secho("Starting inline reply AI review...", fg=typer.colors.CYAN)
+    asyncio.run(run_inline_reply_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("run-summary-reply")
+def run_summary_reply():
+    from argus_review.cli.commands.run_summary_reply_review import run_summary_reply_review_command
+
+    typer.secho("Starting summary reply AI review...", fg=typer.colors.CYAN)
+    asyncio.run(run_summary_reply_review_command())
+    typer.secho("AI review completed successfully!", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("clear-inline")
+def clear_inline():
+    """Remove all AI-generated inline review comments"""
+    from argus_review.cli.commands.run_clear_inline_review import run_clear_inline_review
+
+    typer.secho("Clearing inline AI review comments...", fg=typer.colors.YELLOW)
+    asyncio.run(run_clear_inline_review())
+    typer.secho("Inline AI comments cleared", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("clear-summary")
+def clear_summary():
+    """Remove all AI-generated summary review comments"""
+    from argus_review.cli.commands.run_clear_summary_review import run_clear_summary_review
+
+    typer.secho("Clearing summary AI review comments...", fg=typer.colors.YELLOW)
+    asyncio.run(run_clear_summary_review())
+    typer.secho("Summary AI comments cleared", fg=typer.colors.GREEN, bold=True)
+
+
+@app.command("show-config")
+def show_config():
+    """Show the current resolved configuration"""
+    from argus_review.config import settings
+
+    typer.secho("Loaded ArgusReview configuration:", fg=typer.colors.CYAN, bold=True)
+    typer.echo(settings.model_dump_json(indent=2, exclude_none=True))
+
+
+if __name__ == "__main__":
+    app()

argus_review/clients/azure_devops/client.py

@@ -0,0 +1,39 @@
+from httpx import AsyncClient, AsyncHTTPTransport
+
+from argus_review.clients.azure_devops.pr.client import AzureDevOpsPullRequestsHTTPClient
+from argus_review.clients.azure_devops.tools import build_azure_devops_headers
+from argus_review.config import settings
+from argus_review.libs.http.event_hooks.logger import LoggerEventHook
+from argus_review.libs.http.transports.retry import RetryTransport
+from argus_review.libs.logger import get_logger
+
+
+class AzureDevOpsHTTPClient:
+    def __init__(self, client: AsyncClient):
+        self.pr = AzureDevOpsPullRequestsHTTPClient(client)
+
+
+def get_azure_devops_http_client() -> AzureDevOpsHTTPClient:
+    logger = get_logger("AZURE_DEVOPS_HTTP_CLIENT")
+    logger_event_hook = LoggerEventHook(logger=logger)
+    retry_transport = RetryTransport(
+        logger=logger,
+        transport=AsyncHTTPTransport(
+            proxy=settings.vcs.http_client.proxy_url_value,
+            verify=settings.vcs.http_client.verify
+        )
+    )
+
+    client = AsyncClient(
+        verify=settings.vcs.http_client.verify,
+        timeout=settings.vcs.http_client.timeout,
+        headers=build_azure_devops_headers(),
+        base_url=settings.vcs.http_client.api_url_value,
+        transport=retry_transport,
+        event_hooks={
+            "request": [logger_event_hook.request],
+            "response": [logger_event_hook.response],
+        },
+    )
+
+    return AzureDevOpsHTTPClient(client=client)

argus_review/clients/azure_devops/pr/client.py

@@ -0,0 +1,253 @@
+from httpx import Response, QueryParams
+
+from argus_review.clients.azure_devops.pr.schema.files import (
+    AzureDevOpsPRChangeSchema,
+    AzureDevOpsGetPRFilesQuerySchema,
+    AzureDevOpsGetPRFilesResponseSchema,
+)
+from argus_review.clients.azure_devops.pr.schema.pull_request import AzureDevOpsGetPRResponseSchema
+from argus_review.clients.azure_devops.pr.schema.threads import (
+    AzureDevOpsPRThreadSchema,
+    AzureDevOpsGetPRThreadsQuerySchema,
+    AzureDevOpsGetPRThreadsResponseSchema,
+    AzureDevOpsCreatePRThreadRequestSchema,
+    AzureDevOpsUpdatePRThreadRequestSchema,
+    AzureDevOpsCreatePRThreadResponseSchema,
+    AzureDevOpsCreatePRCommentRequestSchema,
+    AzureDevOpsCreatePRCommentResponseSchema,
+)
+from argus_review.clients.azure_devops.pr.types import AzureDevOpsPullRequestsHTTPClientProtocol
+from argus_review.clients.azure_devops.schema import AzureDevOpsBaseQuerySchema
+from argus_review.clients.azure_devops.tools import azure_devops_extract_continuation_token
+from argus_review.config import settings
+from argus_review.libs.http.client import HTTPClient
+from argus_review.libs.http.handlers import handle_http_error, HTTPClientError
+from argus_review.libs.http.paginate import paginate_with_token
+
+
+class AzureDevOpsPullRequestsHTTPClientError(HTTPClientError):
+    pass
+
+
+class AzureDevOpsPullRequestsHTTPClient(HTTPClient, AzureDevOpsPullRequestsHTTPClientProtocol):
+    @handle_http_error(client="AzureDevOpsPullRequestsHTTPClient", exception=AzureDevOpsPullRequestsHTTPClientError)
+    async def get_pull_request_api(
+            self, organization: str, project: str, repository_id: str, pull_request_id: int
+    ) -> Response:
+        url = f"/{organization}/{project}/_apis/git/repositories/{repository_id}/pullRequests/{pull_request_id}"
+        base_query = AzureDevOpsBaseQuerySchema(api_version=settings.vcs.http_client.api_version)
+        return await self.get(url=url, query=QueryParams(**base_query.model_dump(by_alias=True, exclude_none=True)))
+
+    @handle_http_error(client="AzureDevOpsPullRequestsHTTPClient", exception=AzureDevOpsPullRequestsHTTPClientError)
+    async def get_threads_api(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            query: AzureDevOpsGetPRThreadsQuerySchema,
+    ) -> Response:
+        url = (
+            f"/{organization}/{project}/_apis/git/repositories/"
+            f"{repository_id}/pullRequests/{pull_request_id}/threads"
+        )
+        base_query = AzureDevOpsBaseQuerySchema(api_version=settings.vcs.http_client.api_version)
+        return await self.get(
+            url=url,
+            query=QueryParams(
+                **query.model_dump(by_alias=True, exclude_none=True),
+                **base_query.model_dump(by_alias=True, exclude_none=True),
+            ),
+        )
+
+    @handle_http_error(client="AzureDevOpsPullRequestsHTTPClient", exception=AzureDevOpsPullRequestsHTTPClientError)
+    async def create_thread_api(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            request: AzureDevOpsCreatePRThreadRequestSchema,
+    ) -> Response:
+        url = (
+            f"/{organization}/{project}/_apis/git/repositories/"
+            f"{repository_id}/pullRequests/{pull_request_id}/threads"
+        )
+        base_query = AzureDevOpsBaseQuerySchema(api_version=settings.vcs.http_client.api_version)
+        return await self.post(
+            url=url,
+            json=request.model_dump(by_alias=True, exclude_none=True),
+            query=QueryParams(**base_query.model_dump(by_alias=True, exclude_none=True)),
+        )
+
+    @handle_http_error(client="AzureDevOpsPullRequestsHTTPClient", exception=AzureDevOpsPullRequestsHTTPClientError)
+    async def update_thread_api(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            thread_id: int,
+            request: AzureDevOpsUpdatePRThreadRequestSchema,
+    ) -> Response:
+        url = (
+            f"/{organization}/{project}/_apis/git/repositories/"
+            f"{repository_id}/pullRequests/{pull_request_id}/threads/{thread_id}"
+        )
+        base_query = AzureDevOpsBaseQuerySchema(api_version=settings.vcs.http_client.api_version)
+
+        return await self.patch(
+            url=url,
+            json=request.model_dump(by_alias=True, exclude_none=True),
+            query=QueryParams(**base_query.model_dump(by_alias=True, exclude_none=True)),
+        )
+
+    @handle_http_error(client="AzureDevOpsPullRequestsHTTPClient", exception=AzureDevOpsPullRequestsHTTPClientError)
+    async def create_comment_api(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            thread_id: int,
+            request: AzureDevOpsCreatePRCommentRequestSchema,
+    ) -> Response:
+        url = (
+            f"/{organization}/{project}/_apis/git/repositories/"
+            f"{repository_id}/pullRequests/{pull_request_id}/threads/{thread_id}/comments"
+        )
+        base_query = AzureDevOpsBaseQuerySchema(api_version=settings.vcs.http_client.api_version)
+        return await self.post(
+            url=url,
+            json=request.model_dump(by_alias=True, exclude_none=True),
+            query=QueryParams(**base_query.model_dump(by_alias=True, exclude_none=True)),
+        )
+
+    @handle_http_error(client="AzureDevOpsPullRequestsHTTPClient", exception=AzureDevOpsPullRequestsHTTPClientError)
+    async def get_files_api(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            iteration_id: int,
+            query: AzureDevOpsGetPRFilesQuerySchema,
+    ) -> Response:
+        url = (
+            f"/{organization}/{project}/_apis/git/repositories/"
+            f"{repository_id}/pullRequests/{pull_request_id}/iterations/{iteration_id}/changes"
+        )
+        base_query = AzureDevOpsBaseQuerySchema(api_version=settings.vcs.http_client.api_version)
+        return await self.get(
+            url=url,
+            query=QueryParams(
+                **query.model_dump(by_alias=True, exclude_none=True),
+                **base_query.model_dump(by_alias=True, exclude_none=True),
+            ),
+        )
+
+    async def get_pull_request(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int
+    ) -> AzureDevOpsGetPRResponseSchema:
+        response = await self.get_pull_request_api(organization, project, repository_id, pull_request_id)
+        return AzureDevOpsGetPRResponseSchema.model_validate_json(response.text)
+
+    async def get_threads(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int
+    ) -> AzureDevOpsGetPRThreadsResponseSchema:
+        async def fetch_page(token: str | None) -> Response:
+            query = AzureDevOpsGetPRThreadsQuerySchema(
+                top=settings.vcs.pagination.per_page,
+                continuation_token=[token] if token else None
+            )
+            return await self.get_threads_api(organization, project, repository_id, pull_request_id, query)
+
+        def extract_items(response: Response) -> list[AzureDevOpsPRThreadSchema]:
+            parsed = AzureDevOpsGetPRThreadsResponseSchema.model_validate_json(response.text)
+            return parsed.value
+
+        items = await paginate_with_token(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            extract_token=azure_devops_extract_continuation_token,
+        )
+        return AzureDevOpsGetPRThreadsResponseSchema(value=items)
+
+    async def get_files(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            iteration_id: int
+    ) -> AzureDevOpsGetPRFilesResponseSchema:
+        async def fetch_page(token: str | None) -> Response:
+            query = AzureDevOpsGetPRFilesQuerySchema(
+                top=settings.vcs.pagination.per_page,
+                continuation_token=[token] if token else None
+            )
+            return await self.get_files_api(organization, project, repository_id, pull_request_id, iteration_id, query)
+
+        def extract_items(response: Response) -> list[AzureDevOpsPRChangeSchema]:
+            parsed = AzureDevOpsGetPRFilesResponseSchema.model_validate_json(response.text)
+            return parsed.change_entries
+
+        items = await paginate_with_token(
+            max_pages=settings.vcs.pagination.max_pages,
+            fetch_page=fetch_page,
+            extract_items=extract_items,
+            extract_token=azure_devops_extract_continuation_token,
+        )
+        return AzureDevOpsGetPRFilesResponseSchema(change_entries=items)
+
+    async def create_thread(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            request: AzureDevOpsCreatePRThreadRequestSchema,
+    ) -> AzureDevOpsCreatePRThreadResponseSchema:
+        response = await self.create_thread_api(organization, project, repository_id, pull_request_id, request)
+        return AzureDevOpsCreatePRThreadResponseSchema.model_validate_json(response.text)
+
+    async def delete_thread(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            thread_id: int,
+    ) -> None:
+        request = AzureDevOpsUpdatePRThreadRequestSchema(status="closed")
+        await self.update_thread_api(
+            organization=organization,
+            project=project,
+            repository_id=repository_id,
+            pull_request_id=pull_request_id,
+            thread_id=thread_id,
+            request=request,
+        )
+
+    async def create_comment(
+            self,
+            organization: str,
+            project: str,
+            repository_id: str,
+            pull_request_id: int,
+            thread_id: int,
+            request: AzureDevOpsCreatePRCommentRequestSchema,
+    ) -> AzureDevOpsCreatePRCommentResponseSchema:
+        response = await self.create_comment_api(
+            organization, project, repository_id, pull_request_id, thread_id, request
+        )
+        return AzureDevOpsCreatePRCommentResponseSchema.model_validate_json(response.text)

argus_review/clients/azure_devops/pr/schema/files.py

@@ -0,0 +1,51 @@
+from pydantic import BaseModel, Field, ConfigDict, field_validator
+
+
+class AzureDevOpsFilePositionSchema(BaseModel):
+    """Represents a specific position in a file (line and offset)."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    line: int | None = Field(default=None)
+    offset: int | None = Field(default=None)
+    column: int | None = Field(default=None)
+
+
+class AzureDevOpsPRItemSchema(BaseModel):
+    """Represents a file or item in a PR change entry."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    path: str | None = None
+    object_id: str | None = Field(alias="objectId", default=None)
+
+    @field_validator("path")
+    def normalize_path(cls, value: str | None) -> str | None:
+        if value is None:
+            return None
+
+        return value.lstrip("/")
+
+
+class AzureDevOpsPRChangeSchema(BaseModel):
+    """Represents a single file change within a PR iteration."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    item: AzureDevOpsPRItemSchema
+    change_type: str = Field(alias="changeType")
+    change_tracking_id: int | None = Field(alias="changeTrackingId", default=None)
+
+
+class AzureDevOpsGetPRFilesQuerySchema(BaseModel):
+    """Query params for fetching changed files in a PR iteration."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    top: int = 100
+    continuation_token: list[str] | None = Field(alias="continuationToken", default=None)
+
+
+class AzureDevOpsGetPRFilesResponseSchema(BaseModel):
+    """Response model for listing files changed in a PR iteration."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    count: int | None = None
+    change_entries: list[AzureDevOpsPRChangeSchema] = Field(alias="changeEntries")
+    continuation_token: list[str] | None = Field(alias="continuationToken", default=None)

argus_review/clients/azure_devops/pr/schema/pull_request.py

@@ -0,0 +1,38 @@
+from datetime import datetime
+
+from pydantic import BaseModel, Field, ConfigDict
+
+from argus_review.clients.azure_devops.pr.schema.user import AzureDevOpsUserSchema
+
+
+class AzureDevOpsCommitSchema(BaseModel):
+    """Represents a commit object associated with a PR (e.g., last merge commit)."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    commit_id: str = Field(alias="commitId")
+
+
+class AzureDevOpsRepositorySchema(BaseModel):
+    """Represents a repository in Azure DevOps."""
+    id: str
+    url: str | None = None
+    name: str
+
+
+class AzureDevOpsGetPRResponseSchema(BaseModel):
+    """Represents the main Pull Request object returned by Azure DevOps API."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    title: str
+    status: str | None = None
+    reviewers: list[AzureDevOpsUserSchema] = Field(default_factory=list)
+    created_by: AzureDevOpsUserSchema = Field(alias="createdBy")
+    repository: AzureDevOpsRepositorySchema
+    description: str | None = None
+    creation_date: datetime | None = Field(alias="creationDate", default=None)
+    pull_request_id: int = Field(alias="pullRequestId")
+    source_ref_name: str = Field(alias="sourceRefName")
+    target_ref_name: str = Field(alias="targetRefName")
+    last_merge_commit: AzureDevOpsCommitSchema | None = Field(alias="lastMergeCommit", default=None)
+    last_merge_source_commit: AzureDevOpsCommitSchema | None = Field(alias="lastMergeSourceCommit", default=None)
+    last_merge_target_commit: AzureDevOpsCommitSchema | None = Field(alias="lastMergeTargetCommit", default=None)

argus_review/clients/azure_devops/pr/schema/threads.py

@@ -0,0 +1,108 @@
+from datetime import datetime
+from typing import Literal
+
+from pydantic import BaseModel, Field, ConfigDict
+
+from argus_review.clients.azure_devops.pr.schema.files import AzureDevOpsFilePositionSchema
+from argus_review.clients.azure_devops.pr.schema.user import AzureDevOpsUserSchema
+
+
+class AzureDevOpsPRCommentSchema(BaseModel):
+    """Represents a single comment in a PR thread."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    id: int
+    author: AzureDevOpsUserSchema | None = None
+    content: str | None = Field(default=None)
+    is_deleted: bool = Field(alias="isDeleted", default=False)
+    published_date: datetime | None = Field(alias="publishedDate", default=None)
+    last_updated_date: datetime | None = Field(alias="lastUpdatedDate", default=None)
+
+
+class AzureDevOpsThreadContextSchema(BaseModel):
+    """Represents the code location context for a thread (file path and line)."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    file_path: str | None = Field(alias="filePath", default=None)
+    right_file_end: AzureDevOpsFilePositionSchema | None = Field(alias="rightFileEnd", default=None)
+    right_file_start: AzureDevOpsFilePositionSchema | None = Field(alias="rightFileStart", default=None)
+
+
+class AzureDevOpsIterationContextSchema(BaseModel):
+    """Identifies the iteration range in which a thread applies."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    first_comparing_iteration: int = Field(alias="firstComparingIteration")
+    second_comparing_iteration: int = Field(alias="secondComparingIteration")
+
+
+class AzureDevOpsPullRequestThreadContextSchema(BaseModel):
+    """Provides iteration context for PR thread binding."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    change_tracking_id: int | None = Field(alias="changeTrackingId", default=None)
+    iteration_context: AzureDevOpsIterationContextSchema | None = Field(alias="iterationContext", default=None)
+
+
+class AzureDevOpsPRThreadSchema(BaseModel):
+    """Represents a discussion thread in a Pull Request."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    id: int
+    status: str | None = None
+    comments: list[AzureDevOpsPRCommentSchema] = Field(default_factory=list)
+    is_deleted: bool = Field(alias="isDeleted", default=False)
+    thread_context: AzureDevOpsThreadContextSchema | None = Field(alias="threadContext", default=None)
+
+
+class AzureDevOpsGetPRThreadsQuerySchema(BaseModel):
+    """Pagination and query params for fetching PR threads."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    top: int = 100
+    continuation_token: list[str] | None = Field(alias="continuationToken", default=None)
+
+
+class AzureDevOpsGetPRThreadsResponseSchema(BaseModel):
+    """Response model for fetching PR threads with comments."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    value: list[AzureDevOpsPRThreadSchema]
+    count: int | None = None
+    continuation_token: list[str] | None = Field(alias="continuationToken", default=None)
+
+
+class AzureDevOpsCreatePRCommentRequestSchema(BaseModel):
+    """Request for creating a new comment inside an existing thread."""
+    content: str
+
+
+class AzureDevOpsCreatePRCommentResponseSchema(BaseModel):
+    """Response after creating a comment."""
+    id: int
+    content: str
+
+
+class AzureDevOpsCreatePRThreadRequestSchema(BaseModel):
+    """Request for creating a new thread with an initial comment."""
+    model_config = ConfigDict(populate_by_name=True)
+
+    status: str = "active"
+    comments: list[AzureDevOpsCreatePRCommentRequestSchema]
+    thread_context: AzureDevOpsThreadContextSchema | None = Field(alias="threadContext", default=None)
+    pull_request_thread_context: AzureDevOpsPullRequestThreadContextSchema | None = Field(
+        alias="pullRequestThreadContext", default=None
+    )
+
+
+class AzureDevOpsCreatePRThreadResponseSchema(BaseModel):
+    """Response after creating a new discussion thread."""
+    id: int
+    status: str
+    comments: list[AzureDevOpsPRCommentSchema]
+
+
+class AzureDevOpsUpdatePRThreadRequestSchema(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+    status: Literal["active", "fixed", "closed"] | None = None