Merge pull request #1007 from drikusroor/fix-flake8-issues

Fix flake8 issues

.github/workflows/ci.yml

@@ -31,8 +31,8 @@ jobs:
         pip install -r requirements.txt
 
     - name: Lint with flake8
-      continue-on-error: true
-      run: flake8 scripts/ tests/
+      continue-on-error: false
+      run: flake8 scripts/ tests/ --select E303,W293,W291,W292,E305
 
     - name: Run unittest tests with coverage
       run: |

README.md

@@ -341,4 +341,7 @@ This project uses [flake8](https://flake8.pycqa.org/en/latest/) for linting. To
 
 ```
 flake8 scripts/ tests/
+
+# Or, if you want to run flake8 with the same configuration as the CI:
+flake8 scripts/ tests/ --select E303,W293,W291,W292,E305
 ```

scripts/ai_functions.py

@@ -45,8 +45,6 @@ def improve_code(suggestions: List[str], code: str) -> str:
     result_string = call_ai_function(function_string, args, description_string)
     return result_string
 
-
-
 def write_tests(code: str, focus: List[str]) -> str:
     """
     A function that takes in code and focus topics and returns a response from create chat completion api call.

scripts/browse.py

@@ -41,7 +41,7 @@ def scrape_text(url):
     # Restrict access to local files
     if check_local_file_access(url):
         return "Error: Access to local files is restricted"
-    
+
     # Validate the input URL
     if not is_valid_url(url):
         # Sanitize the input URL

scripts/config.py

@@ -61,7 +61,7 @@ class Config(metaclass=Singleton):
 
         self.use_mac_os_tts = False
         self.use_mac_os_tts = os.getenv("USE_MAC_OS_TTS")
-        
+
         self.google_api_key = os.getenv("GOOGLE_API_KEY")
         self.custom_search_engine_id = os.getenv("CUSTOM_SEARCH_ENGINE_ID")
 

scripts/file_operations.py

@@ -80,4 +80,4 @@ def search_files(directory):
             relative_path = os.path.relpath(os.path.join(root, file), working_directory)
             found_files.append(relative_path)
 
-    return found_files
+    return found_files

scripts/logger.py

@@ -159,6 +159,7 @@ class ConsoleHandler(logging.StreamHandler):
         except Exception:
             self.handleError(record)
 
+
 '''
 Allows to handle custom placeholders 'title_color' and 'message_no_color'.
 To use this formatter, make sure to pass 'color', 'title' as log extras.

scripts/memory/__init__.py

@@ -44,6 +44,7 @@ def get_memory(cfg, init=False):
 def get_supported_memory_backends():
     return supported_memory
 
+
 __all__ = [
     "get_memory",
     "LocalCache",

scripts/speak.py

@@ -61,7 +61,7 @@ def gtts_speech(text):
 def macos_tts_speech(text, voice_index=0):
     if voice_index == 0:
         os.system(f'say "{text}"')
-    else: 
+    else:
         if voice_index == 1:
             os.system(f'say -v "Ava (Premium)" "{text}"')
         else:
@@ -79,7 +79,7 @@ def say_text(text, voice_index=0):
             success = eleven_labs_speech(text, voice_index)
             if not success:
                 gtts_speech(text)
-        
+
         queue_semaphore.release()
 
     queue_semaphore.acquire(True)

tests/integration/memory_tests.py

@@ -45,5 +45,6 @@ class TestLocalCache(unittest.TestCase):
         self.assertEqual(len(relevant_texts), k)
         self.assertIn(self.example_texts[1], relevant_texts)
 
+
 if __name__ == '__main__':
     unittest.main()

tests/local_cache_test.py

@@ -47,5 +47,6 @@ class TestLocalCache(unittest.TestCase):
         stats = self.cache.get_stats()
         self.assertEqual(stats, (1, self.cache.data.embeddings.shape))
 
+
 if __name__ == '__main__':
     unittest.main()

tests/test_browse_scrape_text.py

@@ -33,8 +33,6 @@ Additional aspects:
 - The function uses a generator expression to split the text into lines and chunks, which can improve performance for large amounts of text.
 """
 
-
-
 class TestScrapeText:
 
     # Tests that scrape_text() returns the expected text when given a valid URL.

tests/test_json_parser.py

@@ -66,8 +66,6 @@ class TestParseJson(unittest.TestCase):
         # Assert that this raises an exception:
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
-
-
     def test_invalid_json_leading_sentence_with_gpt(self):
         # Test that a REALLY invalid JSON string raises an error when try_to_fix_with_gpt is False
         json_str = """I will first need to browse the repository (https://github.com/Torantulino/Auto-GPT) and identify any potential bugs that need fixing. I will use the "browse_website" command for this.
@@ -108,6 +106,5 @@ class TestParseJson(unittest.TestCase):
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
 
-
 if __name__ == '__main__':
     unittest.main()

tests/unit/json_tests.py

@@ -68,8 +68,6 @@ class TestParseJson(unittest.TestCase):
         # Assert that this raises an exception:
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
-
-
     def test_invalid_json_leading_sentence_with_gpt(self):
         # Test that a REALLY invalid JSON string raises an error when try_to_fix_with_gpt is False
         json_str = """I will first need to browse the repository (https://github.com/Torantulino/Auto-GPT) and identify any potential bugs that need fixing. I will use the "browse_website" command for this.
@@ -110,6 +108,5 @@ class TestParseJson(unittest.TestCase):
         self.assertEqual(fix_and_parse_json(json_str, try_to_fix_with_gpt=False), good_obj)
 
 
-
 if __name__ == '__main__':
     unittest.main()