chatbot using ollama and openAI

LLMs and AI
1

i am trying to create a chatbot that allows you to toggle between using ollama or chatgpt, but i cannot get the latter to work. does anyone know what the issue is?

      

if "selected_model" not in st.session_state:
    st.session_state.selected_model = "ollama"

if "messages" not in st.session_state:
    st.session_state.messages = []

# chat display prev msgs
for msg in st.session_state.messages:
    with st.chat_message(msg["role"]):
        st.markdown(msg["content"])

#prompt part
if prompt := st.chat_input("type here..."):
    with st.chat_message("user"):
        st.markdown(prompt)

    if st.session_state.selected_model == "ollama":
        response = get_chatbot_response(prompt, st.session_state.messages)
    
    else:
        st.session_state.selected_model = "gpt-3.5-turbo"
        with st.chat_message("assistant"):
            stream = client.chat.completions.create(
                model=st.session_state.selected_model,
                messages=[
                    {"role": m["role"], "content": m["content"]}
                    for m in st.session_state.messages
                ],
                stream=True,
            )
            response = st.write_stream(stream)
    st.session_state.messages.append({"role": "assistant", "content": response})

    with st.chat_message("assistant"):
        st.markdown(response)