sql.py

import streamlit as st

from langchain_community.utilities import SQLDatabase
from langchain_openai import ChatOpenAI
from langchain.chains import create_sql_query_chain

# Set up Langchain SQL access
db = SQLDatabase.from_uri("duckdb:///tmp.db", view_support=True)
parquet = "s3://us-west-2.opendata.source.coop/cboettig/gbif/2024-10-01/**"
db.run(f"create or replace view gbif_h3 as select * from read_parquet('{parquet}');")
llm = ChatOpenAI(model="llama3", temperature=0, api_key=st.secrets["LITELLM_KEY"], base_url = "https://llm.nrp-nautilus.io")
chain = create_sql_query_chain(llm, db)