riazmo commited on
Commit
c98cd56
·
verified ·
1 Parent(s): 61dc3b2

Update langgraph_nodes.py

Browse files
Files changed (1) hide show
  1. langgraph_nodes.py +0 -18
langgraph_nodes.py CHANGED
@@ -375,12 +375,6 @@ def stage1_classification_node(state: ReviewState) -> Dict[str, Any]:
375
  stage1_time = time.time() - start_time
376
  print(f" ✅ Stage 1 complete ({stage1_time:.2f}s)")
377
 
378
- # Save to database
379
- db = EnhancedDatabase()
380
- db.connect()
381
- db.save_stage1_results(state['review_id'], llm1_result, llm2_result, manager_result)
382
- db.close()
383
-
384
  return {
385
  "llm1_result": llm1_result,
386
  "llm2_result": llm2_result,
@@ -534,12 +528,6 @@ def stage2_sentiment_node(state: ReviewState) -> Dict[str, Any]:
534
  stage2_time = time.time() - start_time
535
  print(f" ✅ Stage 2 complete ({stage2_time:.2f}s)")
536
 
537
- # Save to database
538
- db = EnhancedDatabase()
539
- db.connect()
540
- db.save_stage2_results(state['review_id'], best_result, alt_result, layer_result)
541
- db.close()
542
-
543
  return {
544
  "best_sentiment_result": best_result,
545
  "alt_sentiment_result": alt_result,
@@ -690,12 +678,6 @@ Respond ONLY in valid JSON format:
690
  # Calculate total time
691
  total_time = state.get('stage1_time', 0) + state.get('stage2_time', 0) + stage3_time
692
 
693
- # Save to database
694
- db = EnhancedDatabase()
695
- db.connect()
696
- db.save_stage3_results(state['review_id'], result)
697
- db.close()
698
-
699
  return {
700
  "final_result": result,
701
  "final_sentiment": result['final_sentiment'],
 
375
  stage1_time = time.time() - start_time
376
  print(f" ✅ Stage 1 complete ({stage1_time:.2f}s)")
377
 
 
 
 
 
 
 
378
  return {
379
  "llm1_result": llm1_result,
380
  "llm2_result": llm2_result,
 
528
  stage2_time = time.time() - start_time
529
  print(f" ✅ Stage 2 complete ({stage2_time:.2f}s)")
530
 
 
 
 
 
 
 
531
  return {
532
  "best_sentiment_result": best_result,
533
  "alt_sentiment_result": alt_result,
 
678
  # Calculate total time
679
  total_time = state.get('stage1_time', 0) + state.get('stage2_time', 0) + stage3_time
680
 
 
 
 
 
 
 
681
  return {
682
  "final_result": result,
683
  "final_sentiment": result['final_sentiment'],