Allow customizing server port

AI-Engineer-Foundation · Jul 19, 2023 · 2a64901 · 2a64901
1 parent 84b166c
commit 2a64901
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/agent/python/agent_protocol/agent.py b/agent/python/agent_protocol/agent.py
@@ -7,9 +7,6 @@
 from .dependencies import *
 from .models import Step, Task, TaskRequestBody
 
-config = Config()
-config.bind = ["localhost:8000"] # As an example configuration setting
-
 
 from .server import app
 from .models import (
@@ -156,5 +153,7 @@ def handle_task(handler: Callable[[TaskInput | None], Awaitable[StepHandler]]):
  return Agent
 
  @staticmethod
- def start():
+ def start(port: int = 8000):
+ config = Config()
+ config.bind = [f"localhost:{port}"] # As an example configuration setting
  trio.run(serve, app, config)