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

Add error handling for empty system prompt in litellm_ai_handler and type conversion in utils.py

Browse Source
This commit is contained in:
mrT23
2024-08-13 16:26:32 +03:00
parent 38638bd1c4
commit e7e3970874
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -106,6 +106,10 @@ class LiteLLMAIHandler(BaseAiHandler):
deployment_id = self.deployment_id deployment_id = self.deployment_id
if self.azure: if self.azure:
model = 'azure/' + model model = 'azure/' + model
if 'claude' in model and not system:
system = "\n"
get_logger().warning(
"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}]
if img_path: if img_path:
try: try:

2
pr_agent/algo/utils.py
View File

@ -147,7 +147,7 @@ def convert_to_markdown_v2(output_data: dict,
else: else:
markdown_text += f"### {emoji} {key_nice}: {value}\n\n" markdown_text += f"### {emoji} {key_nice}: {value}\n\n"
elif 'relevant tests' in key_nice.lower(): elif 'relevant tests' in key_nice.lower():
value = value.strip().lower() value = str(value).strip().lower()
if gfm_supported: if gfm_supported:
markdown_text += f"<tr><td>" markdown_text += f"<tr><td>"
if is_value_no(value): if is_value_no(value):