Spaces:
Running
Running
Update pipeline.py
Browse files- pipeline.py +4 -3
pipeline.py
CHANGED
@@ -526,9 +526,10 @@ class PipelineState:
|
|
526 |
|
527 |
# self.brand_rag_chain = build_rag_chain2(brand_vs, self.gemini_llm)
|
528 |
# self.wellness_rag_chain = build_rag_chain(wellness_vs, self.gemini_llm)
|
529 |
-
|
530 |
-
self.
|
531 |
-
self.
|
|
|
532 |
|
533 |
def handle_error(self, error: Exception) -> bool:
|
534 |
self.error_count += 1
|
|
|
526 |
|
527 |
# self.brand_rag_chain = build_rag_chain2(brand_vs, self.gemini_llm)
|
528 |
# self.wellness_rag_chain = build_rag_chain(wellness_vs, self.gemini_llm)
|
529 |
+
self.brand_rag_chain = build_rag_chain2(brand_vs, self.groq_fallback_llm)
|
530 |
+
self.wellness_rag_chain = build_rag_chain(wellness_vs, self.groq_fallback_llm)
|
531 |
+
# self.brand_rag_chain_fallback = build_rag_chain2(brand_vs, self.groq_fallback_llm)
|
532 |
+
# self.wellness_rag_chain_fallback = build_rag_chain(wellness_vs, self.groq_fallback_llm)
|
533 |
|
534 |
def handle_error(self, error: Exception) -> bool:
|
535 |
self.error_count += 1
|