diff --git a/viz/backend/sqlchain.py b/viz/backend/sqlchain.py index e3550d9a..2e5c4aa0 100644 --- a/viz/backend/sqlchain.py +++ b/viz/backend/sqlchain.py @@ -9,7 +9,7 @@ def sqlchain(input_text): llm = ChatOpenAI(openai_api_key=config.openai_key, - model_name="gpt-3.5-turbo", temperature=0, verbose=True) + model_name="gpt-3.5-turbo", temperature=0, verbose=True) db = SQLDatabase.from_uri( f"postgresql+psycopg2://postgres:{config.sql_key}@localhost:5432/{config.database_name}") mydb = sqldb.create_engine(