Jesus Sanchez commited on
Commit
d3c12b0
·
1 Parent(s): 3ee3873

missing imports

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -4,6 +4,10 @@ import pandas as pd
4
  import altair as alt
5
  import chat as idf_chat
6
  import sqlite3
 
 
 
 
7
 
8
 
9
  def tables_from_db():
@@ -29,7 +33,7 @@ def from_db(table: str):
29
 
30
  def from_gpt(query: str):
31
  # Create the agent executor
32
- db = SQLDatabase.from_uri("sqlite:///archive/switrs.sqlite")
33
 
34
  llm=OpenAI(temperature=0)
35
  toolkit = SQLDatabaseToolkit(llm=llm,db=db)
 
4
  import altair as alt
5
  import chat as idf_chat
6
  import sqlite3
7
+ from langchain.sql_database import SQLDatabase
8
+ from langchain.agents.agent_toolkits import SQLDatabaseToolkit
9
+ from langchain.agents import create_sql_agent
10
+ from langchain import OpenAI
11
 
12
 
13
  def tables_from_db():
 
33
 
34
  def from_gpt(query: str):
35
  # Create the agent executor
36
+ db = SQLDatabase.from_uri("sqlite:///switrs.sqlite")
37
 
38
  llm=OpenAI(temperature=0)
39
  toolkit = SQLDatabaseToolkit(llm=llm,db=db)