diff --git a/src/options/generate/generator.py b/src/options/generate/generator.py index eae74a3..9a62e34 100644 --- a/src/options/generate/generator.py +++ b/src/options/generate/generator.py @@ -542,7 +542,7 @@ Test scenario: def replace_with_gpt_3_5_turbo_if_possible(pkg): if pkg in ['allennlp', 'bertopic', 'fasttext', 'flair', 'gensim', 'nltk', 'pattern', 'polyglot', 'pytorch-transformers', 'rasa', 'sentence-transformers', - 'spacy', 'stanza', 'textblob', 'textstat', 'transformers']: + 'spacy', 'stanza', 'summarizer', 'textblob', 'textstat', 'transformers']: return 'gpt_3_5_turbo_api' return pkg diff --git a/src/options/generate/templates_system.py b/src/options/generate/templates_system.py index 70a98d1..1d31a5b 100644 --- a/src/options/generate/templates_system.py +++ b/src/options/generate/templates_system.py @@ -139,7 +139,7 @@ contains example: yes () ```json {{ "input": "", - "assertion": "the output is of type " + "assertion": "the output contains the result that is of type " }} ``` diff --git a/src/options/generate/templates_user.py b/src/options/generate/templates_user.py index 873ab81..9f3aeb9 100644 --- a/src/options/generate/templates_user.py +++ b/src/options/generate/templates_user.py @@ -268,7 +268,22 @@ You are given the following files: {all_files_string} To solve this error, you should determine the list of packages that need to be installed via `apt-get install` in the Dockerfile. -Output them as a white space separated list:''' +Output the apt-get packages that need to me placed at {{apt_get_packages}} as json in the following format: +**apt-get-packages.json** +```json +{{"packages": ["", ""]}} +``` +Example for the following requirements.txt file: +**requirements.txt** +``` +numpy==1.19.5 +fitz +``` +The output would be: +**apt-get-packages.json** +```json +{{"packages": []}} +```''' ) @@ -433,7 +448,7 @@ Example for the case where the example is already mentioned in the refined descr ```json {{ "input": "", - "assertion": "the output is of type " + "assertion": "the output contains the result that is of type " }} ``` Note that your response must be either prompt.json or final.json. You must not write both.