style: apply Black formatting

Author: Alcheri

__init__.py

@@ -23,7 +23,9 @@
 __version__ = "1.1"
 
 # XXX Replace this with an appropriate author or supybot.Author instance.
-__author__ = supybot.Author("Barry Suridge", "Alcheri", "barry.suridge@gmail.com")
+__author__ = supybot.Author(
+    "Barry Suridge", "Alcheri", "barry.suridge@gmail.com"
+)
 # This is a dictionary mapping supybot.Author instances to lists of
 # contributions.
 __contributors__ = {}
@@ -47,7 +49,10 @@
     except ImportError as e:
         missing_names = {"test", f"{__name__}.test"}
         missing_test_import = "cannot import name 'test'" in str(e)
-        if getattr(e, "name", None) not in missing_names and not missing_test_import:
+        if (
+            getattr(e, "name", None) not in missing_names
+            and not missing_test_import
+        ):
             raise
 
 Class = plugin.Class

config/runtime.py

@@ -58,9 +58,15 @@ def get_config() -> dict[str, Any]:
         "max_tokens": _to_int(
             plugin_conf.maxUserTokens(), default_config["max_tokens"]
         ),
-        "cooldown": _to_int(plugin_conf.cooldownSeconds(), default_config["cooldown"]),
-        "irc_chunk": _to_int(plugin_conf.ircChunkSize(), default_config["irc_chunk"]),
+        "cooldown": _to_int(
+            plugin_conf.cooldownSeconds(), default_config["cooldown"]
+        ),
+        "irc_chunk": _to_int(
+            plugin_conf.ircChunkSize(), default_config["irc_chunk"]
+        ),
         "botnick": _to_str(plugin_conf.botnick(), default_config["botnick"]),
-        "language": _to_str(plugin_conf.language(), default_config["language"]),
+        "language": _to_str(
+            plugin_conf.language(), default_config["language"]
+        ),
         "debug": _to_bool(plugin_conf.debugMode(), default_config["debug"]),
     }

cooldown.py

@@ -34,7 +34,9 @@ def should_wait_message(self, context_key, now, cooldown_s):
 
         if delta < cd:
             wait_time = int(cd - delta) + 1
-            return "Please wait {}s before sending another request.".format(wait_time)
+            return "Please wait {}s before sending another request.".format(
+                wait_time
+            )
 
         return None
 

core/__init__.py

@@ -1,5 +1,10 @@
 from .chat import chat_with_model, execute_chat_with_input_moderation
-from .text import clean_output, count_tokens, is_likely_math, split_irc_reply_lines
+from .text import (
+    clean_output,
+    count_tokens,
+    is_likely_math,
+    split_irc_reply_lines,
+)
 
 __all__ = [
     "chat_with_model",

core/chat.py

@@ -21,10 +21,14 @@ async def chat_with_model(
     context_key: str,
     config: dict[str, Any],
     *,
-    get_history_fn: Callable[[str, str], list[dict[str, str]]] = get_user_history,
+    get_history_fn: Callable[
+        [str, str], list[dict[str, str]]
+    ] = get_user_history,
     trim_history_fn: Callable[[str], list[dict[str, str]]] = trim_history,
     ensure_openai_client_fn: Callable[[], Any] = ensure_openai_client,
-    create_completion_fn: Callable[..., Any] = create_chat_completion_with_fallback,
+    create_completion_fn: Callable[
+        ..., Any
+    ] = create_chat_completion_with_fallback,
 ) -> str:
     math_mode = is_likely_math(user_message)
 
@@ -82,7 +86,9 @@ async def execute_chat_with_input_moderation(
     *,
     cooldown_manager: Any,
     check_moderation_flag_fn: Callable[[str], Any] = check_moderation_flag,
-    chat_with_model_fn: Callable[[str, str, dict[str, Any]], Any] = chat_with_model,
+    chat_with_model_fn: Callable[
+        [str, str, dict[str, Any]], Any
+    ] = chat_with_model,
     count_tokens_fn: Callable[[str | None], int] = count_tokens,
     now_fn: Callable[[], float] = time.time,
 ) -> str:

core/text.py

@@ -32,7 +32,9 @@ def clean_output(text: str | None) -> str:
     return cleaned.strip()
 
 
-def split_irc_reply_lines(text: str | None, chunk_size: int = 350) -> list[str]:
+def split_irc_reply_lines(
+    text: str | None, chunk_size: int = 350
+) -> list[str]:
     payload = (text or "").strip()
     if not payload:
         return []

plugin.py

@@ -52,7 +52,9 @@ def chat(self, irc, msg, args, user_input):
 
         # ---- Pre-cooldown check (UX polish) ----
         now = time.time()
-        msg_wait = COOLDOWNS.should_wait_message(context_key, now, config["cooldown"])
+        msg_wait = COOLDOWNS.should_wait_message(
+            context_key, now, config["cooldown"]
+        )
 
         if msg_wait:
             irc.reply(msg_wait, prefixNick=False)
@@ -80,21 +82,28 @@ def chat(self, irc, msg, args, user_input):
                 )
             )
 
