File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed
Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change 1414# See the License for the specific language governing permissions and
1515# limitations under the License.
1616#
17+ import json
1718from datetime import datetime
1819from pathlib import Path
1920from typing import Optional
2021from uuid import UUID
2122
2223import click
24+ import yaml
2325from zenml .client import Client
2426
2527from 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
You can’t perform that action at this time.
0 commit comments