| import hashlib |
| from pathlib import Path |
|
|
| from app.config import Settings |
| from app.database.supabase import SupabaseRepository |
| from app.services.embedding_service import JinaEmbeddingService |
| from app.services.trip_indexing import build_trip_embedding_record |
|
|
|
|
| class AdminService: |
| def __init__( |
| self, |
| *, |
| repository: SupabaseRepository, |
| embeddings: JinaEmbeddingService, |
| settings: Settings, |
| info_path: Path | None = None, |
| ) -> None: |
| self.repository = repository |
| self.embeddings = embeddings |
| self.settings = settings |
| self.info_path = info_path or Path("prompts/falzh_info.md") |
|
|
| async def seed_info(self) -> int: |
| text = self.info_path.read_text(encoding="utf-8") |
| chunk_texts = _chunk_markdown(text) |
| embeddings = await self.embeddings.embed_passages(chunk_texts) |
| chunks = [] |
| for chunk, embedding in zip(chunk_texts, embeddings, strict=True): |
| digest = hashlib.sha256(chunk.encode("utf-8")).hexdigest()[:24] |
| chunks.append( |
| { |
| "id": f"info-{digest}", |
| "chunk_text": chunk, |
| "source": str(self.info_path), |
| "embedding": embedding, |
| "embedding_model": self.settings.jina_embedding_model, |
| } |
| ) |
| return await self.repository.upsert_info_chunks(chunks) |
|
|
| async def sync_trips(self) -> int: |
| trips = await self.repository.list_active_trips() |
| trip_records = [ |
| build_trip_embedding_record(trip, self.settings.jina_embedding_model) |
| for trip in trips |
| ] |
| embeddings = await self.embeddings.embed_passages( |
| [record["chunk_text"] for record in trip_records] |
| ) |
| for record, embedding in zip(trip_records, embeddings, strict=True): |
| record["embedding"] = embedding |
| return await self.repository.upsert_trip_embeddings(trip_records) |
|
|
|
|
| def _chunk_markdown(text: str, *, max_chars: int = 1200) -> list[str]: |
| sections = [section.strip() for section in text.split("\n## ") if section.strip()] |
| chunks: list[str] = [] |
| for index, section in enumerate(sections): |
| content = section if index == 0 else f"## {section}" |
| if len(content) <= max_chars: |
| chunks.append(content) |
| continue |
| for start in range(0, len(content), max_chars): |
| chunks.append(content[start : start + max_chars].strip()) |
| return chunks |
|
|
