vibego 1.0.10__py3-none-any.whl → 1.0.11__py3-none-any.whl

bot.py

@@ -2394,6 +2394,9 @@ MODEL_PUSH_SUPPLEMENT_STATUSES: set[str] = {
     "research",
     "test",
 }
+PUSH_MODEL_SUPPLEMENT_IN_PROGRESS_TEXT = (
+    "A supplementary description prompt is already active. Please respond or tap Skip/Cancel."
+)
 
 SUMMARY_COMMAND_PREFIX = "/task_summary_request_"
 SUMMARY_COMMAND_ALIASES: tuple[str, ...] = (
@@ -6849,6 +6852,10 @@ async def on_task_push_model(callback: CallbackQuery, state: FSMContext) -> None
         await callback.answer("Callback parameter error.", show_alert=True)
         return
     _, _, task_id = parts
+    current_state = await state.get_state()
+    existing_context: Dict[str, Any] = {}
+    if current_state == TaskPushStates.waiting_supplement.state:
+        existing_context = await state.get_data()
     task = await TASK_SERVICE.get_task(task_id)
     if task is None:
         await callback.answer("Task does not exist", show_alert=True)
@@ -6859,10 +6866,20 @@ async def on_task_push_model(callback: CallbackQuery, state: FSMContext) -> None
     actor = _actor_from_callback(callback)
     chat_id = callback.message.chat.id if callback.message else callback.from_user.id
     if task.status in MODEL_PUSH_SUPPLEMENT_STATUSES:
+        origin_message = callback.message
+        origin_message_id = origin_message.message_id if origin_message else None
+        if (
+            current_state == TaskPushStates.waiting_supplement.state
+            and existing_context.get("task_id") == task_id
+            and existing_context.get("origin_message_id") == origin_message_id
+        ):
+            await callback.answer(PUSH_MODEL_SUPPLEMENT_IN_PROGRESS_TEXT)
+            return
         await state.clear()
         await state.update_data(
             task_id=task_id,
-            origin_message=callback.message,
+            origin_message=origin_message,
+            origin_message_id=origin_message_id,
             chat_id=chat_id,
             actor=actor,
         )
@@ -6962,16 +6979,31 @@ async def on_task_push_model_fill(callback: CallbackQuery, state: FSMContext) ->
         await callback.answer("Callback parameter error.", show_alert=True)
         return
     _, _, task_id = parts
+    current_state = await state.get_state()
+    existing_context: Dict[str, Any] = {}
+    if current_state == TaskPushStates.waiting_supplement.state:
+        existing_context = await state.get_data()
     task = await TASK_SERVICE.get_task(task_id)
     if task is None:
         await state.clear()
         await callback.answer("Task does not exist", show_alert=True)
         return
     actor = _actor_from_callback(callback)
+    origin_message = callback.message
+    origin_message_id = origin_message.message_id if origin_message else None
+    if (
+        current_state == TaskPushStates.waiting_supplement.state
+        and existing_context.get("task_id") == task_id
+        and existing_context.get("origin_message_id") == origin_message_id
+    ):
+        await callback.answer(PUSH_MODEL_SUPPLEMENT_IN_PROGRESS_TEXT)
+        return
+    await state.clear()
     await state.update_data(
         task_id=task_id,
-        origin_message=callback.message,
-        chat_id=callback.message.chat.id if callback.message else callback.from_user.id,
+        origin_message=origin_message,
+        origin_message_id=origin_message_id,
+        chat_id=origin_message.chat.id if origin_message else callback.from_user.id,
         actor=actor,
     )
     await state.set_state(TaskPushStates.waiting_supplement)
@@ -8141,10 +8173,14 @@ async def on_start(m: Message):
         await m.answer(_format_env_issue_message())
 
 @router.message(F.text)
-async def on_text(m: Message):
+async def on_text(m: Message, state: FSMContext | None = None):
     # Automatically record chat when first received message_id to state document
     _auto_record_chat_id(m.chat.id)
 
+    current_state: Optional[str] = None
+    if state is not None:
+        current_state = await state.get_state()
+
     prompt = (m.text or "").strip()
     if not prompt:
         return await m.answer("Please enter a non-empty prompt word")
@@ -8154,6 +8190,12 @@ async def on_text(m: Message):
         return
     if prompt.startswith("/"):
         return
+    if current_state:
+        worker_log.debug(
+            "Suppressed model dispatch due to active wizard state",
+            extra={**_session_extra(), "chat": m.chat.id, "state": current_state},
+        )
+        return
     await _handle_prompt_dispatch(m, prompt)
 
 

{vibego-1.0.10.dist-info → vibego-1.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vibego
-Version: 1.0.10
+Version: 1.0.11
 Summary: vibego CLI: tools for bootstrapping and managing the Telegram Master Bot
 Author: DavidChan
 License-Expression: LicenseRef-Proprietary

{vibego-1.0.10.dist-info → vibego-1.0.11.dist-info}/RECORD

@@ -1,4 +1,4 @@
-bot.py,sha256=bQ-tadaS-CX9699JhO6wtgBeAiXZsbb21GDRAusXJiQ,300928
+bot.py,sha256=CZhCSzPAqoZnL_LDaO_qSNDTP562aU3JvHvafjPUz4s,302812
 logging_setup.py,sha256=ViXK11TNVakIbA8QMLKXa6ywpAV_j1ydcws0Hx2QqNo,5084
 master.py,sha256=bpcqGsUpk8kHW8sqD6QG6ss-5LGO1OsQ7zsdvkgDJ3E,132763
 project_repository.py,sha256=fdO3xzvt7eBLvek_3mXjXkFpALwllrNWDWeSjtscW2I,17675
@@ -427,15 +427,15 @@ tasks/constants.py,sha256=BNMxSnswF-PTij-p7paMK0G5Tj6wKN-6qpUlMwo1JQA,318
 tasks/fsm.py,sha256=6n0hdFNF3j6ZUcpdJ_TwZBgkrE8aMYHNLHppdGbXGO4,1494
 tasks/models.py,sha256=3OJL3F3nVZIQIejL6LplZkPQxYJXgOhCNA9Rikg8ihk,2461
 tasks/service.py,sha256=rzd2v3kZaRpTi54cEAg2GJbeg45VTufC8bGSL1nGJWc,40115
-vibego-1.0.10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vibego_cli/__init__.py,sha256=FPYqYQIBrm5aP_ujpHbtBhSFUsvBDlKfE0fD-RdZFAo,350
+vibego-1.0.11.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+vibego_cli/__init__.py,sha256=vIcx509KPFkxx_WGt0hrTRW9L4_1JAbHnojdgL2RZ7w,350
 vibego_cli/__main__.py,sha256=lM_m-i1x3yVNQhzRt8wqvuIYeeo1OlLNAEVoJVD7tmw,155
 vibego_cli/config.py,sha256=W-n5y4CTrRpencadWlmhpdrhUHX6tdXQEGXJG1pRM0U,3149
 vibego_cli/deps.py,sha256=nFT-DMsfAiD1lLFnDotxe6aYfeeD9V5TrIPbIbAMTF4,1478
 vibego_cli/main.py,sha256=CZAUHWr45GzzEfALoLuSrnvG2VUuCexiFuCpIvyH8Jc,12506
 vibego_cli/data/worker_requirements.txt,sha256=QSt30DSSSHtfucTFPpc7twk9kLS5rVLNTcvDiagxrZg,62
-vibego-1.0.10.dist-info/METADATA,sha256=gnkCSgLPXn8SwqS3DpNp4G8rIdrXxXZPvy3kRkVrHLI,11901
-vibego-1.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vibego-1.0.10.dist-info/entry_points.txt,sha256=Lsy_zm-dlyxt8-9DL9blBReIwU2k22c8-kifr46ND1M,48
-vibego-1.0.10.dist-info/top_level.txt,sha256=R56CT3nW5H5v3ce0l3QDN4-C4qxTrNWzRTwrxnkDX4U,69
-vibego-1.0.10.dist-info/RECORD,,
+vibego-1.0.11.dist-info/METADATA,sha256=miT7a3hXzn39mpDcBtCQHHRY8gLVz0rnNw4_mHccn0g,11901
+vibego-1.0.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vibego-1.0.11.dist-info/entry_points.txt,sha256=Lsy_zm-dlyxt8-9DL9blBReIwU2k22c8-kifr46ND1M,48
+vibego-1.0.11.dist-info/top_level.txt,sha256=R56CT3nW5H5v3ce0l3QDN4-C4qxTrNWzRTwrxnkDX4U,69
+vibego-1.0.11.dist-info/RECORD,,

vibego_cli/__init__.py

@@ -8,6 +8,6 @@ from __future__ import annotations
 
 __all__ = ["main", "__version__"]
 
-__version__ = "1.0.10"
+__version__ = "1.0.11"
 
 from .main import main  # noqa: E402