Added Langsmith

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import matplotlib.pyplot as plt
 from transformers import HfEngine, ReactCodeAgent
 from transformers.agents import Tool
-
+from langsmith import traceable
 # Height of the Tabs Text Area
 TAB_LINES = 8
 # Load Token
@@ -146,6 +146,11 @@ def get_visualization(question, tool, schema, table_name):
     return results
 
 
+@traceable()
+def query_response(input_prompt, generated_sql):
+  return generated_sql
+
+
 class SQLExecutorTool(Tool):
     name = "sql_engine"
     inputs = {
@@ -176,6 +181,8 @@ def main(table, text_query):
         fig = output.get('fig', None)
         generated_sql = output.get('sql', None)
         data = output.get('data', None)
+        input_prompt = text_query + '\n' + table_name + '\n' + schema
+        _ = query_response(input_prompt, generated_sql)
     except Exception as e:
         gr.Warning(f"❌ Unable to generate the visualization. {e}")
         

requirements.txt

@@ -1,7 +1,8 @@
 torch
-huggingface_hub
-accelerate
-transformers==4.44.2
-duckdb
 seaborn
 plotly
+huggingface_hub
+accelerate==0.34.2
+transformers==4.44.2
+duckdb==1.1.1
+langsmith==0.1.135