From afb0d6449aad47d63f0989ad767d2de7ee38b9eb Mon Sep 17 00:00:00 2001 From: Anton Potapov <47938145+Flagro@users.noreply.github.com> Date: Wed, 11 Dec 2024 10:06:58 +0400 Subject: [PATCH] Simplify the zip method in get_paginated_list_keyboard --- bot/rp_bot/ai_agent/ai.py | 2 +- bot/telegram/keyboards.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/bot/rp_bot/ai_agent/ai.py b/bot/rp_bot/ai_agent/ai.py index 270dadf..20695c9 100644 --- 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 index 363731a..c919130 100644 --- 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