Spaces:
Build error
Build error
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 |
|