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

remove override support

Browse Source
This commit is contained in:
goulustis
2026-03-07 14:50:48 +08:00
parent 3306da038d
commit 5358b46113
1 changed files with 5 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lang_agent/components/server_pipeline_manager.py
View File

@@ -55,7 +55,7 @@ class ServerPipelineManager:
parsed_specs[pipeline_id] = { parsed_specs[pipeline_id] = {
"enabled": bool(spec.get("enabled", True)), "enabled": bool(spec.get("enabled", True)),
"config_file": spec.get("config_file"), "config_file": spec.get("config_file"),
"overrides": spec.get("overrides", {}), "llm_name": spec.get("llm_name"),
} }
if not parsed_specs: if not parsed_specs:
raise ValueError("pipeline registry must define at least one pipeline.") raise ValueError("pipeline registry must define at least one pipeline.")
@@ -152,7 +152,7 @@ class ServerPipelineManager:
) )
config_file = spec.get("config_file") config_file = spec.get("config_file")
overrides = spec.get("overrides", {}) registry_llm_name = spec.get("llm_name")
if config_file: if config_file:
loaded_cfg = load_tyro_conf(self._resolve_config_path(config_file)) loaded_cfg = load_tyro_conf(self._resolve_config_path(config_file))
if hasattr(loaded_cfg, "setup"): if hasattr(loaded_cfg, "setup"):
@@ -166,19 +166,11 @@ class ServerPipelineManager:
) )
else: else:
cfg = copy.deepcopy(self.default_config) cfg = copy.deepcopy(self.default_config)
if not isinstance(overrides, dict): if registry_llm_name is not None and hasattr(cfg, "llm_name"):
raise ValueError( setattr(cfg, "llm_name", registry_llm_name)
f"pipeline `overrides` for `{pipeline_id}` must be an object."
)
for key, value in overrides.items():
if not hasattr(cfg, key):
raise ValueError(
f"unknown override field `{key}` for pipeline `{pipeline_id}`"
)
setattr(cfg, key, value)
p = cfg.setup() p = cfg.setup()
llm_name = getattr(cfg, "llm_name", "unknown-model") llm_name = str(getattr(cfg, "llm_name", registry_llm_name or "unknown-model"))
return p, llm_name return p, llm_name
def _authorize(self, api_key: str, pipeline_id: str) -> None: def _authorize(self, api_key: str, pipeline_id: str) -> None: