NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-22 16:48:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

assertions

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-20 02:16:49 -05:00
parent 559d649a26
commit e615eb1f50
3 changed files with 86 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

23
atroposlib/envs/server_handling/openai_server.py
View file

@ -165,17 +165,12 @@ def resolve_openai_configs(
"""
from atroposlib.envs.server_handling.server_manager import ServerBaseline
print(f"[RESOLVE DEBUG] default_server_configs type = {type(default_server_configs)}")
print(f"[RESOLVE DEBUG] openai_config_dict = {openai_config_dict}")
openai_full_prefix = f"{OPENAI_NAMESPACE}{NAMESPACE_SEP}"
openai_yaml_config = yaml_config.get(OPENAI_NAMESPACE, None)
openai_cli_config = {
k: v for k, v in cli_passed_flags.items() if k.startswith(openai_full_prefix)
}
print(f"[RESOLVE DEBUG] openai_cli_config = {openai_cli_config}")
is_multi_server_yaml = (
isinstance(openai_yaml_config, list) and len(openai_yaml_config) >= 2
)
@ -185,15 +180,6 @@ def resolve_openai_configs(
and len(default_server_configs) >= 2
)
print(
"[RESOLVE DEBUG] is_multi_server_yaml="
f"{is_multi_server_yaml}, is_multi_server_default={is_multi_server_default}"
)
print(
"[RESOLVE DEBUG] isinstance(default_server_configs, ServerBaseline) = "
f"{isinstance(default_server_configs, ServerBaseline)}"
)
if (is_multi_server_yaml or is_multi_server_default) and openai_cli_config:
raise FailedExecutionException(
message=f"CLI overrides for OpenAI settings (--{openai_full_prefix}*) are not supported "
@ -203,7 +189,6 @@ def resolve_openai_configs(
)
if is_multi_server_yaml:
print("[RESOLVE DEBUG] Taking multi-server YAML path")
logger.info(
f"Using multi-server configuration defined in YAML under '{OPENAI_NAMESPACE}'."
)
@ -215,7 +200,6 @@ def resolve_openai_configs(
) from e
elif isinstance(default_server_configs, APIServerConfig):
# Check APIServerConfig BEFORE ServerBaseline since APIServerConfig inherits from ServerBaseline
print("[RESOLVE DEBUG] Taking APIServerConfig merged path")
logger.info("Using single OpenAI server configuration based on merged settings (default/YAML/CLI).")
try:
final_openai_config = APIServerConfig(**openai_config_dict)
@ -227,15 +211,12 @@ def resolve_openai_configs(
server_configs = final_openai_config
elif isinstance(default_server_configs, ServerBaseline):
# Pure ServerBaseline (not APIServerConfig) - no CLI overrides possible
print("[RESOLVE DEBUG] Taking ServerBaseline path")
logger.info("Using ServerBaseline configuration.")
server_configs = default_server_configs
elif is_multi_server_default:
print("[RESOLVE DEBUG] Taking multi-server default path")
logger.info("Using default multi-server configuration (length >= 2).")
server_configs = default_server_configs
else:
print("[RESOLVE DEBUG] Taking single server merged path (fallback)")
logger.info(
"Using single OpenAI server configuration based on merged settings (default/YAML/CLI)."
)
@ -247,12 +228,9 @@ def resolve_openai_configs(
f"Merged Dict: {openai_config_dict}"
) from e
print(f"[RESOLVE DEBUG] final_openai_config = {final_openai_config}")
if isinstance(default_server_configs, APIServerConfig):
print("[RESOLVE DEBUG] Returning final_openai_config directly")
server_configs = final_openai_config
elif isinstance(default_server_configs, list):
print("[RESOLVE DEBUG] Returning [final_openai_config]")
server_configs = [final_openai_config]
else:
logger.warning(
@ -261,5 +239,4 @@ def resolve_openai_configs(
)
server_configs = [final_openai_config]
print(f"[RESOLVE DEBUG] Returning server_configs = {server_configs}")
return server_configs