activate logging to w&b

app.py

@@ -8,6 +8,7 @@ import pandas as pd
 import streamlit as st
 from IPython.core.display import HTML
 from PIL import Image
+from langchain.callbacks import wandb_tracing_enabled
 from chemcrow.agents import ChemCrow, make_tools
 from chemcrow.frontend.streamlit_callback_handler import \
     StreamlitCallbackHandlerChem
@@ -18,6 +19,7 @@ from dotenv import load_dotenv
 load_dotenv()
 ss = st.session_state
 
+os.environ["WANDB_PROJECT"] = "cc-langchain-testing"
 
 icon = Image.open('assets/logo0.png')
 st.set_page_config(
@@ -95,8 +97,9 @@ if prompt := st.chat_input():
             output_placeholder=st.session_state
         )
         try:
-            response = agent.run(prompt, callbacks=[st_callback])
-            st.write(response)
+            with wandb_tracing_enabled():
+                response = agent.run(prompt, callbacks=[st_callback])
+                st.write(response)
         except:
             st.write("Please input a valid OpenAI API key.")
 

requirements.txt

@@ -4,4 +4,5 @@ streamlit
 python-dotenv
 duckduckgo-search
 wikipedia
+wandb
 chemcrow==0.3.7