From e53081905e1815ab67e73fe9e52262110229096e Mon Sep 17 00:00:00 2001 From: pyob-bot Date: Fri, 12 Jun 2026 03:46:04 +0000 Subject: [PATCH] Refactor: Optimize imports and suppress type checking in PromptsAndMemoryMixin --- src/pyob/prompts_and_memory.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/pyob/prompts_and_memory.py b/src/pyob/prompts_and_memory.py index 441f63c..70bb6e1 100644 --- a/src/pyob/prompts_and_memory.py +++ b/src/pyob/prompts_and_memory.py @@ -1,4 +1,5 @@ import os +import random import re from pyob.core_utils import logger @@ -55,8 +56,6 @@ def load_prompt(self, filename: str, **kwargs: str) -> str: path_b = os.path.join(data_dir, f"{base_name}.vB{ext}") if os.path.exists(path_a) and os.path.exists(path_b): - import random - chosen_version = random.choice(["vA", "vB"]) filepath = path_a if chosen_version == "vA" else path_b logger.info(f"A/B Testing: Selected {chosen_version} for prompt {filename}") @@ -145,6 +144,7 @@ def _get_rich_context(self, query_text: str = "") -> str: return context def update_memory(self) -> None: + # type: ignore session_context: list[str] = getattr(self, "session_context", []) if not session_context: return