Improve lint issues

libs/chatmark/step.py

@@ -1,4 +1,4 @@
-from contextlib import AbstractContextManager, contextmanager
+from contextlib import AbstractContextManager
 
 
 class Step(AbstractContextManager):
@@ -25,4 +25,4 @@ class Step(AbstractContextManager):
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         # close the step
-        print(f"\n```", flush=True)
+        print("\n```", flush=True)

unit_tests/assistants/directory_structure/base.py

@@ -2,7 +2,6 @@ from abc import ABC, abstractmethod
 from pathlib import Path
 from typing import Any, Callable, Optional
 
-from openai import OpenAI
 from tools.directory_viewer import mk_repo_file_criteria
 
 

unit_tests/assistants/recommend_test_context.py

@@ -1,5 +1,5 @@
 import json
-from typing import List, Optional, Tuple
+from typing import List, Optional
 
 from model import FuncToTest
 from openai_util import create_chat_completion_content

unit_tests/find_context.py

@@ -1,13 +1,10 @@
-import json
 import os
 import sys
 from collections import defaultdict
-from functools import partial
+from typing import Dict, List, Optional, Set
-from typing import Dict, Iterable, List, Optional, Set, Tuple
 
 from assistants.recommend_test_context import get_recommended_symbols
 from model import FuncToTest
-from tools.file_util import retrieve_file_content
 from tools.symbol_util import (
     find_symbol_nodes,
     get_symbol_content,

unit_tests/tools/symbol_util.py

@@ -8,7 +8,7 @@ from tools.file_util import retrieve_file_content
 
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from libs.ide_services import IDEService, Location, Position, SymbolNode
+from libs.ide_services import Position, SymbolNode
 
 
 def split_tokens(text: str) -> Dict[str, List[int]]:

unit_tests/write_tests.py

@@ -42,7 +42,7 @@ def _mk_write_tests_msg(
 
     context_content = ""
     if symbol_context:
-        context_content += f"\n\nrelevant context\n```\n"
+        context_content += "\n\nrelevant context\n```\n"
         context_content += "\n\n".join(symbol_context)
         context_content += "\n```\n"