Update main.py

main.py

@@ -5,6 +5,7 @@ from wiki import search as search_wikipedia  # Import the search function from t
 from concurrent.futures import ThreadPoolExecutor  # Import ThreadPoolExecutor for concurrent execution
 from llm_handler import send_to_llm  # Import the send_to_llm function from the llm_handler module
 from params import OUTPUT_FILE_PATH, NUM_WORKERS  # Import constants from the params module
+import llamanet
 
 # Import system messages from the system_messages module
 from system_messages import (
@@ -12,6 +13,9 @@ from system_messages import (
 )
 from topics import TOPICS  # Import topics from the topics module
 
+# Initialize LlamaNet
+llamanet.run()
+
 # Set the system messages to those specified in SYSTEM_MESSAGES_VODALUS
 SYSTEM_MESSAGES = SYSTEM_MESSAGES_VODALUS
 
@@ -54,7 +58,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(msg_list)
+    question, _ = send_to_llm("llamanet", msg_list)
 
     # Prepare message list for LLM to generate the answer
     msg_list_answer = [
@@ -63,7 +67,7 @@ async def generate_data(
     ]
 
     # Send to LLM for answer generation
-    answer, _ = send_to_llm(msg_list_answer)
+    answer, _ = send_to_llm("llamanet", msg_list_answer)
 
     # Prepare data for output (excluding usage information)
     data = {