Sort and format import blocks

ask-code/ask-code.py

@@ -1,5 +1,6 @@
 import os
 import sys
+
 from chat.ask_codebase.chains.smart_qa import SmartQA
 
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", "libs"))

commit/commit.py

@@ -1,16 +1,16 @@
 # flake8: noqa: E402
 import os
 import re
-import sys
 import subprocess
+import sys
 
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", "libs"))
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", "..", "libs"))
 sys.path.append(os.path.dirname(__file__))
 
 from chatmark import Checkbox, Form, TextEditor  # noqa: E402
-from llm_api import chat_completion_stream  # noqa: E402
 from ide_services.services import log_info
+from llm_api import chat_completion_stream  # noqa: E402
 
 
 def extract_markdown_block(text):

commit/test/prompt/commit_message_run.py

@@ -1,6 +1,7 @@
-import sys
-import os
 import json
+import os
+import sys
+
 import requests
 
 

libs/chatmark/__init__.py

@@ -1,5 +1,5 @@
-from .widgets import Checkbox, TextEditor, Radio, Button
 from .form import Form
+from .widgets import Button, Checkbox, Radio, TextEditor
 
 __all__ = [
     "Checkbox",

libs/chatmark/chatmark_example/main.py

@@ -1,10 +1,10 @@
-import sys
 import os
+import sys
 
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", "libs"))
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", "..", "libs"))
 
-from chatmark import Checkbox, TextEditor, Radio, Button, Form  # noqa: E402
+from chatmark import Button, Checkbox, Form, Radio, TextEditor  # noqa: E402
 
 
 def main():

libs/chatmark/form.py

@@ -1,6 +1,7 @@
-from typing import List, Optional, Dict, Union
+from typing import Dict, List, Optional, Union
+
 from .iobase import pipe_interaction
-from .widgets import Widget, Button
+from .widgets import Button, Widget
 
 
 class Form:

libs/chatmark/widgets.py

@@ -1,8 +1,9 @@
-from typing import List, Optional, Dict, Tuple
-from .iobase import pipe_interaction
 from abc import ABC, abstractmethod
+from typing import Dict, List, Optional, Tuple
 from uuid import uuid4
 
+from .iobase import pipe_interaction
+
 
 class Widget(ABC):
     """

libs/ide_services/__init__.py

@@ -1,9 +1,9 @@
 from .services import (
     get_lsp_brige_port,
-    install_python_env,
-    update_slash_commands,
-    open_folder,
     ide_language,
+    install_python_env,
+    open_folder,
+    update_slash_commands,
 )
 
 __all__ = [

libs/ide_services/services.py

@@ -1,7 +1,8 @@
-import requests
 import os
 from functools import wraps
 
+import requests
+
 BASE_SERVER_URL = os.environ.get("DEVCHAT_IDE_SERVICE_URL", "http://localhost:3000")
 
 

libs/llm_api/__init__.py

@@ -1,3 +1,3 @@
-from .openai import chat_completion_stream, chat_completion_no_stream_return_json
+from .openai import chat_completion_no_stream_return_json, chat_completion_stream
 
 __all__ = ["chat_completion_stream", "chat_completion_no_stream_return_json"]

libs/llm_api/openai.py

@@ -1,8 +1,8 @@
 # flake8: noqa: E402
-import re
-import os
-import sys
 import json
+import os
+import re
+import sys
 
 import openai
 

unit_tests/i18n.py

@@ -1,5 +1,5 @@
-import os
 import gettext
+import os
 from enum import Enum
 
 

unit_tests/main.py

@@ -1,19 +1,17 @@
 import os
 import sys
+
 import click
-
 from chat.ask_codebase.tools.retrieve_file_content import retrieve_file_content
-
-from propose_test import propose_test
 from find_reference_tests import find_reference_tests
-from write_tests import write_and_print_tests
 from i18n import TUILanguage, get_translation
-
 from model import FuncToTest, TokenBudgetExceededException, UserCancelledException
+from propose_test import propose_test
+from write_tests import write_and_print_tests
 
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", "libs"))
 
-from chatmark import Checkbox, TextEditor, Form  # noqa: E402
+from chatmark import Checkbox, Form, TextEditor  # noqa: E402
 from ide_services import ide_language  # noqa: E402
 
 

unit_tests/model.py

@@ -1,5 +1,6 @@
 from dataclasses import dataclass
 from typing import Optional
+
 from chat.ask_codebase.tools.retrieve_file_content import retrieve_file_content
 
 

unit_tests/openai_util.py

@@ -1,4 +1,5 @@
 from typing import Optional
+
 from openai import OpenAI, Stream
 from openai.types.chat import ChatCompletionChunk
 from tenacity import retry, stop_after_attempt, wait_random_exponential

unit_tests/propose_test.py

@@ -1,13 +1,11 @@
-from typing import List
+import json
 from functools import partial
+from typing import List
 
 import tiktoken
-import json
-
+from model import FuncToTest, TokenBudgetExceededException
 from openai_util import create_chat_completion_content
 from prompts import PROPOSE_TEST_PROMPT
-from model import FuncToTest, TokenBudgetExceededException
-
 
 MODEL = "gpt-3.5-turbo-1106"
 # MODEL = "gpt-4-1106-preview"

unit_tests/write_tests.py

@@ -1,13 +1,11 @@
-from typing import List, Optional
 from functools import partial
+from typing import List, Optional
 
 import tiktoken
-
 from chat.ask_codebase.tools.retrieve_file_content import retrieve_file_content
+from model import FuncToTest, TokenBudgetExceededException
 from openai_util import create_chat_completion_chunks
 from prompts import WRITE_TESTS_PROMPT
-from model import FuncToTest, TokenBudgetExceededException
-
 
 MODEL = "gpt-4-1106-preview"
 TOKEN_BUDGET = int(128000 * 0.9)