studio/docs
Archived
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛(backend) improve sanitizer ai_services

Browse Source 
The json response of the AI service is badly formatted.
This commit improves the sanitizer to try
to handle the response correctly.
This commit is contained in:
Anthony LC
2024-12-11 19:48:54 +01:00
committed by Anthony LC
parent be6da38a08
commit 57c5c394f5
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/backend/core/services/ai_services.py
View File

@@ -69,12 +69,17 @@ class AIService:
content = response.choices[0].message.content content = response.choices[0].message.content
try: try:
sanitized_content = re.sub(r"(?<!\\)\n", "\\\\n", content) sanitized_content = re.sub(r'\s*"answer"\s*:\s*', '"answer": ', content)
sanitized_content = re.sub(r"\s*\}", "}", sanitized_content)
sanitized_content = re.sub(r"(?<!\\)\n", "\\\\n", sanitized_content)
sanitized_content = re.sub(r"(?<!\\)\t", "\\\\t", sanitized_content) sanitized_content = re.sub(r"(?<!\\)\t", "\\\\t", sanitized_content)
json_response = json.loads(sanitized_content) json_response = json.loads(sanitized_content)
except (json.JSONDecodeError, IndexError): except (json.JSONDecodeError, IndexError):
json_response = json.loads(content) try:
json_response = json.loads(content)
except json.JSONDecodeError as err:
raise RuntimeError("AI response is not valid JSON", content) from err
if "answer" not in json_response: if "answer" not in json_response:
raise RuntimeError("AI response does not contain an answer") raise RuntimeError("AI response does not contain an answer")