fix parameters

app.py

@@ -35,7 +35,7 @@ def add_text(history, text):
     return history, gr.Textbox(value="", interactive=False)
 
 
-def bot(history, api_kind, chunk_table, embedding_model, llm_model, cross_encoder, rerank_topk ):
+def bot(history, api_kind, chunk_table, embedding_model, llm_model, cross_encoder, top_k_param, rerank_topk ):
     top_k_param = int(top_k_param)
     query = history[-1][0]
 
@@ -62,12 +62,23 @@ def bot(history, api_kind, chunk_table, embedding_model, llm_model, cross_encode
     prompt = template.render(documents=documents, query=query)
     prompt_html = template_html.render(documents=documents, query=query)
 
-    if api_kind == "HuggingFace":
-         generate_fn = generate_hf
-    elif api_kind == "OpenAI":
-         generate_fn = generate_openai
-    else:
-         raise gr.Error(f"API {api_kind} is not supported")
+    if llm_model == "mistralai/Mistral-7B-Instruct-v0.2":
+        pass
+    if llm_model == "mistralai/Mistral-7B-v0.1":
+        pass
+    if llm_model == "mistralai/Mixtral-8x7B-Instruct-v0.1":
+        pass 
+    if llm_model == "gpt-3.5-turbo":
+        pass
+    if llm_model == "gpt-4-turbo-preview":
+        pass
+
+    #if api_kind == "HuggingFace":
+    #     generate_fn = generate_hf
+    #elif api_kind == "OpenAI":
+    #     generate_fn = generate_openai
+    #else:
+    #     raise gr.Error(f"API {api_kind} is not supported")
 
     history[-1][1] = ""
     for character in generate_fn(prompt, history[:-1]):
@@ -117,6 +128,7 @@ with gr.Blocks() as demo:
             )
     llm_model = gr.Radio(
                 choices=[
+                    "mistralai/Mistral-7B-Instruct-v0.2",
                     "gpt-3.5-turbo",
                     "gpt-4-turbo-preview",
                     "mistralai/Mistral-7B-v0.1",