openvector_dev 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

lein_vector/api/facade.py

@@ -24,6 +24,8 @@ class Memory:
         self._msg_no: dict[int, int] = {}
         self.merge_n = merge_n
 
+        self.score_threshold = 0.72
+
     @classmethod
     async def from_qdrant(
         cls,
@@ -207,33 +209,21 @@ class Memory:
         data = await self.short.window(user_id, bot, n_memories)
         return self._to_openai(data)
 
-    async def _get_long_term(
-        self, user_id: int, embedding: list[float], topk: int = 3
-    ) -> list:
-        """
-        :param user_id: Идентификатор пользователя.
-        :param embedding: Вектор эмбеддинга для поиска.
-        :param topk: Количество возвращаемых чанков.
-        :returns: Список чанков.
-        """
-        return await self.long.retrieve_by_embedding(user_id, embedding, topk)
-
-    async def _add_to_short(self, role: str, text: str) -> None:
-        """
-        :param role: Роль сообщения.
-        :param text: Текст сообщения.
+    async def add_short_msg(self, user_id: int, bot: str, text: str, *, role: str = "user"):
         """
-        await self.short.add(role, text)
-
-    async def _add_to_long(self, user_id: int, chunk: Chunk) -> None:
-        """
-        :param user_id: Идентификатор пользователя.
-        :param chunk: Экземпляр Chunk для сохранения.
+        Добавляет в "короткую память" новое сообщение
+        :param user_id: Идентификатор пользователя
+        :param bot: Кодовое имя бота
+        :param text: Сообщение для записи
+        :param role: Роль в диалоге
+        :type role: str = "user"
+        :return:
         """
-        await self.long.upsert_chunk(user_id, chunk)
+        await self.short.add(bot=bot, user_id=user_id, role=role, text=text)
 
     async def delete_memory(self, user_id: int) -> None:
         """
+        Удаляет
         :param user_id: Идентификатор пользователя.
         """
         await self.long.delete_all(user_id)

lein_vector/sentence_transformer.py

@@ -22,11 +22,10 @@ class EmbeddingProviderGemini(EmbeddingProviderABC):
     async def get_embedding(self, text: str) -> list[float]:
         # В Gemini SDK обычно нет async, значит — обёртка через run_in_executor:
         import asyncio
-
+        if not isinstance(text, str):
+            return
         loop = asyncio.get_running_loop()
-        # SDK call внутри executor
         embedding = await loop.run_in_executor(None, lambda: self._genai_embed(text))
-        # Ответ — dict с 'embedding' или 'values'
         return embedding.embeddings[0].values
 
     async def get_embeddings(self, texts: list[str]) -> list[list[float]]:

{openvector_dev-0.1.6.dist-info → openvector_dev-0.1.7.dist-info}/METADATA

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

{openvector_dev-0.1.6.dist-info → openvector_dev-0.1.7.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=YHBClaBpPjBuR-lxeJ-BSbDZOsXaG4Pw3wV9FckWWfY,11045
+lein_vector/api/facade.py,sha256=f0Cwib5maw4Ae9QEmc5xDkx4Y2ozIzDQUqJ7jOdg2pg,10634
 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=pWuIw_wYlvUlt40jbiE92lWzdYaowZKg3XK0Mq0tQrw,1687
@@ -10,8 +10,8 @@ lein_vector/qdrant_adapter.py,sha256=uPhYAp0KmYQhY3DF29JMO1JTbM__xDWGlnZZPor3VKI
 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=FSUwaFUSZk2ptoxupsdIfj5HYAMnHnu3eBfNpXCUMpM,1195
-lein_vector/sentence_transformer.py,sha256=ep6Y9qu8O4_qK4q14BPNxtSe0gBi-mH04xfy1-dhlUE,1898
+lein_vector/sentence_transformer.py,sha256=gbWtkT01QFFA0IR_Ta-xsfBfb6owDzGnZDKY1g9fvBQ,1851
 lein_vector/short_term.py,sha256=KFot9r26d9TZYLscVT42V0NwufIQUkx-hftOe2p7qLU,1933
-openvector_dev-0.1.6.dist-info/METADATA,sha256=alAGaEp-xEOGjej2VH3f9FaxxWkLOaCRxaHFe5gyMMo,3167
-openvector_dev-0.1.6.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-openvector_dev-0.1.6.dist-info/RECORD,,
+openvector_dev-0.1.7.dist-info/METADATA,sha256=jNE4MlXel9eZ0iNEIg-JaFdIcAOCqBdxVUdVlNoKBqE,3167
+openvector_dev-0.1.7.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+openvector_dev-0.1.7.dist-info/RECORD,,