diff --git a/atroposlib/envs/server_handling/openai_server.py b/atroposlib/envs/server_handling/openai_server.py index ec3796f3..ddeafa0e 100644 --- a/atroposlib/envs/server_handling/openai_server.py +++ b/atroposlib/envs/server_handling/openai_server.py @@ -29,7 +29,7 @@ class OpenAIServer(APIServer): try: await self.openai.chat.completions.create( model=self.config.model_name, - prompt="hi", + messages=[{"role": "user", "content": "hi"}], max_tokens=1, ) self.server_healthy = True