115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
|
"""Calendar event search provider."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import uuid
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from sqlalchemy import text
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
class EventSearchProvider:
|
||
|
|
"""Search provider for CalendarEntry entities."""
|
||
|
|
|
||
|
|
entity_type = "event"
|
||
|
|
|
||
|
|
async def search_fts(
|
||
|
|
self,
|
||
|
|
db: AsyncSession,
|
||
|
|
tsquery: str,
|
||
|
|
tenant_id: uuid.UUID,
|
||
|
|
limit: int,
|
||
|
|
) -> list[dict[str, Any]]:
|
||
|
|
"""Full-text search on calendar_entries.search_tsv."""
|
||
|
|
sql = text(
|
||
|
|
"""
|
||
|
|
SELECT e.*, ts_rank(e.search_tsv, to_tsquery('pg_catalog.german', :q)) AS rank
|
||
|
|
FROM calendar_entries e
|
||
|
|
WHERE e.tenant_id = :tid
|
||
|
|
AND e.deleted_at IS NULL
|
||
|
|
AND e.search_tsv @@ to_tsquery('pg_catalog.german', :q)
|
||
|
|
ORDER BY rank DESC
|
||
|
|
LIMIT :lim
|
||
|
|
"""
|
||
|
|
)
|
||
|
|
result = await db.execute(
|
||
|
|
sql,
|
||
|
|
{"q": tsquery, "tid": tenant_id, "lim": limit},
|
||
|
|
)
|
||
|
|
rows = result.mappings().all()
|
||
|
|
return [dict(r) for r in rows]
|
||
|
|
|
||
|
|
async def search_vector(
|
||
|
|
self,
|
||
|
|
db: AsyncSession,
|
||
|
|
embedding: list[float],
|
||
|
|
tenant_id: uuid.UUID,
|
||
|
|
limit: int,
|
||
|
|
) -> list[dict[str, Any]]:
|
||
|
|
"""Semantic search on calendar_entries.embedding."""
|
||
|
|
sql = text(
|
||
|
|
"""
|
||
|
|
SELECT e.*, 1 - (e.embedding <=> cast(:emb AS vector)) AS score
|
||
|
|
FROM calendar_entries e
|
||
|
|
WHERE e.tenant_id = :tid
|
||
|
|
AND e.deleted_at IS NULL
|
||
|
|
AND e.embedding IS NOT NULL
|
||
|
|
ORDER BY e.embedding <=> cast(:emb AS vector)
|
||
|
|
LIMIT :lim
|
||
|
|
"""
|
||
|
|
)
|
||
|
|
result = await db.execute(
|
||
|
|
sql,
|
||
|
|
{"emb": str(embedding), "tid": tenant_id, "lim": limit},
|
||
|
|
)
|
||
|
|
rows = result.mappings().all()
|
||
|
|
return [dict(r) for r in rows]
|
||
|
|
|
||
|
|
async def get_embedding_text(
|
||
|
|
self,
|
||
|
|
db: AsyncSession,
|
||
|
|
entity_id: uuid.UUID,
|
||
|
|
tenant_id: uuid.UUID,
|
||
|
|
) -> str:
|
||
|
|
"""Get text for embedding generation."""
|
||
|
|
sql = text(
|
||
|
|
"""
|
||
|
|
SELECT title, description, location
|
||
|
|
FROM calendar_entries
|
||
|
|
WHERE id = :eid AND tenant_id = :tid
|
||
|
|
"""
|
||
|
|
)
|
||
|
|
result = await db.execute(sql, {"eid": entity_id, "tid": tenant_id})
|
||
|
|
row = result.mappings().first()
|
||
|
|
if not row:
|
||
|
|
return ""
|
||
|
|
parts = [
|
||
|
|
row.get("title", ""),
|
||
|
|
row.get("description", ""),
|
||
|
|
row.get("location", ""),
|
||
|
|
]
|
||
|
|
return " ".join(str(p) for p in parts if p)
|
||
|
|
|
||
|
|
def to_search_result(self, entity: object) -> dict[str, Any]:
|
||
|
|
"""Convert calendar entry to search result dict."""
|
||
|
|
if isinstance(entity, dict):
|
||
|
|
title = entity.get("title", "")
|
||
|
|
description = entity.get("description", "") or ""
|
||
|
|
entity_id = str(entity.get("id", ""))
|
||
|
|
else:
|
||
|
|
title = getattr(entity, "title", "")
|
||
|
|
description = getattr(entity, "description", "") or ""
|
||
|
|
entity_id = str(getattr(entity, "id", ""))
|
||
|
|
return {
|
||
|
|
"entity_type": self.entity_type,
|
||
|
|
"entity_id": entity_id,
|
||
|
|
"title": title,
|
||
|
|
"snippet": description[:200],
|
||
|
|
"score": 0.0,
|
||
|
|
"data": {},
|
||
|
|
}
|