Add Elasticsearch dependency to production and staging configs

zenml-io · Nov 13, 2024 · 62d3e01 · 62d3e01
1 parent b49d425
commit 62d3e01
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/llm-complete-guide/configs/production/rag.yaml b/llm-complete-guide/configs/production/rag.yaml
@@ -17,6 +17,8 @@ settings:
       - pygithub
       - rerankers[flashrank]
       - matplotlib
+      - elasticsearch
+
     environment:
       ZENML_PROJECT_SECRET_NAME: llm_complete
       ZENML_ENABLE_RICH_TRACEBACK: FALSE

diff --git a/llm-complete-guide/configs/staging/rag.yaml b/llm-complete-guide/configs/staging/rag.yaml
@@ -17,7 +17,8 @@ settings:
       - pygithub
       - rerankers[flashrank]
       - matplotlib
-
+      - elasticsearch
+
     environment:
       ZENML_PROJECT_SECRET_NAME: llm_complete
       ZENML_ENABLE_RICH_TRACEBACK: FALSE