Update app.py

app.py

@@ -201,7 +201,8 @@ def get_file_path(file):
     else:
         return None
 
-# Original functionalities (Tabs 1-4) functions 
+# Original functionalities (Tabs 1-4) functions
+@traceable(name="Biden LLaMA QA")
 def rag_llama_qa(query):
     output = RetrievalQA.from_chain_type(
         llm=llm_local,
@@ -214,9 +215,12 @@ def rag_llama_qa(query):
     idx = lower_text.find("answer:")
     return output[idx + len("answer:"):].strip() if idx != -1 else output
 
+@traceable(name="GPT-4 Document QA")
 def rag_gpt4_qa(query):
     return qa_gpt.run(query)
 
+
+@traceable(name="Upload Document QA")
 def upload_and_chat(file, query):
     file_path = get_file_path(file)
     if file_path is None:
@@ -262,6 +266,7 @@ Refine the summary based on the new content below. Add or update information onl
 """
 )
 
+@traceable(name="Document Summarise")
 def document_summarize(file):
     file_path = get_file_path(file)
     if file_path is None:
@@ -990,6 +995,7 @@ crew = Crew(
     llm=crew_llm
 )
 
+@traceable(name="Multi-Agent Chat")
 def multi_agent_chat_advanced(query: str, file=None) -> str:
     global session_retriever, session_qa_chain, csv_dataframe