openvector_dev 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

lein_vector/__init__.py

@@ -3,14 +3,12 @@ from .memory_manager_qdrant import MemoryManagerQdrant
 from .qdrant_adapter import QdrantAdapter
 from .redis_short_term import RedisShortTermMemory
 from .sentence_transformer import EmbeddingProviderGemini as EmbeddingProvider
-from .short_term import ShortTermMemory
 
 __all__ = [
     "EmbeddingProvider",
     "MemoryManagerQdrant",
     "QdrantAdapter",
     "RedisShortTermMemory",
-    "ShortTermMemory",
     "api",
     "bases",
     "schemas",

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,

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 ролями,

{openvector_dev-0.1.12.dist-info → openvector_dev-0.1.14.dist-info}/METADATA

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

{openvector_dev-0.1.12.dist-info → openvector_dev-0.1.14.dist-info}/RECORD

@@ -1,16 +1,16 @@
-lein_vector/__init__.py,sha256=ieXy65X4eVzUtSHNvlh00s4aMwKReafNni4s2q68z3Q,496
+lein_vector/__init__.py,sha256=48LqJbNqEEpbTFDkXA4BoF7KZ7z8FAKdj9NJ_raKJxk,431
 lein_vector/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lein_vector/api/facade.py,sha256=s8tHif2zUJuS3eqGQQ14Zq-6sHjzW73_2V6P1qFne8M,9871
+lein_vector/api/facade.py,sha256=PpzGUEiCOoApLWtXVB4RTYV9I16ajU_R6_-24kJt6Ug,10237
 lein_vector/bases/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lein_vector/bases/embeding_provider_abc.py,sha256=WBpVC6ra-SYqhJeZs8R8U679wfGebXgwOOVPIii-IvY,265
 lein_vector/bases/memory_manager_abc.py,sha256=NE13Lz-3aX3RjkRhsd5FsezPNk0xju34fFpO3Fa1n9A,1755
-lein_vector/memory_manager_qdrant.py,sha256=jzb04gnESVRH62QSYhnwsl-HpS6Kci721rPA3_6LSDQ,6382
+lein_vector/memory_manager_qdrant.py,sha256=K41sbRYz168-yYO9cb9DOl7XWc7nGfWGFvcEz0iu3KE,6369
 lein_vector/memory_manager_ram.py,sha256=lUATu6U-cZuRHnR6U1L8aJQB8FNK0Wi40vXsZ5fjDgI,6002
 lein_vector/qdrant_adapter.py,sha256=CJsGvlDdTf4gGCFZIRzfYDeOT_7pzw8pf4VT_QIYDWg,4466
 lein_vector/redis_short_term.py,sha256=uCcncjgXAlNAY46b3_Pjvrn_G0NEI16kRRR9hxayzak,3155
 lein_vector/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lein_vector/schemas/chunk.py,sha256=qq2J7UM94AhKXKg_gnErac5GQF9DCoaEeEHyxZg_xuk,1202
 lein_vector/sentence_transformer.py,sha256=SjjIa43Y7JxcU9VkpT2Ml-CVXb8tqmrbMutZN6O_I6s,1853
-openvector_dev-0.1.12.dist-info/METADATA,sha256=dNLKK0aun5DoVYmLFUOFne1WSw8PFjKbhm7W0gttB7I,3168
-openvector_dev-0.1.12.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-openvector_dev-0.1.12.dist-info/RECORD,,
+openvector_dev-0.1.14.dist-info/METADATA,sha256=oKKk2vWWDCgqs9-MGwgGhsKcrOtMbNRCgPNk02bKIpA,3168
+openvector_dev-0.1.14.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+openvector_dev-0.1.14.dist-info/RECORD,,