Tag bot.py, add requirements.txt

adam-cowley · adam-cowley · commit 695b72ed038e · 2023-11-07T11:55:50.000Z
diff --git a/bot.py b/bot.py
@@ -1,6 +1,7 @@
 import streamlit as st
 from utils import write_message
 
+# tag::setup[]
 # Page Config
 st.set_page_config("Ebert", page_icon=":movie_camera:")
 
@@ -9,7 +10,9 @@
     st.session_state.messages = [
         {"role": "assistant", "content": "Hi, I'm the GraphAcademy Chatbot!  How can I help you?"},
     ]
+# end::setup[]
 
+# tag::submit[]
 # Submit handler
 def handle_submit(message):
     """
@@ -22,18 +25,13 @@ def handle_submit(message):
     # Handle the response
     with st.spinner('Thinking...'):
         # # TODO: Replace this with a call to your LLM
-        # from time import sleep
-        # sleep(1)
-        # write_message('assistant', message)
-
-        # from handlers.vector import generate_response
-        # from handlers.cypher import generate_response
-        from handlers.agent import generate_response
-
-        response = generate_response(message)
-        write_message('assistant', response)
+        from time import sleep
+        sleep(1)
+        write_message('assistant', message)
+# end::submit[]
 
 
+# tag::chat[]
 with st.container():
     # Display messages in Session State
     for message in st.session_state.messages:
@@ -46,3 +44,4 @@ def handle_submit(message):
 
         # Generate a response
         handle_submit(prompt)
+# end::chat[]
diff --git a/helloworld/app.py b/helloworld/app.py
@@ -0,0 +1,16 @@
+import streamlit as st
+
+st.set_page_config(
+    page_title="Hello",
+    page_icon="👋",
+)
+
+st.write("# Welcome to Streamlit! 👋")
+
+st.markdown("""
+This component supports **markdown formatting**.
+
+[Check out their documentation](https://docs.streamlit.io) for more information on how to get started.
+""")
+
+st.write('updated')
diff --git a/prompts.py b/prompts.py
@@ -6,6 +6,7 @@
 )
 
 
+# Basic Prompt
 BASIC_PROMPT = PromptTemplate.from_template("""
 You are a movie expert providing customers with movie recommendations.
 Answer the following question to the best of your ability.
@@ -14,7 +15,30 @@
 Answer:
 """)
 
+# RAG Vector Prompts
+GENERAL_SYSTEM_TEMPLATE = """
+You are a movie expert providing customers with movie recommendations.
+Answer the following question to the best of your ability.
+Use the following pieces of context to answer the user's question.
+
+Make sure to rely on information from the answers and not on questions to provide accuate responses.
+
+----
+{summaries}
+----
+
+"""
+
+GENERAL_USER_TEMPLATE = """
+Question:
+
+---
+{question}
+---
+"""
+
 
+# Cypher Prompts
 CYPHER_QA_PROMPT = SystemMessagePromptTemplate.from_template("""
 You are a movie expert providing customers with information about movies.
 Answer the following question to the best of your ability.
@@ -65,4 +89,4 @@
 Use Neo4j 5 Cypher syntax.  When checking a property is not null, use `IS NOT NULL`.
 
 Question: {question}
-"""
+"""
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+langchain==0.0.329
+neo4j-driver==5.12.0
+streamlit==1.28.1

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+langchain==0.0.329`
	`2`	`+neo4j-driver==5.12.0`
	`3`	`+streamlit==1.28.1`