mirror of
https://github.com/StanGirard/quivr.git
synced 2024-12-24 20:03:41 +03:00
512b9b4f37
- Add `QAInterface` which should be implemented by all custom answer generator to be compatible with Quivr - Add `RAGInterface` which should be implemented by all RAG classes
28 lines
763 B
Python
28 lines
763 B
Python
from abc import ABC, abstractmethod
|
|
from uuid import UUID
|
|
|
|
from modules.chat.dto.chats import ChatQuestion
|
|
|
|
|
|
class QAInterface(ABC):
|
|
"""
|
|
Abstract class for all QA interfaces.
|
|
This can be used to implement custom answer generation logic.
|
|
"""
|
|
|
|
@abstractmethod
|
|
def generate_answer(
|
|
self, chat_id: UUID, question: ChatQuestion, should, *custom_params: tuple
|
|
):
|
|
raise NotImplementedError(
|
|
"generate_answer is an abstract method and must be implemented"
|
|
)
|
|
|
|
@abstractmethod
|
|
def generate_stream(
|
|
self, chat_id: UUID, question: ChatQuestion, *custom_params: tuple
|
|
):
|
|
raise NotImplementedError(
|
|
"generate_stream is an abstract method and must be implemented"
|
|
)
|