use CENTML_SERVER_URL instead of IP and PORT (#53)

* Use CENTML_SERVER_URL instead of IP and PORT

Author: destefy

README.md

@@ -35,7 +35,7 @@ To run the server locally, you can use the following CLI command:
 centml server
 ```
 By default, the server will run at the URL `http://0.0.0.0:8090`. \
-You can change this by setting the environment variables `CENTML_SERVER_IP` and `CENTML_SERVER_PORT`
+You can change this by setting the environment variable `CENTML_SERVER_URL`
 
 
 Then, within your python script include the following:
@@ -55,7 +55,7 @@ output = compiled_model(inputs)
 ```
 Note that the centml backend compiler is non-blocking. This means it that until the server returns the compiled model, your python script will use the uncompiled model to generate the output.
 
-Again, make sure your script's environment sets the environment variables `CENTML_SERVER_IP` and `CENTML_SERVER_PORT` to communicate with the desired server.
+Again, make sure your script's environment sets `CENTML_SERVER_URL` to communicate with the desired server.
 
 
 ### Tests

centml/compiler/config.py

@@ -14,9 +14,8 @@ class Config:
     COMPILING_SLEEP_TIME: int = 15
 
     CACHE_PATH: str = os.getenv("CENTML_CACHE_DIR", default=os.path.expanduser("~/.cache/centml"))
-    SERVER_IP: str = os.getenv("CENTML_SERVER_IP", default="0.0.0.0")
-    SERVER_PORT: str = os.getenv("CENTML_SERVER_PORT", default="8090")
-    SERVER_URL: str = f"http://{SERVER_IP}:{SERVER_PORT}"
+
+    SERVER_URL: str = os.getenv("CENTML_SERVER_URL", default="http://0.0.0.0:8090")
 
     BACKEND_BASE_PATH: str = os.path.join(CACHE_PATH, "backend")
     SERVER_BASE_PATH: str = os.path.join(CACHE_PATH, "server")

centml/compiler/server.py

@@ -1,6 +1,7 @@
 import io
 import os
 from http import HTTPStatus
+from urllib.parse import urlparse
 import logging
 import uvicorn
 import torch
@@ -107,7 +108,8 @@ async def download_handler(model_id: str):
 
 
 def run():
-    uvicorn.run(app, host=config_instance.SERVER_IP, port=int(config_instance.SERVER_PORT))
+    parsed = urlparse(config_instance.SERVER_URL)
+    uvicorn.run(app, host=parsed.hostname, port=parsed.port)
 
 
 if __name__ == "__main__":