more models

minimal-example.py

@@ -26,9 +26,12 @@ chatgpt4_llm = ChatOpenAI(model="gpt-4", temperature=0, api_key=st.secrets["OPEN
 # Requires ollama server running locally
 from langchain_community.llms import Ollama
 ## # from langchain_community.llms import ChatOllama
-ollama_llm = Ollama(model="duckdb-nsql", temperature=0)
 
-models = {"ollama": ollama_llm, "chatgpt3.5": chatgpt_llm, "chatgpt4": chatgpt4_llm}
+models = {"duckdb-nsql": Ollama(model="duckdb-nsql", temperature=0),
+          "sqlcoder": Ollama(model="sqlcoder", temperature=0),
+          "gemma": Ollama(model="gemma", temperature=0),
+          "chatgpt3.5": chatgpt_llm, 
+          "chatgpt4": chatgpt4_llm}
 with st.sidebar:
     choice = st.radio("Select an LLM:", models)
     llm = models[choice]