fixes

thammuio · Apr 29, 2024 · 5f86f59 · 5f86f59
1 parent 77174ba
commit 5f86f59
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app/api/routes.py b/app/api/routes.py
@@ -23,7 +23,7 @@ def check_api_status() -> dict[str, str]:
 # Chat
 # Map the model names to the functions
 model_methods = {
-    "llama-2-7B-chat": llama_2_7b_chat
+    "llama2-7B-chat": llama_2_7b_chat
     # "llama-2-13B-chat": llama_2_13b_chat,
     # "Meta-Llama-3-8B-Instruct": llama_3_8b_instruct,
     # "zephyr-7B-alpha": zephyr_7B_alpha,

diff --git a/app/chatbot/mistral_7B_instruct/controller.py b/app/chatbot/mistral_7B_instruct/controller.py
@@ -54,7 +54,7 @@ def mistral_7b_instruct(prompt, temperature, max_tokens, selected_vector_db, use
         context_chunk = get_nearest_chunk_from_pinecone_vectordb(index, vdb_question)
 
     # Step 2: Call the relavent Model in Model Serving
-    MODEL_ACCESS_KEY = get_model_access_key({"name": "llama-2-7B-chat"})
+    MODEL_ACCESS_KEY = get_model_access_key({"name": "Mistral-7B-Instruct"})
 
     question = {'prompt': prompt, "temperature": temperature, "max_tokens": max_tokens, "context": context_chunk, "user": user}
     data = json.dumps({'accessKey': MODEL_ACCESS_KEY, 'request': question})