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

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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.brand_rag_chain_fallback = build_rag_chain2(brand_vs, self.groq_fallback_llm)
531
- self.wellness_rag_chain_fallback = build_rag_chain(wellness_vs, self.groq_fallback_llm)
 
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