|
from abc import ABC, abstractmethod |
|
|
|
|
|
class AStore(ABC): |
|
''' |
|
Classe abstraite de base pour tous les stores : |
|
Chroma |
|
Perso |
|
... |
|
''' |
|
|
|
@abstractmethod |
|
def reset(self)->None: |
|
pass |
|
|
|
@abstractmethod |
|
def print_infos(self)->None: |
|
pass |
|
|
|
@abstractmethod |
|
def add_to_collection(self, collection_name:str, source:str, vectors:list[list[float]], chunks:list[str])->None: |
|
pass |
|
|
|
@abstractmethod |
|
def delete_collection(self, name:str)->None: |
|
pass |
|
|
|
@abstractmethod |
|
def get_similar_vector(self, vector:list[float], collection_name:str)->list[float]: |
|
pass |
|
|
|
@abstractmethod |
|
def get_similar_chunk(self, query_vector:list[float], collection_name:str)->tuple[str, str]: |
|
pass |
|
|
|
@abstractmethod |
|
def get_similar_chunks(self, query_vector:list[float], count:int, collection_name:str): |
|
pass |
|
|
