diff --git a/src/openllm/utils/codegen.py b/src/openllm/utils/codegen.py index b5a11b10..a1ad22a3 100644 --- a/src/openllm/utils/codegen.py +++ b/src/openllm/utils/codegen.py @@ -28,13 +28,14 @@ logger = logging.getLogger(__name__) def write_service(model_name: str, target_path: str, llm_fs: FS): + logger.debug("Generating service for %s to %s", model_name, target_path) service_file = Path(__file__).parent.parent / "_service.py" with open(service_file.__fspath__(), "rb") as f: node = ast.parse(f.read()) generator = ServiceGenerator(model_name) generator.visit(node) llm_fs.writetext( - target_path, f"# GENERATED BY 'openllm build {model_name}'. DO NOT EDIT\n" + "".join(generator.result) + target_path, f"# GENERATED BY 'openllm bundle {model_name}'. DO NOT EDIT\n" + "".join(generator.result) )