Update app.py

app.py

@@ -28,7 +28,7 @@ def brave_search(query, count=3):
         return []
 
 @traceable 
-def query_teapot(context, user_input):
+def query_teapot(context, user_input, teapot_ai):
     response = teapot_ai.query(
         context=context,
         query=user_input
@@ -48,7 +48,7 @@ def handle_chat(user_input, teapot_ai):
 
     context="\n".join(documents)
     
-    response = query_teapot(context, user_input)
+    response = query_teapot(context, user_input, teapot_ai)
     
     # response = teapot_ai.chat([
     #     {