bug fixes

This commit is contained in:
2026-03-04 10:30:22 +08:00
parent 6097e9c24a
commit 16916e183b

View File

@@ -8,8 +8,8 @@ from lang_agent.config.core_config import load_tyro_conf
def opt_to_config(save_path:str, *nargs): def opt_to_config(save_path:str, *nargs):
os.makedirs(osp.dirname(save_path), exist_ok=True) os.makedirs(osp.dirname(save_path), exist_ok=True)
subprocess.run(["python", "--save-path", save_path, *nargs]) subprocess.run(["python", "lang_agent/config/ty_build_config.py",
"--save-path", save_path, *nargs])
def _build_and_load_pipeline_config(pipeline_id: str, def _build_and_load_pipeline_config(pipeline_id: str,
pipeline_config_dir: str, pipeline_config_dir: str,
@@ -44,11 +44,9 @@ def build_route(pipeline_id:str,
prompt_set:str, prompt_set:str,
tool_keys:List[str], tool_keys:List[str],
api_key: str, api_key: str,
entry_pnt:str="fastapi_server/server_dashscope.py",
llm_name:str="qwen-plus", llm_name:str="qwen-plus",
pipeline_config_dir="configs/pipelines"): pipeline_config_dir="configs/pipelines"):
cmd = [ cmd_opt = [
"python", entry_pnt,
"route", # ------------ "route", # ------------
"--llm-name", llm_name, "--llm-name", llm_name,
"--api-key", api_key, "--api-key", api_key,
@@ -62,22 +60,20 @@ def build_route(pipeline_id:str,
] ]
if tool_keys: if tool_keys:
cmd.extend( cmd_opt.extend(
["--tool-manager-config.client-tool-manager.tool-keys", *tool_keys] ["--tool-manager-config.client-tool-manager.tool-keys", *tool_keys]
) )
return _build_and_load_pipeline_config(pipeline_id, pipeline_config_dir, cmd) return _build_and_load_pipeline_config(pipeline_id, pipeline_config_dir, cmd_opt)
def build_react(pipeline_id:str, def build_react(pipeline_id:str,
prompt_set:str, prompt_set:str,
tool_keys:List[str], tool_keys:List[str],
api_key: str, api_key: str,
entry_pnt:str="fastapi_server/server_dashscope.py",
llm_name:str="qwen-plus", llm_name:str="qwen-plus",
pipeline_config_dir="configs/pipelines"): pipeline_config_dir="configs/pipelines"):
cmd = [ cmd_opt = [
"python", entry_pnt,
"react", # ------------ "react", # ------------
"--llm-name", llm_name, "--llm-name", llm_name,
"--api-key", api_key, "--api-key", api_key,
@@ -85,11 +81,11 @@ def build_react(pipeline_id:str,
"--prompt-set-id", prompt_set, "--prompt-set-id", prompt_set,
] ]
if tool_keys: if tool_keys:
cmd.extend( cmd_opt.extend(
["--tool-manager-config.client-tool-manager.tool-keys", *tool_keys] ["--tool-manager-config.client-tool-manager.tool-keys", *tool_keys]
) )
return _build_and_load_pipeline_config(pipeline_id, pipeline_config_dir, cmd) return _build_and_load_pipeline_config(pipeline_id, pipeline_config_dir, cmd_opt)
# {pipeline_id: build_function} # {pipeline_id: build_function}