Update question_generator.py

question_generator.py

@@ -5,7 +5,8 @@ import logging
 import hashlib
 from typing import List, Dict
 from datetime import datetime
-from mistralai import Mistral
+from mistralai.client import MistralClient
+from mistralai.models.chat_completion import ChatMessage
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -19,7 +20,7 @@ if not api_key:
 model = "mistral-large-latest"
 
 # Initialize Mistral client
-client = Mistral(api_key=api_key)
+client = MistralClient(api_key=api_key)
 
 def load_csv_data(file_path: str) -> List[Dict[str, str]]:
     """Load data from a CSV file."""
@@ -135,32 +136,26 @@ def generate_microbiology_question() -> Dict[str, str]:
 
     Format the response as a JSON object with the following keys:
     
-    {
+    {{
         "question": "The question text",
-        "options": {
+        "options": {{
             "A": "Option A text",
             "B": "Option B text",
             "C": "Option C text",
             "D": "Option D text",
             "E": "Option E text"
-        },
+        }},
         "correct_answer": "The letter of the correct answer (A, B, C, D, or E)",
         "explanation": "The explanation text",
         "medical_reasoning": "The detailed medical reasoning text"
-    }
+    }}
     """
     
-    chat_response = client.chat.complete(
+    chat_response = client.chat(
         model=model,
         messages=[
-            {
-                "role": "system",
-                "content": "You are a medical educator creating unique microbiology questions for the NBME exam. Ensure each question is distinct from previously generated ones and follows the specified template."
-            },
-            {
-                "role": "user",
-                "content": prompt
-            }
+            ChatMessage(role="system", content="You are a medical educator creating unique microbiology questions for the NBME exam. Ensure each question is distinct from previously generated ones and follows the specified template."),
+            ChatMessage(role="user", content=prompt)
         ]
     )