openvector_dev 0.1.9__tar.gz → 0.1.11__tar.gz

{openvector_dev-0.1.9 → openvector_dev-0.1.11}/PKG-INFO

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

{openvector_dev-0.1.9 → openvector_dev-0.1.11}/pyproject.toml

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

{openvector_dev-0.1.9 → openvector_dev-0.1.11}/src/lein_vector/api/facade.py

@@ -1,5 +1,6 @@
 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
@@ -114,10 +115,12 @@ class Memory:
                 extra={"msg_no": curr_no},
             )
             await self.long.upsert_chunk_with_vector(new_chunk, vector)
-            # self.short.clear_until(block_size)
 
         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
 
     async def get_long_memories(
         self, user_id: int, bot: str, search_terms: list[str], topk: int = 3
@@ -218,8 +221,9 @@ class Memory:
         """
         Удаляет
         :param user_id: Идентификатор пользователя.
+        :param bot: Codename бота
         """
-        await self.long.delete_all(user_id)
+        await self.long.delete_all(user_id, bot)
 
     @staticmethod
     def _chunk_texts(chunks: Sequence[Chunk | ChunkPayload]) -> list[str]:

{openvector_dev-0.1.9 → openvector_dev-0.1.11}/src/lein_vector/qdrant_adapter.py

@@ -109,13 +109,21 @@ class QdrantAdapter:
     async def delete_collection(self) -> None:
         await self.client.delete_collection(collection_name=self.collection)
 
-    async def get_all_chunks(self, user_id: int) -> list[ChunkPayload]:
+    async def get_all_chunks(self, user_id: int, bot: str) -> list[ChunkPayload]:
+        """
+        Вернуть ВСЕ чанки заданного пользователя и конкретного бота.
+        """
         q_filter = Filter(
-            must=[FieldCondition(key="user_id", match=MatchValue(value=user_id))]
+            must=[
+                FieldCondition(key="user_id", match=MatchValue(value=user_id)),
+                FieldCondition(key="bot",      match=MatchValue(value=bot)),
+            ]
         )
+
         scroll = await self.client.scroll(
             collection_name=self.collection,
             scroll_filter=q_filter,
             limit=2048,
         )
+
         return [ChunkPayload(**p.payload) for p in scroll[0]]