Update app.py

app.py

@@ -21,11 +21,6 @@ def reset_conversation():
     st.session_state.messages = []
     return None
 
-# Define the available models
-model_links = {
-    "Mistral-7B": "mistralai/Mistral-7B-Instruct-v0.2"
-}
-
 # Set the temperature value directly in the code
 temperature = 0.5
 
@@ -62,9 +57,8 @@ if prompt := st.chat_input("Type your message here..."):
             response = client.chat.completions.create(
                 model=model_link,
                 messages=[
-                    system_message,
-                    {"role": m["role"], "content": m["content"]}
-                    for m in st.session_state.messages
+                    system_message,  # System message
+                    *[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages]  # User and assistant messages
                 ],
                 temperature=temperature,
                 max_tokens=3000,