Improve sglang router (#2148)

This commit is contained in:
Byron Hsu
2024-11-23 17:34:24 -08:00
committed by GitHub
parent 7921690893
commit 32293a299c
3 changed files with 9 additions and 3 deletions

View File

@@ -3,6 +3,11 @@
name: Release SGLang Router to PyPI name: Release SGLang Router to PyPI
on: on:
push:
branches:
- main
paths:
- rust/pyproject.toml"
workflow_dispatch: workflow_dispatch:
jobs: jobs:

View File

@@ -2,6 +2,7 @@ import argparse
import copy import copy
import multiprocessing as mp import multiprocessing as mp
import os import os
import random
import signal import signal
import sys import sys
import time import time
@@ -11,7 +12,7 @@ import requests
from sglang_router.launch_router import RouterArgs, launch_router from sglang_router.launch_router import RouterArgs, launch_router
from sglang.srt.server import launch_server from sglang.srt.server import launch_server
from sglang.srt.server_args import ServerArgs, prepare_server_args from sglang.srt.server_args import ServerArgs
from sglang.srt.utils import is_port_available from sglang.srt.utils import is_port_available
from sglang.utils import get_exception_traceback from sglang.utils import get_exception_traceback
@@ -94,7 +95,7 @@ def find_available_ports(base_port: int, count: int) -> List[int]:
while len(available_ports) < count: while len(available_ports) < count:
if is_port_available(current_port): if is_port_available(current_port):
available_ports.append(current_port) available_ports.append(current_port)
current_port += 1 current_port += random.randint(100, 1000)
return available_ports return available_ports

View File

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
[project] [project]
name = "sglang-router" name = "sglang-router"
version = "0.0.6" version = "0.0.7"
description = "SGLang router is a standalone module implemented in Rust to achieve data parallelism across SGLang instances." description = "SGLang router is a standalone module implemented in Rust to achieve data parallelism across SGLang instances."
authors = [{name = "Byron Hsu", email = "byronhsu1230@gmail.com"}] authors = [{name = "Byron Hsu", email = "byronhsu1230@gmail.com"}]
requires-python = ">=3.8" requires-python = ">=3.8"