ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 10:55:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: D350490E39D5F5AD
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
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: