Update pr_agent/algo/ai_handlers/litellm_ai_handler.py

Co-authored-by: codiumai-pr-agent-pro[bot] <151058649+codiumai-pr-agent-pro[bot]@users.noreply.github.com>
This commit is contained in:
Tal
2024-07-27 18:02:57 +03:00
committed by GitHub
parent 778d7ce1ed
commit 3904eebf85

View File

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