From 7eadb45c09ede8cc3b515ef01e431a0ca2df8277 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Sat, 27 Jul 2024 17:23:42 +0300 Subject: [PATCH] Refactor seed handling logic in litellm_ai_handler to improve readability and error checking --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index 8075376f..8df78dff 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -129,9 +129,10 @@ class LiteLLMAIHandler(BaseAiHandler): "force_timeout": get_settings().config.ai_timeout, "api_base": self.api_base, } - if get_settings().config.get("seed", -1) > 0: - if temperature > 0: - raise ValueError("Seed is not supported with temperature > 0") + seed = get_settings().config.get("seed", -1) + if temperature > 0 and seed >= 0: + raise ValueError("Seed is not supported with temperature > 0") + elif seed > 0: kwargs["seed"] = get_settings().config.seed if self.repetition_penalty: