fix linting

Author: Yifu Cai

aide/backend/__init__.py

@@ -2,7 +2,6 @@
 
 import logging
 
-from nbformat import current
 from . import backend_anthropic, backend_openai, backend_openrouter, backend_gdm
 from .utils import FunctionSpec, OutputType, PromptType, compile_prompt_to_md
 
@@ -149,7 +148,7 @@ def query(
         **model_kwargs,
     )
     logger.info(f"response: {output}", extra={"verbose": True})
-    logger.info(f"---Query complete---", extra={"verbose": True})
+    logger.info("---Query complete---", extra={"verbose": True})
     if token_counter is not None:
         token_counter.add_tokens(model, input_tokens=in_tok_count, output_tokens=out_tok_count)
 

aide/backend/utils.py

@@ -5,12 +5,12 @@
 import jsonschema
 from dataclasses_json import DataClassJsonMixin
 
+import backoff
+
 PromptType = str | dict | list
 FunctionCallType = dict
 OutputType = str | FunctionCallType
 
-import backoff
-
 logger = logging.getLogger("aide")
 
 

aide/utils/config.py

@@ -10,7 +10,6 @@
 from omegaconf import OmegaConf
 from rich.syntax import Syntax
 import shutup
-from rich.logging import RichHandler
 import logging
 
 from aide.journal import Journal, filter_journal
@@ -240,4 +239,4 @@ def output_file_or_placeholder(file: Path):
         else:
             return json.dumps(json.loads(file.read_text()), indent=4)
     else:
-        return f"File not found."
+        return "File not found."