From 83a8a7cb4fc6ffb85bb47237c29c392b222ba477 Mon Sep 17 00:00:00 2001 From: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com> Date: Thu, 25 May 2023 23:00:34 +0000 Subject: [PATCH] docs(codegen): make sure the generated dostring is correct Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com> Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com> --- src/openllm/utils/codegen.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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) )