Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- 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 |
|