rishiad commited on
Commit
6c27519
·
unverified ·
1 Parent(s): c172525

feat: bind to 0.0.0.0

Browse files
Files changed (1) hide show
  1. main.py +5 -1
main.py CHANGED
@@ -26,6 +26,9 @@ def setup_logging(level=logging.INFO):
26
  def main():
27
  """Main entry point."""
28
  parser = argparse.ArgumentParser(description="Start the agent server")
 
 
 
29
  parser.add_argument(
30
  "--port", type=int, default=8000, help="Port to bind the server to"
31
  )
@@ -51,7 +54,7 @@ def main():
51
 
52
  # Start the server
53
  try:
54
- start_server(agent, "0.0.0.0", args.port)
55
  except KeyboardInterrupt:
56
  logger.info("Server stopped by user")
57
  except Exception as e:
@@ -61,3 +64,4 @@ def main():
61
 
62
  if __name__ == "__main__":
63
  main()
 
 
26
  def main():
27
  """Main entry point."""
28
  parser = argparse.ArgumentParser(description="Start the agent server")
29
+ parser.add_argument(
30
+ "--host", type=str, default="0.0.0.0", help="Host to bind the server to"
31
+ )
32
  parser.add_argument(
33
  "--port", type=int, default=8000, help="Port to bind the server to"
34
  )
 
54
 
55
  # Start the server
56
  try:
57
+ start_server(agent, args.host, args.port)
58
  except KeyboardInterrupt:
59
  logger.info("Server stopped by user")
60
  except Exception as e:
 
64
 
65
  if __name__ == "__main__":
66
  main()
67
+