From a7d5cec0bdd62c8f533fe12ceadf6f1b48d8ded7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= <hoenicke.florian@gmail.com>
Date: Fri, 26 May 2023 12:58:59 +0200
Subject: [PATCH] =?UTF-8?q?=E2=8F=AA=20fix:=20fix=20bring=20back=20the=20t?=
 =?UTF-8?q?est=20description=20for=20func=20generation?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dev_gpt/constants.py       |  2 +-
 test/unit/test_packages.py | 17 ++++++++++++++++-
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/dev_gpt/constants.py b/dev_gpt/constants.py
index 77403ef..06f194b 100644
--- a/dev_gpt/constants.py
+++ b/dev_gpt/constants.py
@@ -53,7 +53,7 @@ BLACKLISTED_PACKAGES = [
     'clearbit'  # because of installation issues on latest version
 ]
 UNNECESSARY_PACKAGES = [
-    'fastapi', 'uvicorn', 'starlette'  # because the wrappers are used instead
+    'flask', 'fastapi', 'uvicorn', 'starlette'  # because the wrappers are used instead
 ]
 
 LANGUAGE_PACKAGES = [
diff --git a/test/unit/test_packages.py b/test/unit/test_packages.py
index 73a3cd0..af6c4ca 100644
--- a/test/unit/test_packages.py
+++ b/test/unit/test_packages.py
@@ -17,4 +17,19 @@ def test_package_parsing():
         [],
         ['gpt_3_5_turbo'],
     ]):
-        assert set(parsed) == set(expected)
\ No newline at end of file
+        assert set(parsed) == set(expected)
+
+def test_package_parsing_blacklist():
+    packages_json_string = '''\
+    [
+      ["Flask"],
+      [],
+      []
+    ]'''
+
+    parsed_packages = Generator.process_packages_json_string(packages_json_string)
+    assert parsed_packages == [
+        [],
+        [],
+        [],
+    ]
\ No newline at end of file