diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 51d56334f..068be9acd 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -10,7 +10,7 @@ repos: - id: flake8 args: ["--show-source"] - repo: https://github.com/pycqa/isort - rev: 5.11.1 + rev: 5.12.0 hooks: - id: isort args: ["--profile", "black"] diff --git a/aider/coder.py b/aider/coder.py index 24e87517f..af761d62f 100755 --- a/aider/coder.py +++ b/aider/coder.py @@ -2,21 +2,19 @@ import os import sys -import traceback import time -from openai.error import RateLimitError - -from rich.console import Console -from rich.live import Live -from rich.markdown import Markdown +import traceback from pathlib import Path import git import openai +from openai.error import RateLimitError +from rich.console import Console +from rich.live import Live +from rich.markdown import Markdown # from aider.dump import dump -from aider import utils -from aider import prompts +from aider import prompts, utils from aider.commands import Commands from aider.repomap import RepoMap diff --git a/aider/commands.py b/aider/commands.py index 758bfb22c..56906e4e6 100644 --- a/aider/commands.py +++ b/aider/commands.py @@ -1,9 +1,11 @@ -import sys import os -import git -import subprocess import shlex +import subprocess +import sys + +import git from prompt_toolkit.completion import Completion + from aider import prompts diff --git a/aider/dump.py b/aider/dump.py index de6dcbb6d..2c8bf31c2 100644 --- a/aider/dump.py +++ b/aider/dump.py @@ -1,5 +1,5 @@ -import traceback import json +import traceback def cvt(s): diff --git a/aider/io.py b/aider/io.py index b76d34543..014f720f7 100644 --- a/aider/io.py +++ b/aider/io.py @@ -1,19 +1,18 @@ import os import os.path -from pygments.lexers import guess_lexer_for_filename -from pygments.token import Token -from prompt_toolkit.styles import Style -from pygments.util import ClassNotFound -from prompt_toolkit.shortcuts import PromptSession, prompt +from collections import defaultdict +from datetime import datetime +from pathlib import Path + from prompt_toolkit.completion import Completer, Completion from prompt_toolkit.history import FileHistory -from prompt_toolkit.shortcuts import CompleteStyle +from prompt_toolkit.shortcuts import CompleteStyle, PromptSession, prompt +from prompt_toolkit.styles import Style +from pygments.lexers import guess_lexer_for_filename +from pygments.token import Token +from pygments.util import ClassNotFound from rich.console import Console from rich.text import Text -from pathlib import Path -from datetime import datetime - -from collections import defaultdict class FileContentCompleter(Completer): diff --git a/aider/main.py b/aider/main.py index 6fb51cb06..9913c171f 100644 --- a/aider/main.py +++ b/aider/main.py @@ -1,7 +1,9 @@ import os import sys -import git + import configargparse +import git + from aider.coder import Coder from aider.io import InputOutput diff --git a/aider/repomap.py b/aider/repomap.py index bd52f1eb9..510a075fb 100644 --- a/aider/repomap.py +++ b/aider/repomap.py @@ -1,10 +1,11 @@ -import os import json -import sys +import os import subprocess -import tiktoken +import sys import tempfile +import tiktoken + from aider import prompts # Global cache for tags diff --git a/aider/utils.py b/aider/utils.py index e1734ab40..e6cedff47 100644 --- a/aider/utils.py +++ b/aider/utils.py @@ -1,6 +1,5 @@ -import re import math - +import re from difflib import SequenceMatcher from pathlib import Path diff --git a/setup.py b/setup.py index 6b922cfd1..20e70ba88 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,4 @@ -from setuptools import setup, find_packages +from setuptools import find_packages, setup with open("requirements.txt") as f: requirements = f.read().splitlines() diff --git a/tests/test_coder.py b/tests/test_coder.py index 86e7487b6..52ffabdb9 100644 --- a/tests/test_coder.py +++ b/tests/test_coder.py @@ -1,6 +1,7 @@ import os import unittest from unittest.mock import MagicMock + from aider.coder import Coder diff --git a/tests/test_commands.py b/tests/test_commands.py index 7cdca3a95..5f3d596d1 100644 --- a/tests/test_commands.py +++ b/tests/test_commands.py @@ -1,10 +1,11 @@ import os +import shutil import tempfile from unittest import TestCase + from aider.commands import Commands from aider.io import InputOutput -import shutil class TestCommands(TestCase): def setUp(self): diff --git a/tests/test_io.py b/tests/test_io.py index 4a51bcd0d..dfc3806ff 100644 --- a/tests/test_io.py +++ b/tests/test_io.py @@ -1,13 +1,16 @@ import os import unittest from unittest.mock import patch + from aider.io import InputOutput + class TestInputOutput(unittest.TestCase): def test_no_color_environment_variable(self): with patch.dict(os.environ, {"NO_COLOR": "1"}): io = InputOutput() self.assertFalse(io.pretty) + if __name__ == "__main__": unittest.main() diff --git a/tests/test_main.py b/tests/test_main.py index f8b891bb3..eb35435dc 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,14 +1,16 @@ import os -import tempfile -import subprocess import shutil +import subprocess +import tempfile +from io import StringIO from unittest import TestCase from unittest.mock import patch -from aider.main import main + from prompt_toolkit.input import create_input -from io import StringIO from prompt_toolkit.output import DummyOutput +from aider.main import main + class TestMain(TestCase): def setUp(self): diff --git a/tests/test_repomap.py b/tests/test_repomap.py index c52f24d00..b43fc7779 100644 --- a/tests/test_repomap.py +++ b/tests/test_repomap.py @@ -1,9 +1,11 @@ import os import tempfile import unittest -from aider.repomap import RepoMap -from unittest.mock import patch from subprocess import CompletedProcess +from unittest.mock import patch + +from aider.repomap import RepoMap + class TestRepoMap(unittest.TestCase): def test_get_tags_map(self): @@ -65,7 +67,14 @@ def my_function(arg1, arg2): def test_check_for_ctags_success(self): with patch("subprocess.run") as mock_run: - mock_run.return_value = CompletedProcess(args=["ctags", "--version"], returncode=0, stdout=b'{"_type": "tag", "name": "status", "path": "aider/main.py", "pattern": "/^ status = main()$/", "kind": "variable"}') + mock_run.return_value = CompletedProcess( + args=["ctags", "--version"], + returncode=0, + stdout=( + b'{"_type": "tag", "name": "status", "path": "aider/main.py", "pattern": "/^ ' + b' status = main()$/", "kind": "variable"}' + ), + ) repo_map = RepoMap(use_ctags=True) result = repo_map.check_for_ctags() self.assertTrue(result) diff --git a/tests/test_utils.py b/tests/test_utils.py index bddeb67dc..5b6b940c5 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -1,6 +1,7 @@ # flake8: noqa: E501 import unittest + from aider import utils