diff --git a/dev_gpt/constants.py b/dev_gpt/constants.py index 1a9c150..e78c84f 100644 --- a/dev_gpt/constants.py +++ b/dev_gpt/constants.py @@ -63,7 +63,7 @@ LANGUAGE_PACKAGES = [ 'allennlp', 'bertopic', 'gpt-3', 'fasttext', 'flair', 'gensim', 'nltk', 'openai', 'pattern', 'polyglot', 'pytorch-transformers', 'rasa', 'sentence-transformers', 'spacy', 'stanza', 'summarizer', 'sumy', 'textblob', 'textstat', 'transformers', - 'vadersentiment' + 'vadersentiment', 'language-tool-python' ] SEARCH_PACKAGES = [ diff --git a/dev_gpt/options/generate/chains/auto_refine_description.py b/dev_gpt/options/generate/chains/auto_refine_description.py index 1404e41..350cdce 100644 --- a/dev_gpt/options/generate/chains/auto_refine_description.py +++ b/dev_gpt/options/generate/chains/auto_refine_description.py @@ -12,6 +12,7 @@ def auto_refine_description(context): identity_parser, context_string=context_to_string(context) ) + context['request_schema'] = ask_gpt( generate_request_schema_prompt, optional_tripple_back_tick_parser, diff --git a/test/integration/test_generator.py b/test/integration/test_generator.py index 8213447..6c9a02a 100644 --- a/test/integration/test_generator.py +++ b/test/integration/test_generator.py @@ -93,7 +93,7 @@ def test_generation_level_2_svg(microservice_dir, mock_input_sequence): assert generator.generate() == 0 -@pytest.mark.parametrize('mock_input_sequence', [['y', 'yfinance.Ticker("MSFT").info']], indirect=True) +@pytest.mark.parametrize('mock_input_sequence', [['y', 'ticker = yf.Ticker(symbol); data = ticker.history(start=start_date, end=end_date); [row[\'Close\'] for row in data.to_dict(\'records\')]']], indirect=True) def test_generation_level_3(microservice_dir, mock_input_sequence): """ Requirements: