Phoenix21 commited on
Commit
449be2c
·
verified ·
1 Parent(s): 022d277

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -1
pipeline.py CHANGED
@@ -223,7 +223,10 @@ try:
223
  gemini_llm = LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
224
  wellness_rag_chain = build_rag_chain(gemini_llm, wellness_vectorstore)
225
  brand_rag_chain = build_rag_chain(gemini_llm, brand_vectorstore)
226
-
227
  print("Pipeline initialized successfully!")
228
  except Exception as e:
229
  print(f"Error initializing pipeline: {str(e)}")
 
 
 
 
223
  gemini_llm = LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
224
  wellness_rag_chain = build_rag_chain(gemini_llm, wellness_vectorstore)
225
  brand_rag_chain = build_rag_chain(gemini_llm, brand_vectorstore)
226
+
227
  print("Pipeline initialized successfully!")
228
  except Exception as e:
229
  print(f"Error initializing pipeline: {str(e)}")
230
+
231
+ def run_with_chain(query: str) -> str:
232
+ return run_pipeline(query)