From e99131973d4230e0fdfb8075396fc0b2a3279ca2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= Date: Mon, 1 May 2023 12:26:33 +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 | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/options/generate/generator.py b/src/options/generate/generator.py index 37eb171..60bfc76 100644 --- a/src/options/generate/generator.py +++ b/src/options/generate/generator.py @@ -187,6 +187,8 @@ metas: tag_name=REQUIREMENTS_FILE_TAG, )[REQUIREMENTS_FILE_NAME] + # I deactivated this because 3.5-turbo was halucinating packages that were not needed + # now, in the first iteration the default dockerfile is used # self.generate_and_persist_file( # section_title='Generate Dockerfile', # template=template_generate_apt_get_install, @@ -535,6 +537,6 @@ Test scenario: @staticmethod def replace_with_gpt_3_5_turbo_if_possible(pkg): - if pkg in ['nltk', 'textblob', 'spacy', 'transformers', 'textstat']: + if pkg in ['nltk', 'textblob', 'spacy', 'transformers', 'textstat', 'gensim']: return 'gpt_3_5_turbo_api' return pkg