xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-06 22:00:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: add TypedDict and type hints to todo item formatter

Browse Source
This commit is contained in:
Judy
2025-05-27 17:52:36 +09:00
parent 7d38814cae
commit 53b913a4cb
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
pr_agent/algo/utils.py
View File

@ -14,7 +14,7 @@ import traceback
from datetime import datetime from datetime import datetime
from enum import Enum from enum import Enum
from importlib.metadata import PackageNotFoundError, version from importlib.metadata import PackageNotFoundError, version
from typing import Any, List, Tuple from typing import Any, List, Tuple, TypedDict
import html2text import html2text
import requests import requests
@ -37,21 +37,31 @@ def get_model(model_type: str = "model_weak") -> str:
return get_settings().config.model_reasoning return get_settings().config.model_reasoning
return get_settings().config.model return get_settings().config.model
class Range(BaseModel): class Range(BaseModel):
line_start: int # should be 0-indexed line_start: int # should be 0-indexed
line_end: int line_end: int
column_start: int = -1 column_start: int = -1
column_end: int = -1 column_end: int = -1
class ModelType(str, Enum): class ModelType(str, Enum):
REGULAR = "regular" REGULAR = "regular"
WEAK = "weak" WEAK = "weak"
REASONING = "reasoning" REASONING = "reasoning"
class TodoItem(TypedDict):
relevant_file: str
line_number: int
content: str
class PRReviewHeader(str, Enum): class PRReviewHeader(str, Enum):
REGULAR = "## PR Reviewer Guide" REGULAR = "## PR Reviewer Guide"
INCREMENTAL = "## Incremental PR Reviewer Guide" INCREMENTAL = "## Incremental PR Reviewer Guide"
class ReasoningEffort(str, Enum): class ReasoningEffort(str, Enum):
HIGH = "high" HIGH = "high"
MEDIUM = "medium" MEDIUM = "medium"
@ -109,6 +119,7 @@ def unique_strings(input_list: List[str]) -> List[str]:
seen.add(item) seen.add(item)
return unique_list return unique_list
def convert_to_markdown_v2(output_data: dict, def convert_to_markdown_v2(output_data: dict,
gfm_supported: bool = True, gfm_supported: bool = True,
incremental_review=None, incremental_review=None,
@ -211,7 +222,7 @@ def convert_to_markdown_v2(output_data: dict,
value = emphasize_header(value.strip(), only_markdown=True) value = emphasize_header(value.strip(), only_markdown=True)
markdown_text += f"{value}\n\n" markdown_text += f"{value}\n\n"
elif 'todo sections' in key_nice.lower(): elif 'todo sections' in key_nice.lower():
def format_todo_item(todo_item): def format_todo_item(todo_item: TodoItem) -> str:
relevant_file = todo_item.get('relevant_file', '').strip() relevant_file = todo_item.get('relevant_file', '').strip()
line_number = todo_item.get('line_number', '') line_number = todo_item.get('line_number', '')
content = todo_item.get('content', '') content = todo_item.get('content', '')