thisisdev commited on
Commit
e95c19b
·
verified ·
1 Parent(s): bb7ce57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import os
3
  import keyfile
4
  import warnings
5
  import streamlit as st
 
6
  warnings.filterwarnings("ignore")
7
 
8
  # Langchain packages
@@ -13,6 +14,11 @@ from langchain.schema import HumanMessage, SystemMessage, AIMessage
13
  st.set_page_config(page_title = "Magical Healer")
14
  st.header("Welcome, What help do you need?")
15
 
 
 
 
 
 
16
  # initializing the sessionMessages
17
  if "sessionMessages" not in st.session_state:
18
  st.session_state["sessionMessages"] = []
 
3
  import keyfile
4
  import warnings
5
  import streamlit as st
6
+ from pydantic import BaseModel
7
  warnings.filterwarnings("ignore")
8
 
9
  # Langchain packages
 
14
  st.set_page_config(page_title = "Magical Healer")
15
  st.header("Welcome, What help do you need?")
16
 
17
+
18
+
19
+ class AIMessage(BaseModel):
20
+ content: str
21
+
22
  # initializing the sessionMessages
23
  if "sessionMessages" not in st.session_state:
24
  st.session_state["sessionMessages"] = []