From cf091d04dda874525f328d590c78abda595c143e Mon Sep 17 00:00:00 2001 From: Hamza Tahir Date: Mon, 11 Nov 2024 22:50:28 +0100 Subject: [PATCH] Add pipeline_args to main function for caching configuration --- llm-complete-guide/gh_action_rag.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/llm-complete-guide/gh_action_rag.py b/llm-complete-guide/gh_action_rag.py index 79858c77..d1ce776f 100644 --- a/llm-complete-guide/gh_action_rag.py +++ b/llm-complete-guide/gh_action_rag.py @@ -99,6 +99,8 @@ def main( zenml_model_name (str): The ZenML model name. zenml_model_version (str): The ZenML model version. """ + pipeline_args = {"enable_cache": not no_cache} + client = Client() config_path = Path(__file__).parent / "configs" / config @@ -154,7 +156,7 @@ def main( name_id_or_prefix="LLM Complete (production)", allow_name_prefix_match=True, ) - except ZenKeyError: + except KeyError: if not event_source_id: raise RuntimeError( "An event source is required for this workflow."