From d9a654170cf47903e41a44e32a6c066f8703a997 Mon Sep 17 00:00:00 2001 From: goulustis Date: Sat, 28 Feb 2026 10:59:11 +0800 Subject: [PATCH] save arch img to frontend --- scripts/gen_arch_img.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 scripts/gen_arch_img.py diff --git a/scripts/gen_arch_img.py b/scripts/gen_arch_img.py new file mode 100644 index 0000000..1575b6b --- /dev/null +++ b/scripts/gen_arch_img.py @@ -0,0 +1,17 @@ +from lang_agent.graphs import ReactGraphConfig, ReactGraph, RoutingConfig,RoutingGraph +from lang_agent.base import GraphBase + +import os.path as osp +from tqdm import tqdm + +def main(): + + save_dir = osp.join(osp.dirname(osp.dirname(__file__)), "frontend/assets/images/graph_arch") + 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")) + +if __name__ == "__main__": + main() \ No newline at end of file