Remove metadata

pages/chatbot.py

@@ -36,11 +36,6 @@ def display_messages():
 
 def launchQuery(query: str = None):
 
-    metadata = {
-        "production": os.environ.get("env") == "production"
-    }
-    print(metadata)
-
     # Initialize the assistant's response
     full_response = st.write_stream(
         st.session_state["assistant"].ask(
@@ -48,7 +43,7 @@ def launchQuery(query: str = None):
             prompt_system=st.session_state.prompt_system,
             messages=st.session_state[chapter_session_key]["messages"] if "messages" in st.session_state[chapter_session_key] else [],
             # variables=st.session_state["data_dict"],
-            metadata=metadata
+            # metadata=metadata
         ))
 
     # Temporary placeholder AI message in chat history

rag.py

@@ -79,7 +79,7 @@ class Rag:
             },
         )
 
-    def ask(self, query: str, prompt_system: str, messages: list, variables: list = None, metadata: dict = {}):
+    def ask(self, query: str, prompt_system: str, messages: list, variables: list = None):
         self.chain = self.prompt | self.model | StrOutputParser()
         
         # Retrieve the context document
@@ -111,7 +111,7 @@ class Rag:
             chain_input.update(extra_vars)
         
 
-        return self.chain.stream(chain_input, { metadata : metadata })
+        return self.chain.stream(chain_input)
 
     def clear(self):
         self.document_vector_store = None