diff --git a/src/server/dcp_server/config.yaml b/src/server/dcp_server/config.yaml index 5652469..1facae5 100644 --- a/src/server/dcp_server/config.yaml +++ b/src/server/dcp_server/config.yaml @@ -8,7 +8,8 @@ "runner_name": "bento_runner", "bento_model_path": "cells", "service_name": "data-centric-platform", - "port": 7010 + "port": 7010, + "timeout": 1000, }, "model": { diff --git a/src/server/dcp_server/main.py b/src/server/dcp_server/main.py index 9c149b5..339c2be 100644 --- a/src/server/dcp_server/main.py +++ b/src/server/dcp_server/main.py @@ -23,7 +23,7 @@ def main() -> None: "service", config_path=path.join(dir_name, "config.yaml") ) port = str(service_config["port"]) - + timeout = str(service_config["timeout"]) subprocess.run( [ "bentoml", @@ -33,6 +33,7 @@ def main() -> None: "service:svc", "--reload", "--port=" + port, + "--timeout=" + timeout ] )