TomData commited on
Commit
cef6758
·
1 Parent(s): 85e76f5

change llm response formatting

Browse files
Files changed (1) hide show
  1. src/chatbot.py +1 -1
src/chatbot.py CHANGED
@@ -55,7 +55,7 @@ db = get_vectorstore(embeddings=embeddings, folder_path=folder_path, index_name=
55
 
56
  def chatbot(message, history, db=db, llm=llm, prompt=prompt2):
57
  raw_response = RAG(llm=llm, prompt=prompt, db=db, question=message)
58
- response = raw_response['answer'].split("Antwort: ")[1]
59
  return response
60
 
61
  # Retrieve speech contents based on keywords
 
55
 
56
  def chatbot(message, history, db=db, llm=llm, prompt=prompt2):
57
  raw_response = RAG(llm=llm, prompt=prompt, db=db, question=message)
58
+ response = raw_response['answer']#.split("Antwort: ")[1]
59
  return response
60
 
61
  # Retrieve speech contents based on keywords