Spaces:
Running
Running
| """Base class for memory providers.""" | |
| import abc | |
| import openai | |
| from autogpt.config import AbstractSingleton, Config | |
| cfg = Config() | |
| def get_ada_embedding(text): | |
| text = text.replace("\n", " ") | |
| if cfg.use_azure: | |
| return openai.Embedding.create( | |
| input=[text], | |
| engine=cfg.get_azure_deployment_id_for_model("text-embedding-ada-002"), | |
| )["data"][0]["embedding"] | |
| else: | |
| return openai.Embedding.create(input=[text], model="text-embedding-ada-002")[ | |
| "data" | |
| ][0]["embedding"] | |
| class MemoryProviderSingleton(AbstractSingleton): | |
| def add(self, data): | |
| pass | |
| def get(self, data): | |
| pass | |
| def clear(self): | |
| pass | |
| def get_relevant(self, data, num_relevant=5): | |
| pass | |
| def get_stats(self): | |
| pass | |