From 222ca015689fade8b71aea24e6d1d65cdaffca72 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= Date: Mon, 1 May 2023 10:51:12 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20test:=20level=202?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/options/generate/generator.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/options/generate/generator.py b/src/options/generate/generator.py index 7860a48..f9a6539 100644 --- a/src/options/generate/generator.py +++ b/src/options/generate/generator.py @@ -183,6 +183,7 @@ metas: }), file_name_purpose=REQUIREMENTS_FILE_NAME, file_name_s=[REQUIREMENTS_FILE_NAME], + parse_result_fn=self.parse_result_fn_requirements, tag_name=REQUIREMENTS_FILE_TAG, )[REQUIREMENTS_FILE_NAME] @@ -217,6 +218,16 @@ metas: docker_file_template = self.read_docker_template() return {DOCKER_FILE_NAME: docker_file_template.replace('{{apt_get_packages}}', '{apt_get_packages}').format(apt_get_packages=content_raw)} + def parse_result_fn_requirements(self, content_raw: str): + lines = content_raw.split('\n') + lines = [line for line in lines if not any([pkg in line for pkg in ['jina', 'docarray', 'openai', 'pytest']])] + content_modified = f'''jina==3.15.1.dev14 +docarray==0.21.0 +openai>=0.26.0 +pytest +{os.linesep.join(lines)}''' + return {REQUIREMENTS_FILE_NAME: content_modified} + def generate_playground(self, microservice_name, microservice_path): print_colored('', '\n\n############# Playground #############', 'blue')