Update main.py

main.py

@@ -4,10 +4,7 @@ import numpy as np  # Provides support for large, multi-dimensional arrays and m
 from wiki import search as search_wikipedia  # Import the search function from the wiki module and rename it
 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, PROVIDER  # Import constants from the params module
-
-# Set the provider for the language model to "local-model"
-PROVIDER = "local-model"
+from params import OUTPUT_FILE_PATH, NUM_WORKERS  # Import constants from the params module
 
 # Import system messages from the system_messages module
 from system_messages import (
@@ -39,8 +36,7 @@ async def generate_data(
     topic_selected,
     system_message_generation,
     system_message_selected,
-    output_file_path,
-    llm_provider
+    output_file_path
 ):
     # Fetch Wikipedia content for the selected topic
     wikipedia_info = search_wikipedia(topic_selected)
@@ -58,7 +54,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(llm_provider, msg_list)
+    question, _ = send_to_llm(msg_list)
 
     # Prepare message list for LLM to generate the answer
     msg_list_answer = [
@@ -67,7 +63,7 @@ async def generate_data(
     ]
 
     # Send to LLM for answer generation
-    answer, _ = send_to_llm(llm_provider, msg_list_answer)
+    answer, _ = send_to_llm(msg_list_answer)
 
     # Prepare data for output (excluding usage information)
     data = {
@@ -101,8 +97,7 @@ def main():
                     topic_selected,
                     system_message_generation,
                     system_message_selected,
-                    OUTPUT_FILE_PATH,
-                    PROVIDER
+                    OUTPUT_FILE_PATH
                 )
             )