test code smith

app.py

@@ -1,6 +1,7 @@
 from AssistantService import GPTAssistant
 from openai.error import AuthenticationError
 import streamlit as st
+from langsmith.run_helpers import traceable
 import configparser
 import os
 
@@ -66,7 +67,19 @@ if 'output_format' in st.session_state:
             st.write("Complete the API key field")
         except AuthenticationError:
             st.write("Invalid API key")
-
+            
+@traceable(run_type="tool")
+def test_the_code(code, full_content):
+    exec(code, globals())
+    if result:
+        st.write("data extracted successfully")
+        # show data in table
+        st.table(result)
+    else:
+        st.write("error extracting data")
+        
+    return result or "error"
+    
 
 if 'code_generated' in st.session_state:
     python_function_label = st.write("Here is your python function:")
@@ -76,11 +89,5 @@ if 'code_generated' in st.session_state:
     if full_content and test_code:
         html_data = full_content
         result = None
-        exec(st.session_state['code_generated_exec'], globals())
-        if result:
-            st.write("data extracted successfully")
-            # show data in table
-            st.table(result)
-        else:
-            st.write("error extracting data")
+        test_the_code(st.session_state['code_generated_exec'], full_content=full_content)