Update main.py

main.py

@@ -3,7 +3,7 @@ import json
 import numpy as np
 from wiki import search as search_wikipedia
 from concurrent.futures import ThreadPoolExecutor
-from llm_handler import send_to_llm
+from llm_handler import send_to_llm, agent, settings
 from params import OUTPUT_FILE_PATH, NUM_WORKERS
 from system_messages import SYSTEM_MESSAGES_VODALUS
 from topics import TOPICS
@@ -50,7 +50,7 @@ async def generate_data(
     msg_list = [msg_context, {"role": "user", "content": f"Generate a question based on the SUBJECT_AREA: {topic_selected}"}]
 
     # Send to LLM for question generation
-    question, _ = send_to_llm("llamanet", msg_list)
+    question, _ = send_to_llm(agent, msg_list)
 
     # Prepare message list for LLM to generate the answer
     msg_list_answer = [
@@ -59,7 +59,7 @@ async def generate_data(
     ]
 
     # Send to LLM for answer generation
-    answer, _ = send_to_llm("llamanet", msg_list_answer)
+    answer, _ = send_to_llm(agent, msg_list_answer)
 
     # Prepare data for output (excluding usage information)
     data = {