Update backend/query_llm.py

backend/query_llm.py

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