Selaa lähdekoodia

Merge pull request #94 from mwmbl/rishabh-port-configuration

Allow configuration of port
Rishabh Singh Ahluwalia 2 vuotta sitten
vanhempi
commit
38a5dbbf3c
1 muutettua tiedostoa jossa 2 lisäystä ja 1 poistoa
  1. 2 1
      mwmbl/main.py

+ 2 - 1
mwmbl/main.py

@@ -27,6 +27,7 @@ def setup_args():
     parser = argparse.ArgumentParser(description="Mwmbl API server and background task processor")
     parser.add_argument("--num-pages", type=int, help="Number of pages of memory (4096 bytes) to use for the index", default=2560)
     parser.add_argument("--data", help="Path to the data folder for storing index and cached batches", default="./devdata")
+    parser.add_argument("--port", type=int, help="Port for the server to listen at", default=5000)
     parser.add_argument("--background", help="Enable running the background tasks to process batches",
                         action='store_true')
     args = parser.parse_args()
@@ -70,7 +71,7 @@ def run():
         app.include_router(crawler_router)
 
         # Initialize uvicorn server using global app instance and server config params
-        uvicorn.run(app, host="0.0.0.0", port=5000)
+        uvicorn.run(app, host="0.0.0.0", port=args.port)
 
 
 if __name__ == "__main__":