Agent Zero
8cebb4f4e9
- unified_search: Hybride Suche (PostgreSQL FTS + pgvector + RRF Fusion) - 5 Search Providers (Contact, Company, Mail, File, Event) - KI Query Understanding (Fuzzy, Facetten via LiteLLM) - DMS Text-Extraction (PDF, DOCX, XLSX, PPTX) - Embedding Pipeline (ollama/nomic-embed-text, 768 Dim) - Background Jobs für Indexierung - Plugin-basierte Provider Registry - ai_proactive: Proaktiver KI-Agent - Context-Tracking (Frontend → Backend → Event Bus) - Proactive Engine mit LLM Suggestion-Generierung - SSE Real-time Push an Frontend - 6 AI Tools für Tool Registry - Rate-Limiting + User Settings - Deep Analysis Background Jobs - Frontend Integration: - useAIContext Hook, SuggestionSidebar, SuggestionBadge - ProactiveAISettings Page, Search API Client - Globale Suche auf neue API umgestellt - Tests: test_unified_search.py + test_ai_proactive.py (alle bestanden) - Config: Ollama Cloud DeepSeek V4 als Default, konfigurierbar - Dependencies: PyMuPDF, python-docx, python-pptx, pgvector - Bugfixes: notification type_key length, migration IF NOT EXISTS
134 lines
3.7 KiB
Python
134 lines
3.7 KiB
Python
"""SearchProvider protocol and global registry."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import TYPE_CHECKING, Protocol, runtime_checkable
|
|
|
|
from sqlalchemy import select
|
|
|
|
if TYPE_CHECKING:
|
|
import uuid
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@runtime_checkable
|
|
class SearchProvider(Protocol):
|
|
"""Protocol for entity-specific search providers."""
|
|
|
|
entity_type: str
|
|
|
|
async def search_fts(
|
|
self,
|
|
db: AsyncSession,
|
|
tsquery: str,
|
|
tenant_id: uuid.UUID,
|
|
limit: int,
|
|
) -> list[dict]:
|
|
"""Full-text search using PostgreSQL tsquery."""
|
|
...
|
|
|
|
async def search_vector(
|
|
self,
|
|
db: AsyncSession,
|
|
embedding: list[float],
|
|
tenant_id: uuid.UUID,
|
|
limit: int,
|
|
) -> list[dict]:
|
|
"""Semantic vector search using pgvector."""
|
|
...
|
|
|
|
async def get_embedding_text(
|
|
self,
|
|
db: AsyncSession,
|
|
entity_id: uuid.UUID,
|
|
tenant_id: uuid.UUID,
|
|
) -> str:
|
|
"""Get text representation for embedding generation."""
|
|
...
|
|
|
|
def to_search_result(self, entity: object) -> dict:
|
|
"""Convert an ORM entity to a search result dict."""
|
|
...
|
|
|
|
|
|
class SearchProviderRegistry:
|
|
"""In-memory registry of search providers."""
|
|
|
|
def __init__(self) -> None:
|
|
self._providers: dict[str, SearchProvider] = {}
|
|
|
|
def register(self, provider: SearchProvider) -> None:
|
|
"""Register a search provider."""
|
|
self._providers[provider.entity_type] = provider
|
|
logger.debug("Registered search provider: %s", provider.entity_type)
|
|
|
|
def unregister(self, entity_type: str) -> None:
|
|
"""Unregister a search provider by entity type."""
|
|
self._providers.pop(entity_type, None)
|
|
|
|
def get(self, entity_type: str) -> SearchProvider | None:
|
|
"""Get a provider by entity type."""
|
|
return self._providers.get(entity_type)
|
|
|
|
def get_all(self) -> list[SearchProvider]:
|
|
"""Get all registered providers."""
|
|
return list(self._providers.values())
|
|
|
|
def get_entity_types(self) -> list[str]:
|
|
"""Get all registered entity types."""
|
|
return list(self._providers.keys())
|
|
|
|
def clear(self) -> None:
|
|
"""Clear all registered providers."""
|
|
self._providers.clear()
|
|
|
|
|
|
# Global singleton
|
|
_registry = SearchProviderRegistry()
|
|
|
|
|
|
def get_search_registry() -> SearchProviderRegistry:
|
|
"""Get the global search provider registry."""
|
|
return _registry
|
|
|
|
|
|
async def auto_register_providers(db: AsyncSession) -> None:
|
|
"""Auto-register providers for active plugins.
|
|
|
|
Checks which plugins are active and registers corresponding providers.
|
|
"""
|
|
from app.plugins.builtins.unified_search.providers.contact_provider import (
|
|
ContactSearchProvider,
|
|
)
|
|
from app.plugins.builtins.unified_search.providers.company_provider import (
|
|
CompanySearchProvider,
|
|
)
|
|
from app.plugins.builtins.unified_search.providers.mail_provider import (
|
|
MailSearchProvider,
|
|
)
|
|
from app.plugins.builtins.unified_search.providers.file_provider import (
|
|
FileSearchProvider,
|
|
)
|
|
from app.plugins.builtins.unified_search.providers.event_provider import (
|
|
EventSearchProvider,
|
|
)
|
|
|
|
registry = get_search_registry()
|
|
registry.clear()
|
|
|
|
# Register all built-in providers
|
|
for provider_cls in [
|
|
ContactSearchProvider,
|
|
CompanySearchProvider,
|
|
MailSearchProvider,
|
|
FileSearchProvider,
|
|
EventSearchProvider,
|
|
]:
|
|
try:
|
|
registry.register(provider_cls())
|
|
except Exception:
|
|
logger.exception("Failed to register %s", provider_cls.__name__)
|