Update pages/sql.py

pages/sql.py

@@ -51,7 +51,7 @@ Question: {input}
 '''
 
 with st.sidebar:
-    model = st.selectbox("LLM:", ["gorilla", "llama3", "olmo"])
+    model = st.selectbox("LLM:", ["gorilla", "llama3", "olmo", "groq-tools"])
 
 
 from langchain_core.prompts import PromptTemplate
@@ -59,7 +59,7 @@ prompt = PromptTemplate.from_template(template, partial_variables = {"dialect":
 
 # Now we are ready to create our model and start querying!
 from langchain_openai import ChatOpenAI
-llm = ChatOpenAI(model="gorilla", # Try: llama3, gorilla, or groq-tools, or other models
+llm = ChatOpenAI(model=model, # Try: llama3, gorilla, or groq-tools, or other models
                  temperature=0, 
                  api_key=api_key, 
                  base_url = "https://llm.nrp-nautilus.io")