feat: Expand LLM support with PromptModel, PromptNode, and PromptTemp…

…late (#3667) Co-authored-by: ZanSara <[email protected]>
deepset-ai · Dec 21, 2022 · a0934f8 · a0934f8
1 parent 71de1ca
commit a0934f8
Show file tree

Hide file tree

Showing 6 changed files with 1,322 additions and 1 deletion.
diff --git a/haystack/nodes/__init__.py b/haystack/nodes/__init__.py
@@ -23,6 +23,7 @@
 from haystack.nodes.label_generator import PseudoLabelGenerator
 from haystack.nodes.other import Docs2Answers, JoinDocuments, RouteDocuments, JoinAnswers, DocumentMerger
 from haystack.nodes.preprocessor import BasePreProcessor, PreProcessor
+from haystack.nodes.prompt import PromptNode, PromptTemplate, PromptModel
 from haystack.nodes.query_classifier import SklearnQueryClassifier, TransformersQueryClassifier
 from haystack.nodes.question_generator import QuestionGenerator
 from haystack.nodes.ranker import BaseRanker, SentenceTransformersRanker

diff --git a/haystack/nodes/prompt/__init__.py b/haystack/nodes/prompt/__init__.py
@@ -0,0 +1 @@
+from haystack.nodes.prompt.prompt_node import PromptNode, PromptTemplate, PromptModel