openvector_dev 0.1__tar.gz → 0.1.1__tar.gz

{openvector_dev-0.1 → openvector_dev-0.1.1}/PKG-INFO

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

{openvector_dev-0.1 → openvector_dev-0.1.1}/pyproject.toml

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

{openvector_dev-0.1 → openvector_dev-0.1.1}/src/lein_vector/api/facade.py

@@ -1,8 +1,9 @@
+from typing import Sequence, Union
 from uuid import uuid4
 from datetime import datetime, UTC
 
 from src.lein_vector import ShortTermMemory, QdrantAdapter, MemoryManagerQdrant
-from src.lein_vector.schemas.chunk import Chunk
+from src.lein_vector.schemas.chunk import Chunk, ChunkPayload
 from src.lein_vector.sentence_transformer import EmbeddingProviderGemini
 
 
@@ -72,6 +73,7 @@ class MemoryFacade:
         """
         data = await self.step_user(user_id, user_msg, topk=topk, history_n=history_n)
         data["short_term"] = self._to_openai(data["short_term"])
+        data["long_term"] = self._chunk_texts(data["long_term"])
         return data
 
     @staticmethod
@@ -134,3 +136,8 @@ class MemoryFacade:
 
     async def add_to_long(self, user_id: int, chunk: Chunk) -> None:
         await self.long.upsert_chunk(user_id, chunk)
+
+    @staticmethod
+    def _chunk_texts(chunks: Sequence[Union[Chunk, ChunkPayload]]) -> list[str]:
+        """Вернуть список текстов из любых Chunk/ChunkPayload."""
+        return [c.text for c in chunks]

{openvector_dev-0.1 → openvector_dev-0.1.1}/src/lein_vector/schemas/chunk.py

@@ -36,4 +36,3 @@ class ChunkPayload(BaseModel):
 
     def to_chunk(self, last_hit: datetime = None, hit_count: int = 0) -> Chunk:
         return Chunk(**self.model_dump(), last_hit=last_hit or datetime.now(UTC), hit_count=hit_count)
-