openvector_dev 0.1.13__tar.gz → 0.1.15__tar.gz

{openvector_dev-0.1.13 → openvector_dev-0.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openvector_dev
-Version: 0.1.13
+Version: 0.1.15
 Summary: 
 Author: p00ler
 Author-email: liveitspain@gmail.com

{openvector_dev-0.1.13 → openvector_dev-0.1.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openvector_dev"
-version = "0.1.13"
+version = "0.1.15"
 description = ""
 authors = [
     {name = "p00ler",email = "liveitspain@gmail.com"}

{openvector_dev-0.1.13 → openvector_dev-0.1.15}/src/lein_vector/api/facade.py

@@ -1,6 +1,6 @@
+import random
 from collections.abc import Sequence
 from datetime import UTC, datetime
-from typing import Optional, Any
 from uuid import UUID, uuid4
 
 import redis.asyncio as aioredis
@@ -24,6 +24,10 @@ class Memory:
         self._msg_no: dict[int, int] = {}
         self.merge_n = merge_n
 
+        self._gift_base = 0.02
+        self._gift_growth = 1.15
+        self._gift_fail: dict[tuple[int, str], int] = {}
+
         self.score_threshold = 0.72
 
     @classmethod
@@ -119,7 +123,9 @@ class Memory:
             await self.long.upsert_chunk_with_vector(new_chunk, vector)
 
         if curr_no % (block_size * self.merge_n) == 0:
-            merged_text, used_ids = await self.long.merge_old_chunks(user_id=user_id, bot=bot, chunk_type="type0", n=self.merge_n)
+            merged_text, used_ids = await self.long.merge_old_chunks(
+                user_id=user_id, bot=bot, chunk_type="type0", n=self.merge_n
+            )
             return merged_text, used_ids
         else:
             return None, None
@@ -159,19 +165,35 @@ class Memory:
         return self._to_openai(data)
 
     async def add_short_msg(
-        self, user_id: int, bot: str, text: str, *, role: str = "user"
-    ):
+        self,
+        user_id: int,
+        bot: str,
+        text: str,
+        *,
+        role: str = "user",
+    ) -> bool:
         """
-        Добавляет в "короткую память" новое сообщение
-        :param user_id: Идентификатор пользователя
-        :param bot: Кодовое имя бота
-        :param text: Сообщение для записи
-        :param role: Роль в диалоге
-        :type role: str = "user"
-        :return:
+        Записывает сообщение и возвращает флаг `is_gift_hint`.
+        Шанс = min(1, p0 * growth**fails).
         """
         await self.short.add(bot=bot, user_id=user_id, role=role, text=text)
 
+        if role != "user":
+            return False
+
+        key = (user_id, bot)
+        fails = self._gift_fail.get(key, 0)
+
+        p0, g = self._gift_base, self._gift_growth
+        p = min(1.0, p0 * (g**fails))
+
+        if random.random() < p:
+            self._gift_fail[key] = 0
+            return True
+        else:
+            self._gift_fail[key] = fails + 1
+            return False
+
     async def add_summary_chunk(
         self,
         user_id: int,
@@ -221,12 +243,22 @@ class Memory:
 
     async def delete_memory(self, user_id: int, bot: str) -> None:
         """
-        Удаляет
-        :param user_id: Идентификатор пользователя.
-        :param bot: Codename бота
+        Полная очистка памяти данного (user_id, bot):
+          • short-term  (Redis)
+          • long-term   (Qdrant)
+          • внутренние счётчики фасада (_msg_no, _gift_fail)
         """
+        # 1) short-term (Redis list)
+        await self.short.clear(user_id=user_id, bot=bot)
+
+        # 2) long-term  (Qdrant)
         await self.long.delete_all(user_id, bot)
 
+        # 3) локальные счётчики
+        self._msg_no.pop((user_id, bot), None)
+        self._gift_fail.pop((user_id, bot), None)
+
+
     @staticmethod
     def _chunk_texts(chunks: Sequence[Chunk | ChunkPayload]) -> list[str]:
         """

{openvector_dev-0.1.13 → openvector_dev-0.1.15}/src/lein_vector/memory_manager_qdrant.py

@@ -1,6 +1,6 @@
 import asyncio
 from datetime import UTC, datetime
-from typing import Any, Tuple, List
+from typing import Any
 from uuid import UUID
 
 from lein_vector.bases.memory_manager_abc import MemoryManagerABC
@@ -95,7 +95,7 @@ class MemoryManagerQdrant(MemoryManagerABC):
             bot: str,
             chunk_type: str,
             n: int = 5,
-    ) -> Tuple[str | None, List[UUID]]:
+    ) -> tuple[str | None, list[UUID]]:
         """
         Собрать n старых чанков и вернуть:
           • merged_text — склеенные сообщения c ролями,