From 28fd205294013ee361869014956cf5ae9d7e758e Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Tue, 30 May 2023 11:02:30 +0800
Subject: [PATCH] Update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 qa_prompt.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/qa_prompt.py b/qa_prompt.py
index a24724f..60001d5 100644
--- a/qa_prompt.py
+++ b/qa_prompt.py
@@ -1,7 +1,9 @@
 from typing import List, Tuple, Dict, Optional
 
+from towhee.operator import PyOperator
 
-class QAPrompt:
+
+class QAPrompt(PyOperator):
     def __init__(self):
         super().__init__()
         self._template = """Use the following pieces of context to answer the question at the end.