Spaces:

Phoenix21
/

ChatBotAgenticRAG

Sleeping

Phoenix21 commited on Jan 18

Commit

76f6ca7

verified ·

1 Parent(s): b30f596

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -219,9 +219,9 @@ brand_store_dir = "faiss_brand_store"
 wellness_vectorstore = build_or_load_vectorstore(wellness_csv, wellness_store_dir)
 brand_vectorstore = build_or_load_vectorstore(brand_csv, brand_store_dir)
-gemini_llm = LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
-wellness_rag_chain = build_rag_chain(gemini_llm, wellness_vectorstore)
-brand_rag_chain = build_rag_chain(gemini_llm, brand_vectorstore)
 # Function to wrap up and run the chain
 def run_with_chain(query: str) -> str:

 wellness_vectorstore = build_or_load_vectorstore(wellness_csv, wellness_store_dir)
 brand_vectorstore = build_or_load_vectorstore(brand_csv, brand_store_dir)
+# gemini_llm = LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
+wellness_rag_chain = build_rag_chain( wellness_vectorstore)
+brand_rag_chain = build_rag_chain( brand_vectorstore)
 # Function to wrap up and run the chain
 def run_with_chain(query: str) -> str: