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

format spaces in build_server_utils.py

Browse Source
This commit is contained in:
goulustis
2026-03-04 15:20:32 +08:00
parent bb6d98c9f4
commit 9284e8f928
1 changed files with 41 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
lang_agent/front_api/build_server_utils.py
View File

@@ -18,9 +18,10 @@ def opt_to_config(save_path: str, *nargs):
cwd=_PROJECT_ROOT,
)
def _build_and_load_pipeline_config(pipeline_id: str,
pipeline_config_dir: str,
cmd: List[str]):
def _build_and_load_pipeline_config(
pipeline_id: str, pipeline_config_dir: str, cmd: List[str]
):
save_config_f = osp.join(pipeline_config_dir, f"{pipeline_id}.yml")
opt_to_config(save_config_f, *cmd)
@@ -28,53 +29,55 @@ def _build_and_load_pipeline_config(pipeline_id: str,
return load_tyro_conf(save_config_f)
def update_pipeline_registry(pipeline_id:str, # the agent name -- xiaozhan, blueberry
prompt_set:str, # the version of the prompt for xiaozhan/blueberry
graph_id: str, # what type of graph is this pipeline.
config_file: str,
llm_name: str,
enabled: bool = True,
registry_f:str="configs/pipeline_registry.json"):
def update_pipeline_registry(
pipeline_id: str,
graph_id: str,
config_file: str,
llm_name: str,
enabled: bool = True,
registry_f: str = "configs/pipeline_registry.json",
):
if not osp.isabs(registry_f):
registry_f = osp.join(_PROJECT_ROOT, registry_f)
os.makedirs(osp.dirname(registry_f), exist_ok=True)
if not osp.exists(registry_f):
with open(registry_f, "w", encoding="utf-8") as f:
json.dump({"routes": {}, "api_keys": {}}, f, indent=4)
json.dump({"pipelines": {}, "api_keys": {}}, f, indent=4)
with open(registry_f, "r") as f:
registry = json.load(f)
routes: Dict[str, Dict[str, Any]] = registry.setdefault("routes", {})
route = routes.setdefault(pipeline_id, {})
route["enabled"] = bool(enabled)
route["config_file"] = config_file
route["prompt_pipeline_id"] = prompt_set
route["graph_id"] = graph_id
route["overrides"] = {"llm_name": llm_name}
pipelines: Dict[str, Dict[str, Any]] = registry.setdefault("pipelines", {})
pipeline = pipelines.setdefault(pipeline_id, {})
pipeline["enabled"] = bool(enabled)
pipeline["config_file"] = config_file
pipeline["graph_id"] = graph_id
pipeline["overrides"] = {"llm_name": llm_name}
with open(registry_f, "w", encoding="utf-8") as f:
json.dump(registry, f, indent=4)
def build_route(pipeline_id:str,
prompt_set:str,
tool_keys:List[str],
api_key: str,
llm_name:str="qwen-plus",
pipeline_config_dir="configs/pipelines"):
def build_route(
pipeline_id: str,
prompt_set: str,
tool_keys: List[str],
api_key: str,
llm_name: str = "qwen-plus",
pipeline_config_dir="configs/pipelines",
):
cmd_opt = [
"route", # ------------
"route", # ------------
"--llm-name", llm_name,
"--api-key", api_key,
"--pipeline-id", pipeline_id,
"--prompt-set-id", prompt_set,
"tool_node", # ------------
"tool_node", # ------------
"--llm-name", llm_name,
"--api-key", api_key,
"--pipeline-id", pipeline_id,
"--prompt-set-id", prompt_set,
]
]
if tool_keys:
cmd_opt.extend(
@@ -84,14 +87,16 @@ def build_route(pipeline_id:str,
return _build_and_load_pipeline_config(pipeline_id, pipeline_config_dir, cmd_opt)
def build_react(pipeline_id:str,
prompt_set:str,
tool_keys:List[str],
api_key: str,
llm_name:str="qwen-plus",
pipeline_config_dir="configs/pipelines"):
def build_react(
pipeline_id: str,
prompt_set: str,
tool_keys: List[str],
api_key: str,
llm_name: str = "qwen-plus",
pipeline_config_dir="configs/pipelines",
):
cmd_opt = [
"react", # ------------
"react", # ------------
"--llm-name", llm_name,
"--api-key", api_key,
"--pipeline-id", pipeline_id,