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

Refactor seed handling logic in litellm_ai_handler to improve readability and error checking

Browse Source
This commit is contained in:
mrT23
2024-07-27 17:23:42 +03:00
parent ac247dbc2c
commit 7eadb45c09
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -129,9 +129,10 @@ class LiteLLMAIHandler(BaseAiHandler):
"force_timeout": get_settings().config.ai_timeout, "force_timeout": get_settings().config.ai_timeout,
"api_base": self.api_base, "api_base": self.api_base,
} }
if get_settings().config.get("seed", -1) > 0: seed = get_settings().config.get("seed", -1)
if temperature > 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:
kwargs["seed"] = get_settings().config.seed kwargs["seed"] = get_settings().config.seed
if self.repetition_penalty: if self.repetition_penalty: