Skip to content

Commit 40f501e

Browse files
committed
add parameter to configure uvicorn workers.
1 parent f1f8fce commit 40f501e

File tree

3 files changed

+14
-1
lines changed

3 files changed

+14
-1
lines changed

config.json

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,8 @@
5656
"host": "127.0.0.1",
5757
"port": 8080,
5858
"auto_open_webpage": true,
59-
"log_level": "INFO"
59+
"log_level": "INFO",
60+
"workers": 1
6061
},
6162
"agent": {
6263
"prompt": "",

oxygent/config.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ class Config:
9292
"port": 8080,
9393
"auto_open_webpage": True,
9494
"log_level": "INFO",
95+
"workers": 1,
9596
},
9697
"agent": {
9798
"prompt": "",
@@ -528,6 +529,16 @@ def set_server_log_level(cls, log_level):
528529
def get_server_log_level(cls):
529530
return cls.get_module_config("server", "log_level")
530531

532+
@classmethod
533+
def set_server_workers(cls, workers=None):
534+
if workers is None:
535+
workers = os.cpu_count() * 2 + 1
536+
cls.set_module_config("server", "workers", workers)
537+
538+
@classmethod
539+
def get_server_workers(cls):
540+
return cls.get_module_config("server", "workers")
541+
531542
""" agent """
532543

533544
@classmethod

oxygent/mas.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1045,6 +1045,7 @@ async def run_uvicorn():
10451045
port=port,
10461046
log_level=Config.get_server_log_level().lower(),
10471047
log_config=None,
1048+
workers=Config.set_server_workers(),
10481049
)
10491050
server = uvicorn.Server(config)
10501051

0 commit comments

Comments
 (0)