From af16b87b0e6633993adaed08570499c1d2d05a56 Mon Sep 17 00:00:00 2001 From: goulustis Date: Tue, 3 Mar 2026 14:11:07 +0800 Subject: [PATCH] rename --- scripts/{gen_arch_img.py => misc_tasks.py} | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) rename scripts/{gen_arch_img.py => misc_tasks.py} (50%) diff --git a/scripts/gen_arch_img.py b/scripts/misc_tasks.py similarity index 50% rename from scripts/gen_arch_img.py rename to scripts/misc_tasks.py index 1575b6b..595a85c 100644 --- a/scripts/gen_arch_img.py +++ b/scripts/misc_tasks.py @@ -1,17 +1,26 @@ from lang_agent.graphs import ReactGraphConfig, ReactGraph, RoutingConfig,RoutingGraph +from lang_agent.pipeline import PipelineConfig from lang_agent.base import GraphBase import os.path as osp +import os from tqdm import tqdm +import yaml +import tyro -def main(): +def gen_arch_imgs(save_dir="frontend/assets/images/graph_arch"): - save_dir = osp.join(osp.dirname(osp.dirname(__file__)), "frontend/assets/images/graph_arch") + save_dir = osp.join(osp.dirname(osp.dirname(__file__)), save_dir) confs:GraphBase = [ReactGraphConfig(), RoutingConfig()] for conf in tqdm(confs): graph:GraphBase = conf.setup() img = graph.show_graph(ret_img=True) img.save(osp.join(save_dir, f"arch_{conf.__class__.__name__}.png")) +def make_save_conf(pipeline_config:PipelineConfig, save_path:str): + os.makedirs(osp.dirname(save_path), exist_ok=True) + pipeline_config.save_config(save_path) + if __name__ == "__main__": - main() \ No newline at end of file + # gen_arch_imgs() + tyro.cli(make_save_conf) \ No newline at end of file