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

clean

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-03-17 12:23:35 -04:00
parent 7aba0d3fc8
commit 79baac1ea7
2 changed files with 10 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -402,16 +402,12 @@ def resolve_openai_configs(
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)."
)
logger.info("Using single OpenAI server configuration.")
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}"
f"Error creating final OpenAI configuration: {e}"
) from e
server_configs = [final_openai_config]
elif isinstance(default_server_configs, ServerBaseline):
@ -421,15 +417,12 @@ def resolve_openai_configs(
logger.info("Using default multi-server configuration (length >= 2).")
server_configs = default_server_configs
else:
logger.info(
"Using single OpenAI server configuration based on merged settings (default/YAML/CLI)."
)
logger.info("Using single OpenAI server configuration.")
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}"
f"Error creating final OpenAI configuration: {e}"
) from e
if isinstance(default_server_configs, APIServerConfig):
@ -439,7 +432,7 @@ def resolve_openai_configs(
else:
logger.warning(
f"Unexpected type for default_server_configs: {type(default_server_configs)}. "
"Proceeding with single OpenAI server configuration based on merged settings."
"Proceeding with single OpenAI server configuration."
)
server_configs = [final_openai_config]