fixed tracing with document agent

langchain-streamlit-demo/app.py

@@ -499,7 +499,7 @@ if st.session_state.llm:
  doc_chain_tool = Tool.from_function(
  func=lambda s: st.session_state.doc_chain.invoke(
  s,
- # config=get_config(callbacks),
+ config=get_config(callbacks),
  ),
  name="user-document-chat",
  description="this assistant returns a response based on the user's custom context. ",
@@ -510,10 +510,12 @@ if st.session_state.llm:
  doc_question_tool = Tool.from_function(
  func=lambda s: doc_chain_agent.invoke(
  s,
+ config=get_config(callbacks),
  ),
  name="document-question-tool",
  description="this assistant answers a question based on the user's custom context. "
  "this assistant responds to fully formed questions."
+ "Do not send anything besides a question. It already has context."
  "if the user's meaning is unclear, perhaps the answer is here. "
  "generally speaking, try this tool before conducting web research.",
  )