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

fix: update error logging messages and system prompt handling in litellm_ai_handler.py

Browse Source
This commit is contained in:
mrT23
2024-09-15 08:07:59 +03:00
parent cc0e432247
commit 8f943a0d44
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -167,7 +167,7 @@ class LiteLLMAIHandler(BaseAiHandler):
if self.azure: if self.azure:
model = 'azure/' + model model = 'azure/' + model
if 'claude' in model and not system: if 'claude' in model and not system:
system = "\n" system = "No system prompt provided"
get_logger().warning( get_logger().warning(
"Empty system prompt for claude model. Adding a newline character to prevent OpenAI API error.") "Empty system prompt for claude model. Adding a newline character to prevent OpenAI API error.")
messages = [{"role": "system", "content": system}, {"role": "user", "content": user}] messages = [{"role": "system", "content": system}, {"role": "user", "content": user}]
@ -215,13 +215,13 @@ class LiteLLMAIHandler(BaseAiHandler):
response = await acompletion(**kwargs) response = await acompletion(**kwargs)
except (openai.APIError, openai.APITimeoutError) as e: except (openai.APIError, openai.APITimeoutError) as e:
get_logger().warning("Error during OpenAI inference: ", e) get_logger().warning("Error during LLM inference: {e}")
raise raise
except (openai.RateLimitError) as e: except (openai.RateLimitError) as e:
get_logger().error("Rate limit error during OpenAI inference: ", e) get_logger().error("Rate limit error during LLM inference: {e}")
raise raise
except (Exception) as e: except (Exception) as e:
get_logger().warning("Unknown error during OpenAI inference: ", e) get_logger().warning("Unknown error during LLM inference: {e}")
raise openai.APIError from e raise openai.APIError from e
if response is None or len(response["choices"]) == 0: if response is None or len(response["choices"]) == 0:
raise openai.APIError raise openai.APIError