Spaces:
Sleeping
Sleeping
eeshwar143 commited on
Commit ·
5d2ef04
1
Parent(s): edf1c38
Handle fallback clients without connect method
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|