Add option to disable automatic client.py generation (#131)

Lightning-AI · Jun 11, 2024 · 65af629 · 65af629
1 parent ac38699
commit 65af629
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/litserve/server.py b/src/litserve/server.py
@@ -574,8 +574,9 @@ def generate_client_file(self):
         except Exception as e:
             print(f"Error copying file: {e}")
 
-    def run(self, port: Union[str, int] = 8000, log_level: str = "info", **kwargs):
-        self.generate_client_file()
+    def run(self, port: Union[str, int] = 8000, log_level: str = "info", generate_client_file: bool = True, **kwargs):
+        if generate_client_file:
+            self.generate_client_file()
 
         port_msg = f"port must be a value from 1024 to 65535 but got {port}"
         try: