add arg option host

Signed-off-by: Ceng2333 <[email protected]>

Author: Ceng23333

enhanced_babysitter.py

@@ -37,10 +37,11 @@
 logger = logging.getLogger(__name__)
 
 class EnhancedServiceBabysitter:
-    def __init__(self, config_file, port, service_name=None,
+    def __init__(self, config_file, host, port, service_name=None,
                  registry_url="http://localhost:8081", router_url="http://localhost:8080",
                  max_restarts=10, restart_delay=5, heartbeat_interval=30):
         self.config_file = config_file
+        self.host = host
         self.port = port
         self.service_name = service_name or f"infinilm-rust-{port}"
         self.registry_url = registry_url
@@ -65,9 +66,9 @@ def __init__(self, config_file, port, service_name=None,
         # Service info
         self.service_info = ServiceInfo(
             name=self.service_name,
-            host="localhost",
+            host=self.host,
             port=self.babysitter_port,
-            url=f"http://localhost:{self.babysitter_port}",
+            url=f"http://{self.host}:{self.babysitter_port}",
             status="running",
             metadata={
                 "type": "rust",
@@ -144,9 +145,9 @@ async def info_handler(self, request):
         """Service info endpoint"""
         return web.json_response({
             "name": self.service_name,
-            "host": "localhost",
+            "host": self.host,
             "port": self.babysitter_port,
-            "url": f"http://localhost:{self.babysitter_port}",
+            "url": f"http://{self.host}:{self.babysitter_port}",
             "metadata": self.service_info.metadata,
             "xtask_port": self.xtask_port,
             "uptime": time.time() - getattr(self, 'start_time', time.time())
@@ -216,9 +217,9 @@ def register_xtask_with_registry(self) -> bool:
 
         xtask_service_info = ServiceInfo(
             name=f"{self.service_name}-xtask",
-            host="localhost",
+            host=self.host,
             port=self.xtask_port,
-            url=f"http://localhost:{self.xtask_port}",
+            url=f"http://{self.host}:{self.xtask_port}",
             status="running",
             metadata={
                 "type": "openai-api",
@@ -345,7 +346,7 @@ def start_service(self):
     def check_service_health(self) -> bool:
         """Check if the service is responding to health checks"""
         try:
-            response = requests.get(f"http://localhost:{self.port}/health", timeout=5)
+            response = requests.get(f"http://{self.host}:{self.port}/health", timeout=5)
             return response.status_code == 200
         except:
             return False
@@ -396,6 +397,7 @@ async def run(self):
         logger.info("Starting Enhanced InfiniLM service babysitter...")
         logger.info(f"Service name: {self.service_name}")
         logger.info(f"Config file: {self.config_file}")
+        logger.info(f"Host: {self.host}")
         logger.info(f"Port: {self.port}")
         logger.info(f"Registry URL: {self.registry_url}")
         logger.info(f"Router URL: {self.router_url}")
@@ -446,6 +448,7 @@ def main():
 
     parser = argparse.ArgumentParser(description="Enhanced Babysitter for InfiniLM service with Registry integration")
     parser.add_argument("config_file", help="Path to the service configuration file (e.g., service.toml)")
+    parser.add_argument("--host", default="localhost", help="Host to run the service on (default: localhost)")
     parser.add_argument("-p", "--port", type=int, default=5000, help="Port to run the service on (default: 5000)")
     parser.add_argument("-n", "--name", help="Service name (default: infinilm-rust-{port})")
     parser.add_argument("--registry", default="http://localhost:8081", help="Registry URL (default: http://localhost:8081)")
@@ -464,6 +467,7 @@ def main():
     # Create and run enhanced babysitter
     babysitter = EnhancedServiceBabysitter(
         config_file=args.config_file,
+        host=args.host,
         port=args.port,
         service_name=args.name,
         registry_url=args.registry,
@@ -477,3 +481,5 @@ def main():
 
 if __name__ == "__main__":
     main()
+
+## python3 enhanced_babysitter.py service156_generated.toml --port 5002 --name service156 --registry http://172.22.163.154:8081 --router http://172.22.163.154:8080 --host 172.22.163.156