-            chunks = split_irc_reply_lines(response, chunk_size=config["irc_chunk"])
+            chunks = split_irc_reply_lines(
+                response, chunk_size=config["irc_chunk"]
+            )
             if not chunks:
                 irc.reply("AI returned no response.", prefixNick=False)
             else:
                 for chunk in chunks:
                     irc.reply(chunk, prefixNick=False)
 
             if config["debug"]:
-                log.info("[Asyncio DEBUG] {}: {}".format(context_key, response))
+                log.info(
+                    "[Asyncio DEBUG] {}: {}".format(context_key, response)
+                )
 
         except Exception as error:
             log.error(
-                "[Asyncio] Exception in chat command: {}".format(error), exc_info=True
+                "[Asyncio] Exception in chat command: {}".format(error),
+                exc_info=True,
+            )
+            irc.reply(
+                "An unexpected error occurred. Check logs.", prefixNick=False
             )
-            irc.reply("An unexpected error occurred. Check logs.", prefixNick=False)
 
     def resetCommand(self, irc, msg, args):
         """Reset your conversation memory for this channel (or PM)."""

services/openai_client.py

@@ -27,7 +27,9 @@ def _chat_model_candidates() -> list[str]:
     """Ordered model fallback list for chat completions."""
     env_value = os.getenv("OPENAI_CHAT_MODELS", "")
     if env_value.strip():
-        models = [model.strip() for model in env_value.split(",") if model.strip()]
+        models = [
+            model.strip() for model in env_value.split(",") if model.strip()
+        ]
         if models:
             return models
 
@@ -70,7 +72,9 @@ def create_chat_completion_with_fallback(
                 **kwargs,
             )
             if state.active_chat_model != model_name:
-                log.warning("[Asyncio] Chat model switched to '{}'".format(model_name))
+                log.warning(
+                    "[Asyncio] Chat model switched to '{}'".format(model_name)
+                )
             state.active_chat_model = model_name
             return response
         except Exception as error:
@@ -89,7 +93,9 @@ def create_chat_completion_with_fallback(
     raise RuntimeError("No chat model candidates configured.")
 
 
-def ensure_openai_client(runtime_state: OpenAIRuntimeState | None = None) -> Any:
+def ensure_openai_client(
+    runtime_state: OpenAIRuntimeState | None = None,
+) -> Any:
     state = runtime_state or OPENAI_RUNTIME
     if state.client is not None:
         return state.client

state/memory.py

@@ -11,9 +11,13 @@ def make_context_key(msg: Any) -> str:
     return f"{channel}:{nick}"
 
 
-def get_user_history(context_key: str, system_prompt: str) -> list[dict[str, str]]:
+def get_user_history(
+    context_key: str, system_prompt: str
+) -> list[dict[str, str]]:
     if context_key not in USER_HISTORIES:
-        USER_HISTORIES[context_key] = [{"role": "system", "content": system_prompt}]
+        USER_HISTORIES[context_key] = [
+            {"role": "system", "content": system_prompt}
+        ]
         return USER_HISTORIES[context_key]
 
     history = USER_HISTORIES[context_key]
@@ -27,7 +31,9 @@ def get_user_history(context_key: str, system_prompt: str) -> list[dict[str, str
     return history
 
 
-def trim_history(context_key: str, max_messages: int = 12) -> list[dict[str, str]]:
+def trim_history(
+    context_key: str, max_messages: int = 12
+) -> list[dict[str, str]]:
     history = USER_HISTORIES.get(context_key, [])
     if len(history) > max_messages:
         USER_HISTORIES[context_key] = [history[0]] + history[-10:]

tests/test_core_chat.py

@@ -23,7 +23,9 @@ def should_wait_message(self, context_key, now, cooldown):
         return "Please wait 1s before sending another request."
 
     def record(self, context_key, now):
-        raise AssertionError("record should not be called when cooldown blocks")
+        raise AssertionError(
+            "record should not be called when cooldown blocks"
+        )
 
 
 class CoreChatTestCase(unittest.IsolatedAsyncioTestCase):
@@ -59,7 +61,13 @@ async def fake_chat_with_model(user_request, context_key, config):
         self.assertEqual(response, "ok")
         self.assertEqual(
             events,
-            ["cooldown_check", "cooldown_record", "count_tokens", "moderation", "chat"],
+            [
+                "cooldown_check",
+                "cooldown_record",
+                "count_tokens",
+                "moderation",
+                "chat",
+            ],
         )
 
     async def test_cooldown_blocks_early(self):
@@ -120,4 +128,6 @@ async def fake_chat_with_model(_a, _b, _c):
         )
 
         self.assertIn("flagged as inappropriate", response)
-        self.assertEqual(events, ["cooldown_check", "cooldown_record", "moderation"])
+        self.assertEqual(
+            events, ["cooldown_check", "cooldown_record", "moderation"]
+        )