add log10_session scope

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from log10.load import log10
+from log10.load import log10, log10_session
 import openai
 
 @st.cache_resource
@@ -36,6 +36,10 @@ index = init_vector_db();
 
 st.title('State of the Union')
 query = st.text_input("Question:", "What did the president say about Ketanji Brown Jackson?")
-answer = index.query_with_sources(query, llm=ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo"))
+#answer = index.query_with_sources(query, llm=ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo"))
+
+# If you want to combine all queries into a single session can comment the 2 lines below and uncomment the answer=... line above
+with log10_session():
+    answer = index.query_with_sources(query, llm=ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo"))
 
 st.write(answer)

requirements.txt

@@ -1,4 +1,5 @@
 chromadb
+tiktoken
 streamlit
 langchain
 openai