✨ project template

llm_term/cli.py

@@ -2,7 +2,7 @@
 LLM CLI
 """
 
-from typing import Optional, Tuple
+from __future__ import annotations
 
 import click
 import rich.traceback
@@ -85,8 +85,8 @@
 )
 def cli(
     model: str,
-    chat: Tuple[str, ...],
-    system: Optional[str],
+    chat: tuple[str, ...],
+    system: str | None,
     api_key: str,
     stream: bool,
     console: int,

llm_term/utils.py

@@ -2,10 +2,12 @@
 Helper functions for the CLI
 """
 
+from __future__ import annotations
+
 from datetime import datetime, timezone
 from pathlib import Path
 from textwrap import dedent
-from typing import Any, Dict, Generator, List, Optional
+from typing import Any, Generator
 
 import click
 import openai
@@ -54,7 +56,7 @@ def check_credentials(api_key: str) -> None:
         raise click.ClickException(msg)
 
 
-def setup_system_message(model: str, message: Optional[str] = None) -> Message:
+def setup_system_message(model: str, message: str | None = None) -> Message:
     """
     Set up the system message
     """
@@ -86,7 +88,7 @@ def chat_session(
     history_file = Path().home() / ".llm-term-history.txt"
     history = FileHistory(str(history_file))
     session: PromptSession = PromptSession(history=history, erase_when_done=True)
-    messages: List[Message] = [system_message]
+    messages: list[Message] = [system_message]
     if chat_message.strip() != "":
         messages.append(Message(role="user", content=chat_message))
     message_counter = 0
@@ -142,7 +144,7 @@ def print_response(
 
 
 def render_streamed_response(
-    response: Generator[Dict[str, Any], None, None], console: Console, panel: bool
+    response: Generator[dict[str, Any], None, None], console: Console, panel: bool
 ) -> Message:
     """
     Render the streamed response and a spinner

pyproject.toml

@@ -208,10 +208,6 @@ select = [
   "YTT"
 ]
 target-version = "py38"
-unfixable = [
-  # Don't touch unused imports
-  "F401"
-]
 
 [tool.ruff.flake8-tidy-imports]
 ban-relative-imports = "all"