Simplify the zip method in get_paginated_list_keyboard

Flagro · Dec 11, 2024 · afb0d64 · afb0d64
1 parent d0b8ec4
commit afb0d64
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bot/rp_bot/ai_agent/ai.py b/bot/rp_bot/ai_agent/ai.py
@@ -1,6 +1,6 @@
 import tiktoken
 import io
-from typing import AsyncIterator, List, Dict, Any
+from typing import AsyncIterator
 from openai import OpenAI
 
 from ...models.config.ai_config import AIConfig

diff --git a/bot/telegram/keyboards.py b/bot/telegram/keyboards.py
@@ -15,9 +15,9 @@ def get_paginated_list_keyboard(
     modes_per_page: Optional[int] = 5,
 ) -> InlineKeyboardMarkup:
     value_names = list(value_id_to_name.values())
-    capitalized_value_names = [value_name.capitalize() for value_name in value_names]
-    value_ids = list(value_id_to_name.keys())
-    values = list(zip(capitalized_value_names, value_ids))
+    values = list(
+        zip(map(lambda s: s.capitalize(), value_names), value_id_to_name.keys())
+    )
     page_start = page_index * modes_per_page
     page_end = (page_index + 1) * modes_per_page