Skip to content

Commit c40d7e0

Browse files
committed
Add config to action
1 parent 0eb3715 commit c40d7e0

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

llm-complete-guide/gh_action_rag.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,14 @@
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
1616
#
17+
import json
1718
from datetime import datetime
1819
from pathlib import Path
1920
from typing import Optional
2021
from uuid import UUID
2122

2223
import click
24+
import yaml
2325
from zenml.client import Client
2426

2527
from pipelines.llm_basic_rag import llm_basic_rag
@@ -75,6 +77,9 @@ def main(
7577
client = Client()
7678
config_path = Path(__file__).parent / "configs" / config
7779

80+
with (open(config_path,"r") as file):
81+
config = yaml.safe_load(file)
82+
7883
if create_template:
7984
# run pipeline
8085
run = llm_basic_rag.with_options(
@@ -90,7 +95,8 @@ def main(
9095
client.update_action(
9196
name_id_or_prefix=UUID(action_id),
9297
configuration={
93-
"template_id": str(rt.id)
98+
"template_id": str(rt.id),
99+
"run_config": config
94100
}
95101
)
96102

0 commit comments

Comments
 (0)