Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import keyfile
 import warnings
 import streamlit as st
+from pydantic import BaseModel
 warnings.filterwarnings("ignore")
 
 # Langchain packages
@@ -13,6 +14,11 @@ from langchain.schema import HumanMessage, SystemMessage, AIMessage
 st.set_page_config(page_title = "Magical Healer")
 st.header("Welcome, What help do you need?")
 
+
+
+class AIMessage(BaseModel):
+    content: str
+    
 # initializing the sessionMessages
 if "sessionMessages" not in st.session_state:
     st.session_state["sessionMessages"] = []