VcRlAgent commited on
Commit
795d61e
·
1 Parent(s): 9e24dfc

Starter LLM Inference Call

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -38,6 +38,10 @@ df.to_sql("tickets", engine, if_exists="replace", index=False)
38
  # Create agent with table description
39
  db = SQLDatabase(engine)
40
 
 
 
 
 
41
  # Add table descriptions for better context
42
  table_info = """
43
  The 'tickets' table contains Jira ticket data with columns:
 
38
  # Create agent with table description
39
  db = SQLDatabase(engine)
40
 
41
+ print("Tables:", db.get_table_names())
42
+ print("\nSchema:", db.get_table_info())
43
+ print("\nSample Data:", db.run("SELECT * FROM tickets LIMIT 5"))
44
+
45
  # Add table descriptions for better context
46
  table_info = """
47
  The 'tickets' table contains Jira ticket data with columns: