blob: 3ce997e102e6dad52d4da14348e899a7b7357ee1 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
from abc import abstractmethod
from typing import Any, Generator, List
from rag.message import Message
from rag.retriever.vector import Document
class AbstractGenerator(type):
_instances = {}
def __call__(cls, *args, **kwargs):
if cls not in cls._instances:
instance = super().__call__(*args, **kwargs)
cls._instances[cls] = instance
return cls._instances[cls]
@abstractmethod
def generate(self, messages: List[Message], documents: List[Document]) -> Generator[Any, Any, Any]:
pass
|
