MikeMann commited on
Commit
9547e98
·
1 Parent(s): fa52948

added stepback-Prompting

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -357,11 +357,11 @@ class BSIChatbot:
357
  # )
358
 
359
  # Alles außer letzte Useranfrage, Normaler Query
360
- query, context = self.retrieval(query, True)
361
  if stepBackPrompt == True:
362
  stepBackQuery = self.stepBackPrompt(query)
363
  print("DBG stepBackQuery:" + stepBackQuery)
364
- stepBackQuery, stepBackContext = self.retrieval(stepBackQuery, True)
365
  print("DBG stepBackContext:" + stepBackContext)
366
  sysPrompt = """
367
  You are an helpful Chatbot for the BSI IT-Grundschutz. Using the information contained in the context,
 
357
  # )
358
 
359
  # Alles außer letzte Useranfrage, Normaler Query
360
+ query, context = self.retrieval(query, rerankingStep)
361
  if stepBackPrompt == True:
362
  stepBackQuery = self.stepBackPrompt(query)
363
  print("DBG stepBackQuery:" + stepBackQuery)
364
+ stepBackQuery, stepBackContext = self.retrieval(stepBackQuery, rerankingStep)
365
  print("DBG stepBackContext:" + stepBackContext)
366
  sysPrompt = """
367
  You are an helpful Chatbot for the BSI IT-Grundschutz. Using the information contained in the context,