UPDATE: follow ups

functions.py

@@ -242,7 +242,7 @@ def answerQuery(query: str, vectorstore: str, llmModel: str = "llama3-70b-8192")
  history_messages_key="chatHistory"
  )
  chain = RunnablePassthrough.assign(messages_trimmed=trimMessages) | messageChain
- followUpChain = prompt | ChatGroq(model_name="gemma2-9b-it", temperature=0) | jsonParser
+ followUpChain = followUpPrompt | ChatGroq(model_name="gemma2-9b-it", temperature=0) | jsonParser
  output = chain.invoke(
  {"question": query},
  {"configurable": {"session_id": vectorStoreName}}