Spaces:
Sleeping
Sleeping
Update langgraph_nodes.py
Browse files- 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'],
|