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

move server_type to serverbaseline instead so it can be used as well for server instantiation

Browse source
This commit is contained in:
dmahan93 2025-05-13 10:21:58 -05:00
parent 6fc356e76e
commit 8b864e9786
2 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
atroposlib/envs/server_handling/server_baseline.py
View file

@ -108,6 +108,9 @@ class ServerBaseline(BaseModel):
rolling_buffer_length: int = Field(
default=1000, description="Length of the rolling buffer to store metrics."
)
server_type: Literal["openai", "trl"] = Field(
default="openai", description="Type of server to use, openai or trl"
)
class APIServerConfig(ServerBaseline):
@ -117,9 +120,6 @@ class APIServerConfig(ServerBaseline):
api_key: Optional[str] = Field(default="", description="API key for the server.")
base_url: Optional[str] = Field(default="", description="Base URL for the server.")
server_type: Literal["openai", "trl"] = Field(
default="openai", description="Type of server to use, openai or trl"
)
n_kwarg_is_ignored: bool = Field(
default=False, description="Whether the n kwarg is ignored by this API server."
)