diff --git a/pr_agent/algo/ai_handler.py b/pr_agent/algo/ai_handler.py index ac73a325..348056ff 100644 --- a/pr_agent/algo/ai_handler.py +++ b/pr_agent/algo/ai_handler.py @@ -26,7 +26,7 @@ class AiHandler: try: openai.api_key = get_settings().openai.key litellm.openai_key = get_settings().openai.key - litellm.debugger = get_settings().config.litellm_debugger + litellm.debugger = get_settings().litellm.debugger self.azure = False if get_settings().get("OPENAI.ORG", None): litellm.organization = get_settings().openai.org @@ -48,8 +48,8 @@ class AiHandler: litellm.replicate_key = get_settings().replicate.key if get_settings().get("HUGGINGFACE.KEY", None): litellm.huggingface_key = get_settings().huggingface.key - if get_settings().get("LITELLM.USER_EMAIL"): - litellm.email = get_settings().get("LITELLM.USER_EMAIL", None) + if get_settings().get("LITELLM.DEBUGGER") and get_settings().get("LITELLM.EMAIL"): + litellm.email = get_settings().get("LITELLM.EMAIL", None) except AttributeError as e: raise ValueError("OpenAI key is required") from e diff --git a/pr_agent/settings/configuration.toml b/pr_agent/settings/configuration.toml index c46412ed..8281d7d9 100644 --- a/pr_agent/settings/configuration.toml +++ b/pr_agent/settings/configuration.toml @@ -10,7 +10,6 @@ use_repo_settings_file=true ai_timeout=180 max_description_tokens = 500 max_commits_tokens = 500 -litellm_debugger=false secret_provider="google_cloud_storage" [pr_reviewer] # /review # @@ -95,3 +94,7 @@ polling_interval_seconds = 30 # patch_server_endpoint = "http://127.0.0.1:5000/patch" # token to authenticate in the patch server # patch_server_token = "" + +[litellm] +debugger=false +#email="youremail@example.com"