mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 16:54:59 +00:00
isort
This commit is contained in:
parent
95be55a2b6
commit
92cd2f3ce6
15 changed files with 56 additions and 38 deletions
|
@ -10,7 +10,7 @@ repos:
|
||||||
- id: flake8
|
- id: flake8
|
||||||
args: ["--show-source"]
|
args: ["--show-source"]
|
||||||
- repo: https://github.com/pycqa/isort
|
- repo: https://github.com/pycqa/isort
|
||||||
rev: 5.11.1
|
rev: 5.12.0
|
||||||
hooks:
|
hooks:
|
||||||
- id: isort
|
- id: isort
|
||||||
args: ["--profile", "black"]
|
args: ["--profile", "black"]
|
||||||
|
|
|
@ -2,21 +2,19 @@
|
||||||
|
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import traceback
|
|
||||||
import time
|
import time
|
||||||
from openai.error import RateLimitError
|
import traceback
|
||||||
|
|
||||||
from rich.console import Console
|
|
||||||
from rich.live import Live
|
|
||||||
from rich.markdown import Markdown
|
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
import git
|
import git
|
||||||
import openai
|
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.dump import dump
|
||||||
from aider import utils
|
from aider import prompts, utils
|
||||||
from aider import prompts
|
|
||||||
from aider.commands import Commands
|
from aider.commands import Commands
|
||||||
from aider.repomap import RepoMap
|
from aider.repomap import RepoMap
|
||||||
|
|
||||||
|
|
|
@ -1,9 +1,11 @@
|
||||||
import sys
|
|
||||||
import os
|
import os
|
||||||
import git
|
|
||||||
import subprocess
|
|
||||||
import shlex
|
import shlex
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import git
|
||||||
from prompt_toolkit.completion import Completion
|
from prompt_toolkit.completion import Completion
|
||||||
|
|
||||||
from aider import prompts
|
from aider import prompts
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
import traceback
|
|
||||||
import json
|
import json
|
||||||
|
import traceback
|
||||||
|
|
||||||
|
|
||||||
def cvt(s):
|
def cvt(s):
|
||||||
|
|
19
aider/io.py
19
aider/io.py
|
@ -1,19 +1,18 @@
|
||||||
import os
|
import os
|
||||||
import os.path
|
import os.path
|
||||||
from pygments.lexers import guess_lexer_for_filename
|
from collections import defaultdict
|
||||||
from pygments.token import Token
|
from datetime import datetime
|
||||||
from prompt_toolkit.styles import Style
|
from pathlib import Path
|
||||||
from pygments.util import ClassNotFound
|
|
||||||
from prompt_toolkit.shortcuts import PromptSession, prompt
|
|
||||||
from prompt_toolkit.completion import Completer, Completion
|
from prompt_toolkit.completion import Completer, Completion
|
||||||
from prompt_toolkit.history import FileHistory
|
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.console import Console
|
||||||
from rich.text import Text
|
from rich.text import Text
|
||||||
from pathlib import Path
|
|
||||||
from datetime import datetime
|
|
||||||
|
|
||||||
from collections import defaultdict
|
|
||||||
|
|
||||||
|
|
||||||
class FileContentCompleter(Completer):
|
class FileContentCompleter(Completer):
|
||||||
|
|
|
@ -1,7 +1,9 @@
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import git
|
|
||||||
import configargparse
|
import configargparse
|
||||||
|
import git
|
||||||
|
|
||||||
from aider.coder import Coder
|
from aider.coder import Coder
|
||||||
from aider.io import InputOutput
|
from aider.io import InputOutput
|
||||||
|
|
||||||
|
|
|
@ -1,10 +1,11 @@
|
||||||
import os
|
|
||||||
import json
|
import json
|
||||||
import sys
|
import os
|
||||||
import subprocess
|
import subprocess
|
||||||
import tiktoken
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
|
|
||||||
|
import tiktoken
|
||||||
|
|
||||||
from aider import prompts
|
from aider import prompts
|
||||||
|
|
||||||
# Global cache for tags
|
# Global cache for tags
|
||||||
|
|
|
@ -1,6 +1,5 @@
|
||||||
import re
|
|
||||||
import math
|
import math
|
||||||
|
import re
|
||||||
from difflib import SequenceMatcher
|
from difflib import SequenceMatcher
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
|
2
setup.py
2
setup.py
|
@ -1,4 +1,4 @@
|
||||||
from setuptools import setup, find_packages
|
from setuptools import find_packages, setup
|
||||||
|
|
||||||
with open("requirements.txt") as f:
|
with open("requirements.txt") as f:
|
||||||
requirements = f.read().splitlines()
|
requirements = f.read().splitlines()
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
import os
|
import os
|
||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import MagicMock
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
from aider.coder import Coder
|
from aider.coder import Coder
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,10 +1,11 @@
|
||||||
import os
|
import os
|
||||||
|
import shutil
|
||||||
import tempfile
|
import tempfile
|
||||||
from unittest import TestCase
|
from unittest import TestCase
|
||||||
|
|
||||||
from aider.commands import Commands
|
from aider.commands import Commands
|
||||||
from aider.io import InputOutput
|
from aider.io import InputOutput
|
||||||
|
|
||||||
import shutil
|
|
||||||
|
|
||||||
class TestCommands(TestCase):
|
class TestCommands(TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
|
|
|
@ -1,13 +1,16 @@
|
||||||
import os
|
import os
|
||||||
import unittest
|
import unittest
|
||||||
from unittest.mock import patch
|
from unittest.mock import patch
|
||||||
|
|
||||||
from aider.io import InputOutput
|
from aider.io import InputOutput
|
||||||
|
|
||||||
|
|
||||||
class TestInputOutput(unittest.TestCase):
|
class TestInputOutput(unittest.TestCase):
|
||||||
def test_no_color_environment_variable(self):
|
def test_no_color_environment_variable(self):
|
||||||
with patch.dict(os.environ, {"NO_COLOR": "1"}):
|
with patch.dict(os.environ, {"NO_COLOR": "1"}):
|
||||||
io = InputOutput()
|
io = InputOutput()
|
||||||
self.assertFalse(io.pretty)
|
self.assertFalse(io.pretty)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|
|
@ -1,14 +1,16 @@
|
||||||
import os
|
import os
|
||||||
import tempfile
|
|
||||||
import subprocess
|
|
||||||
import shutil
|
import shutil
|
||||||
|
import subprocess
|
||||||
|
import tempfile
|
||||||
|
from io import StringIO
|
||||||
from unittest import TestCase
|
from unittest import TestCase
|
||||||
from unittest.mock import patch
|
from unittest.mock import patch
|
||||||
from aider.main import main
|
|
||||||
from prompt_toolkit.input import create_input
|
from prompt_toolkit.input import create_input
|
||||||
from io import StringIO
|
|
||||||
from prompt_toolkit.output import DummyOutput
|
from prompt_toolkit.output import DummyOutput
|
||||||
|
|
||||||
|
from aider.main import main
|
||||||
|
|
||||||
|
|
||||||
class TestMain(TestCase):
|
class TestMain(TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
|
|
|
@ -1,9 +1,11 @@
|
||||||
import os
|
import os
|
||||||
import tempfile
|
import tempfile
|
||||||
import unittest
|
import unittest
|
||||||
from aider.repomap import RepoMap
|
|
||||||
from unittest.mock import patch
|
|
||||||
from subprocess import CompletedProcess
|
from subprocess import CompletedProcess
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
from aider.repomap import RepoMap
|
||||||
|
|
||||||
|
|
||||||
class TestRepoMap(unittest.TestCase):
|
class TestRepoMap(unittest.TestCase):
|
||||||
def test_get_tags_map(self):
|
def test_get_tags_map(self):
|
||||||
|
@ -65,7 +67,14 @@ def my_function(arg1, arg2):
|
||||||
|
|
||||||
def test_check_for_ctags_success(self):
|
def test_check_for_ctags_success(self):
|
||||||
with patch("subprocess.run") as mock_run:
|
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)
|
repo_map = RepoMap(use_ctags=True)
|
||||||
result = repo_map.check_for_ctags()
|
result = repo_map.check_for_ctags()
|
||||||
self.assertTrue(result)
|
self.assertTrue(result)
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
# flake8: noqa: E501
|
# flake8: noqa: E501
|
||||||
|
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
from aider import utils
|
from aider import utils
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue