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

Support multiple model types for different reasoning tasks

Browse Source
This commit is contained in:
mrT23
2025-04-27 08:50:03 +03:00
parent 60a887ffe1
commit f53bd524c5
5 changed files with 27 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pr_agent/algo/utils.py
View File

@ -30,12 +30,13 @@ from pr_agent.config_loader import get_settings, global_settings
from pr_agent.log import get_logger
def get_weak_model() -> str:
if get_settings().get("config.model_weak"):
def get_model(model_type: str = "model_weak") -> str:
if model_type == "model_weak" and get_settings().get("config.model_weak"):
return get_settings().config.model_weak
elif model_type == "model_reasoning" and get_settings().get("config.model_reasoning"):
return get_settings().config.model_reasoning
return get_settings().config.model
class Range(BaseModel):
line_start: int # should be 0-indexed
line_end: int
@ -45,6 +46,7 @@ class Range(BaseModel):
class ModelType(str, Enum):
REGULAR = "regular"
WEAK = "weak"
REASONING = "reasoning"
class PRReviewHeader(str, Enum):
REGULAR = "## PR Reviewer Guide"