minhvtt commited on
Commit
885208c
verified
1 Parent(s): 5929e46

Update scenario_handlers/event_recommendation.py

Browse files
scenario_handlers/event_recommendation.py CHANGED
@@ -69,7 +69,7 @@ class EventRecommendationHandler(BaseScenarioHandler):
69
  # Save results to scenario data
70
  scenario_data['rag_results'] = formatted_events
71
  scenario_data['available_events'] = [
72
- r['metadata'].get('id_use', metadata.get('original_id', f'event_{i+1}'))
73
  for i, r in enumerate(results)
74
  ]
75
 
 
69
  # Save results to scenario data
70
  scenario_data['rag_results'] = formatted_events
71
  scenario_data['available_events'] = [
72
+ r['metadata'].get('id_use', r['metadata'].get('original_id', f'event_{i+1}'))
73
  for i, r in enumerate(results)
74
  ]
75