Update main.py

main.py

@@ -47,12 +47,16 @@ async def root():
 @app.post("/generate_quiz/")
 async def generate_quiz(request: QuizRequest):
     # fetch & store context
-    context_docs = wikipedia_query(request.search_query)
+
+    llm = get_llm()
+    search_query=llm.invoke(f'write one word which should be search for this question {request.search_query}')
+    
+    context_docs = wikipedia_query(search_query)
     context_text = str(context_docs)
     STORE["context"] = context_text
 
     # generate quiz
-    llm = get_llm()
+    
     prompt = f"""
 You are a quiz generator assistant. Create a quiz for the given context.