eeshwar143 commited on
Commit
5d2ef04
·
1 Parent(s): edf1c38

Handle fallback clients without connect method

Browse files
Files changed (1) hide show
  1. inference.py +5 -1
inference.py CHANGED
@@ -225,7 +225,11 @@ def heuristic_action(observation: SupportQueueObservation) -> SupportQueueAction
225
  async def build_env() -> SupportQueueEnv:
226
  if ENV_BASE_URL:
227
  env = SupportQueueEnv(base_url=ENV_BASE_URL)
228
- await env.connect()
 
 
 
 
229
  return env
230
  return await SupportQueueEnv.from_docker_image(LOCAL_IMAGE_NAME or "support-queue-openenv")
231
 
 
225
  async def build_env() -> SupportQueueEnv:
226
  if ENV_BASE_URL:
227
  env = SupportQueueEnv(base_url=ENV_BASE_URL)
228
+ connect = getattr(env, "connect", None)
229
+ if callable(connect):
230
+ maybe_coro = connect()
231
+ if asyncio.iscoroutine(maybe_coro):
232
+ await maybe_coro
233
  return env
234
  return await SupportQueueEnv.from_docker_image(LOCAL_IMAGE_NAME or "support-queue-openenv")
235