NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-29 17:35:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

adding tests

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-03-13 17:23:40 -04:00
parent 6c564799bc
commit 1b8ff075c4
3 changed files with 84 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

17
atroposlib/envs/server_handling/vllm_server.py
View file

@ -401,6 +401,19 @@ def resolve_openai_configs(
raise FailedExecutionException(
f"Error parsing multi-server OpenAI configuration from YAML under '{OPENAI_NAMESPACE}': {e}"
) from e
elif isinstance(default_server_configs, APIServerConfig):
# Check APIServerConfig BEFORE ServerBaseline since APIServerConfig inherits from ServerBaseline
logger.info(
"Using single OpenAI server configuration based on merged settings (default/YAML/CLI)."
)
try:
final_openai_config = APIServerConfig(**openai_config_dict)
except Exception as e:
raise FailedExecutionException(
f"Error creating final OpenAI configuration from merged settings: {e}\n"
f"Merged Dict: {openai_config_dict}"
) from e
server_configs = [final_openai_config]
elif isinstance(default_server_configs, ServerBaseline):
logger.info("Using ServerBaseline configuration.")
server_configs = default_server_configs
@ -419,9 +432,7 @@ def resolve_openai_configs(
f"Merged Dict: {openai_config_dict}"
) from e
if isinstance(default_server_configs, APIServerConfig):
server_configs = [final_openai_config]
elif isinstance(default_server_configs, list):
if isinstance(default_server_configs, list):
server_configs = [final_openai_config]
else:
logger.warning(