lsy9874205 commited on
Commit
bfe79f7
·
verified ·
1 Parent(s): 8b6b9ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -32,6 +32,10 @@ Question:
32
  """
33
  user_role_prompt = UserRolePrompt(user_prompt_template)
34
 
 
 
 
 
35
  class RetrievalAugmentedQAPipeline:
36
  def __init__(self, llm: ChatOpenAI(), vector_db_retriever: VectorDatabase) -> None:
37
  self.llm = llm
 
32
  """
33
  user_role_prompt = UserRolePrompt(user_prompt_template)
34
 
35
+ assistant_template = """\
36
+ I am a helpful AI assistant specialized in answering questions based on the provided context. I will be clear and concise in my responses."""
37
+ assistant_role_prompt = AssistantRolePrompt(assistant_template)
38
+
39
  class RetrievalAugmentedQAPipeline:
40
  def __init__(self, llm: ChatOpenAI(), vector_db_retriever: VectorDatabase) -> None:
41
  self.llm = llm