diff --git a/llm-complete-guide/gh_action_rag.py b/llm-complete-guide/gh_action_rag.py index bfd3dbd3..30d3281a 100644 --- a/llm-complete-guide/gh_action_rag.py +++ b/llm-complete-guide/gh_action_rag.py @@ -21,6 +21,8 @@ import click import yaml +from zenml.enums import PluginSubType + from pipelines.llm_index_and_evaluate import llm_index_and_evaluate from zenml.client import Client from zenml import Model @@ -175,6 +177,8 @@ def main( }, service_account_id=service_account_id, auth_window=0, + flavor="builtin", + action_type=PluginSubType.PIPELINE_RUN ).id client.create_trigger( name="Production Trigger LLM-Complete",