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

Update seed handling: log fixed seed usage; adjust default seed and temperature in config

Browse Source
This commit is contained in:
mrT23
2024-07-27 17:50:59 +03:00
parent 70f7a90226
commit 3067afbcb3
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View File

@ -132,7 +132,8 @@ class LiteLLMAIHandler(BaseAiHandler):
seed = get_settings().config.get("seed", -1) seed = get_settings().config.get("seed", -1)
if temperature > 0 and seed >= 0: if temperature > 0 and seed >= 0:
raise ValueError("Seed is not supported with temperature > 0") raise ValueError("Seed is not supported with temperature > 0")
elif seed > 0: elif seed >= 0:
get_logger().info(f"Using fixed seed of {seed}")
kwargs["seed"] = get_settings().config.seed kwargs["seed"] = get_settings().config.seed
if self.repetition_penalty: if self.repetition_penalty:

4
pr_agent/settings/configuration.toml
View File

@ -27,8 +27,8 @@ output_relevant_configurations=false
large_patch_policy = "clip" # "clip", "skip" large_patch_policy = "clip" # "clip", "skip"
is_auto_command=false is_auto_command=false
# seed # seed
seed=0 # set positive value to fix the seed seed=-1 # set positive value to fix the seed (and ensure temperature=0)
temperature=0 temperature=0.2
[pr_reviewer] # /review # [pr_reviewer] # /review #
# enable/disable features # enable/disable features