Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- 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)
|