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

lein_vector/api/facade.py

@@ -32,6 +32,7 @@ class Memory:
         host: str,
         port: int,
         collection: str,
+        redis_url: str,
         vector_size: int = 768,
         api_key: str | None = None,
         short_maxlen: int = 20,
@@ -41,12 +42,13 @@ class Memory:
         :param host: Адрес Qdrant.
         :param port: Порт Qdrant.
         :param collection: Название коллекции Qdrant.
+        :param redis_url: Ссылка на подключение к Redis
         :param vector_size: Размерность векторного пространства.
         :param api_key: Ключ для эмбеддера (если требуется).
         :param short_maxlen: Максимальная длина окна кратковременной памяти.
         :returns: Экземпляр Memory с инициализированными зависимостями.
         """
-        _redis = aioredis.from_url("redis://localhost:6379", decode_responses=True)
+        _redis = aioredis.from_url(redis_url, decode_responses=True)
         short_mem = RedisShortTermMemory(_redis, maxlen=short_maxlen)
         embedder = EmbeddingProviderGemini(api_key=api_key)
         adapter = QdrantAdapter(host, port, collection, vector_size)
@@ -117,7 +119,8 @@ class Memory:
             await self.long.upsert_chunk_with_vector(new_chunk, vector)
 
         if curr_no % (block_size * self.merge_n) == 0:
-            return 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
 

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

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

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

@@ -1,6 +1,6 @@
 lein_vector/__init__.py,sha256=ieXy65X4eVzUtSHNvlh00s4aMwKReafNni4s2q68z3Q,496
 lein_vector/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lein_vector/api/facade.py,sha256=Z5jHOoU4iYiJ3hGrhpOEJbJvx_grp2YLgmeQbuCrkTo,9725
+lein_vector/api/facade.py,sha256=s8tHif2zUJuS3eqGQQ14Zq-6sHjzW73_2V6P1qFne8M,9871
 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
@@ -11,7 +11,6 @@ lein_vector/redis_short_term.py,sha256=uCcncjgXAlNAY46b3_Pjvrn_G0NEI16kRRR9hxayz
 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
-lein_vector/short_term.py,sha256=KFot9r26d9TZYLscVT42V0NwufIQUkx-hftOe2p7qLU,1933
-openvector_dev-0.1.10.dist-info/METADATA,sha256=KxlvCA-i9l8m4M8e_6aVRWW8FFTiwo5UYSAhz6k3uwM,3168
-openvector_dev-0.1.10.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-openvector_dev-0.1.10.dist-info/RECORD,,
+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,,

lein_vector/short_term.py

@@ -1,48 +0,0 @@
-from collections import deque
-from datetime import datetime
-
-
-class ShortTermMemory:
-    def __init__(self, maxlen: int = 10):
-        self._buffer: deque = deque(maxlen=maxlen)
-
-    def add(
-        self,
-        codename: str,
-        user_id: int,
-        role: str,
-        text: str,
-        ts: datetime | None = None,
-        **extra
-    ) -> None:
-        """Добавить сообщение в память (роль, текст, ts - время, по умолчанию now)."""
-        if ts is None:
-            ts = datetime.now()
-        self._buffer.append({"role": role, "text": text, "ts": ts, **extra})
-
-    def window(self, n: int | None = None) -> list[dict]:
-        """Получить последние n сообщений (по умолчанию все)."""
-        if n is None or n > len(self._buffer):
-            return list(self._buffer)
-        return list(self._buffer)[-n:]
-
-    def clear(self) -> None:
-        """Очистить память."""
-        self._buffer.clear()
-
-    def load(self, history: list[dict]) -> None:
-        """Инициализировать память списком сообщений."""
-        self._buffer.clear()
-        for msg in history[-self._buffer.maxlen :]:
-            self._buffer.append(msg)
-
-    def to_list(self) -> list[dict]:
-        """Выгрузить всю память как список."""
-        return list(self._buffer)
-
-    def chunk_for_vector(self, chunk_size: int = 6) -> list[dict] | None:
-        """Сформировать чанк для векторной БД — N последних сообщений по хронологии."""
-        if len(self._buffer) < chunk_size:
-            return None
-        # Забираем chunk_size старейших, но не очищаем (внешний код решает)
-        return list(self._buffer)[-chunk_size:]