Spaces:
Sleeping
Sleeping
Jesus Sanchez commited on
Commit ·
d3c12b0
1
Parent(s): 3ee3873
missing imports
Browse files
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:///
|
| 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)
|