Update backend/query_llm.py

backend/query_llm.py

@@ -52,13 +52,16 @@ def format_prompt(message: str, api_kind: str):
     """
 
     # Create a list of message dictionaries with role and content
-    messages: List[Dict[str, Any]] = [{'role': 'user', 'content': message}]
+    messages1: List[Dict[str, Any]] = [{'role': 'user', 'content': message}]
+    messages2: List[Dict[str, Any]] = [{'role': 'user', 'content': message}]
 
     if api_kind == "openai":
-        return messages
+        return messages1
     elif api_kind == "hf":
         return tokenizer.apply_chat_template(messages, tokenize=False)
-    elif api_kind:
+    elif api_kind=="gemini":
+        return messages2
+    else:
         raise ValueError("API is not supported")