amit/quantum-ai
1
0
Fork 0
mirror of https://github.com/thecyberlearn/quantum-ai.git synced 2026-08-18 08:33:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix UNIQUE constraint error in Data Analyzer

Browse Source 
- Replace objects.create() with get_or_create() to prevent duplicate responses
- Handle case where response already exists by updating it instead of creating new one
- Fix both success and error response creation
- Prevents 'UNIQUE constraint failed: data_analyzer_responses.request_id' error

Issue: OneToOneField relationship allows only one response per request
Solution: Use get_or_create() pattern to handle existing responses gracefully
This commit is contained in:
Claude 2025-07-16 22:54:09 +05:30
parent f6ba431514
commit 65c713ef64
1 changed files with 34 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
data_analyzer/processor.py
View File

@ -106,17 +106,29 @@ class DataAnalysisAgentProcessor(StandardWebhookProcessor):
# Determine success based on N8N status
success = status == 'success' and bool(analysis_text)
# Create response object
response_obj = DataAnalysisAgentResponse.objects.create(
# Create or update response object (prevent duplicate responses)
response_obj, created = DataAnalysisAgentResponse.objects.get_or_create(
request=request_obj,
success=success,
processing_time=response_data.get('processing_time', 0),
analysis_results=analysis_results,
insights_summary=insights_summary,
report_text=report_text,
raw_response=raw_response,
defaults={
'success': success,
'processing_time': response_data.get('processing_time', 0),
'analysis_results': analysis_results,
'insights_summary': insights_summary,
'report_text': report_text,
'raw_response': raw_response,
}
)
# If response already exists, update it
if not created:
response_obj.success = success
response_obj.processing_time = response_data.get('processing_time', 0)
response_obj.analysis_results = analysis_results
response_obj.insights_summary = insights_summary
response_obj.report_text = report_text
response_obj.raw_response = raw_response
response_obj.save()
# Only deduct wallet balance after successful processing
if success:
request_obj.user.deduct_balance(
@ -138,12 +150,21 @@ class DataAnalysisAgentProcessor(StandardWebhookProcessor):
request_obj.status = 'failed'
request_obj.save()
# Create error response
error_response = DataAnalysisAgentResponse.objects.create(
# Create or update error response (prevent duplicate responses)
error_response, created = DataAnalysisAgentResponse.objects.get_or_create(
request=request_obj,
success=False,
error_message=str(e),
processing_time=response_data.get('processing_time', 0) if response_data else 0
defaults={
'success': False,
'error_message': str(e),
'processing_time': response_data.get('processing_time', 0) if response_data else 0
}
)
# If response already exists, update it with error info
if not created:
error_response.success = False
error_response.error_message = str(e)
error_response.processing_time = response_data.get('processing_time', 0) if response_data else 0
error_response.save()
raise Exception(f"Failed to process Data Analysis Agent response: {e}")