| """Abstract base class for vector database providers""" |
|
|
| from abc import ABC, abstractmethod |
| from typing import List, Dict, Any, Optional |
|
|
|
|
| class VectorDBProvider(ABC): |
| """Abstract base class for vector database implementations""" |
|
|
| @abstractmethod |
| def initialize(self, config: Dict[str, Any]) -> None: |
| """Initialize database connection""" |
| pass |
|
|
| @abstractmethod |
| def add_vectors( |
| self, |
| vectors: List[List[float]], |
| ids: List[str], |
| metadata: List[Dict[str, Any]], |
| ) -> bool: |
| """Add vectors to database""" |
| pass |
|
|
| @abstractmethod |
| def search( |
| self, |
| query_vector: List[float], |
| top_k: int = 5, |
| ) -> List[Dict[str, Any]]: |
| """Search for similar vectors""" |
| pass |
|
|
| @abstractmethod |
| def delete(self, doc_id: str) -> bool: |
| """Delete document from database""" |
| pass |
|
|
| @abstractmethod |
| def health_check(self) -> bool: |
| """Check database connection health""" |
| pass |
|
|
| @abstractmethod |
| def get_stats(self) -> Dict[str, Any]: |
| """Get database statistics""" |
| pass |
|
|
