coding-alt/AutoGPT

1"""Base class for memory providers."""2import abc3 4import openai5 6from autogpt.config import AbstractSingleton, Config7 8cfg = Config()9 10 11def get_ada_embedding(text):12    text = text.replace("\n", " ")13    if cfg.use_azure:14        return openai.Embedding.create(15            input=[text],16            engine=cfg.get_azure_deployment_id_for_model("text-embedding-ada-002"),17        )["data"][0]["embedding"]18    else:19        return openai.Embedding.create(input=[text], model="text-embedding-ada-002")[20            "data"21        ][0]["embedding"]22 23 24class MemoryProviderSingleton(AbstractSingleton):25    @abc.abstractmethod26    def add(self, data):27        pass28 29    @abc.abstractmethod30    def get(self, data):31        pass32 33    @abc.abstractmethod34    def clear(self):35        pass36 37    @abc.abstractmethod38    def get_relevant(self, data, num_relevant=5):39        pass40 41    @abc.abstractmethod42    def get_stats(self):43        pass44