fix bug alloc server

modelscope · Sep 5, 2024 · 5d3bd26 · 5d3bd26
1 parent 145c4ba
commit 5d3bd26
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/src/agentscope/studio/_app.py b/src/agentscope/studio/_app.py
@@ -381,13 +381,18 @@ def _alloc_server() -> Response:
     # TODO: allocate based on server's cpu and memory usage
     # currently random select a server
     servers = _ServerTable.query.all()
+    if len(servers) == 0:
+        return jsonify({"status": "fail"})
     server = choice(servers)
-    return jsonify(
-        {
-            "host": server.host,
-            "port": server.port,
-        },
-    )
+    if RpcClient(host=server.host, port=server.port).is_alive():
+        return jsonify(
+            {
+                "host": server.host,
+                "port": server.port,
+            },
+        )
+    else:
+        return jsonify({"status": "fail"})
 
 
 @_app.route("/api/messages/push", methods=["POST"])