summaryrefslogtreecommitdiff
path: root/rag/generator/abstract.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-23 00:49:48 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-23 00:49:48 +0200
commitf5adcbb62b10110dc7417c5a07ef6461782f5a10 (patch)
tree397388946c321244be6e4e4f5959b9a0159a1a09 /rag/generator/abstract.py
parentd6ce5be4889e462fa69968feb90f8ebad30cdb0f (diff)
Remove chat
Diffstat (limited to 'rag/generator/abstract.py')
-rw-r--r--rag/generator/abstract.py8
1 files changed, 1 insertions, 7 deletions
diff --git a/rag/generator/abstract.py b/rag/generator/abstract.py
index 71edfc4..1beacfb 100644
--- a/rag/generator/abstract.py
+++ b/rag/generator/abstract.py
@@ -1,5 +1,5 @@
from abc import abstractmethod
-from typing import Any, Dict, Generator, List
+from typing import Any, Generator
from .prompt import Prompt
@@ -16,9 +16,3 @@ class AbstractGenerator(type):
@abstractmethod
def generate(self, prompt: Prompt) -> Generator[Any, Any, Any]:
pass
-
- @abstractmethod
- def chat(
- self, prompt: Prompt, messages: List[Dict[str, str]]
- ) -> Generator[Any, Any, Any]:
- pass