Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
goulustis
2026-03-03 14:11:07 +08:00
parent 6b0e50c532
commit af16b87b0e
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/gen_arch_img.py → scripts/misc_tasks.py
View File

@@ -1,17 +1,26 @@
from lang_agent.graphs import ReactGraphConfig, ReactGraph, RoutingConfig,RoutingGraph from lang_agent.graphs import ReactGraphConfig, ReactGraph, RoutingConfig,RoutingGraph
from lang_agent.pipeline import PipelineConfig
from lang_agent.base import GraphBase from lang_agent.base import GraphBase
import os.path as osp import os.path as osp
import os
from tqdm import tqdm 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()] confs:GraphBase = [ReactGraphConfig(), RoutingConfig()]
for conf in tqdm(confs): for conf in tqdm(confs):
graph:GraphBase = conf.setup() graph:GraphBase = conf.setup()
img = graph.show_graph(ret_img=True) img = graph.show_graph(ret_img=True)
img.save(osp.join(save_dir, f"arch_{conf.__class__.__name__}.png")) 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__": if __name__ == "__main__":
main() # gen_arch_imgs()
tyro.cli(make_save_conf)