Phoenix21 commited on
Commit
8235467
·
verified ·
1 Parent(s): f6b1707

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -3
pipeline.py CHANGED
@@ -715,10 +715,10 @@ def run_with_chain(query: str) -> str:
715
  # brand vs wellness
716
  if classification == "brand":
717
  rag_chain_main = pipeline_state.brand_rag_chain
718
- rag_chain_fallback = pipeline_state.brand_rag_chain_fallback
719
  else:
720
  rag_chain_main = pipeline_state.wellness_rag_chain
721
- rag_chain_fallback = pipeline_state.wellness_rag_chain_fallback
722
 
723
  # RAG with fallback
724
  try:
@@ -727,7 +727,7 @@ def run_with_chain(query: str) -> str:
727
  except Exception as e_main:
728
  if "resource exhausted" in str(e_main).lower():
729
  logger.warning("Gemini resource exhausted. Falling back to Groq.")
730
- rag_output = rag_chain_fallback({"query": query})
731
  else:
732
  raise
733
 
 
715
  # brand vs wellness
716
  if classification == "brand":
717
  rag_chain_main = pipeline_state.brand_rag_chain
718
+ # rag_chain_fallback = pipeline_state.brand_rag_chain_fallback
719
  else:
720
  rag_chain_main = pipeline_state.wellness_rag_chain
721
+ # rag_chain_fallback = pipeline_state.wellness_rag_chain_fallback
722
 
723
  # RAG with fallback
724
  try:
 
727
  except Exception as e_main:
728
  if "resource exhausted" in str(e_main).lower():
729
  logger.warning("Gemini resource exhausted. Falling back to Groq.")
730
+ # rag_output = rag_chain_fallback({"query": query})
731
  else:
732
  raise
733