Update llm_handler.py

llm_handler.py

@@ -1,17 +1,16 @@
 from openai import OpenAI
-from params import OPENAI_MODEL, OPENAI_API_KEY
 import llamanet
 
 # Initialize LlamaNet
-llamanet.run("start", "https://huggingface.co/arcee-ai/Arcee-Spark-GGUF/blob/main/Arcee-Spark-IQ4_XS.gguf")
+llamanet.run()
 
-# Create an instance of the OpenAI class
-client = OpenAI(api_key=OPENAI_API_KEY)
+# Create an instance of the OpenAI class (used by LlamaNet)
+client = OpenAI()
 
-def send_to_chatgpt(msg_list):
+def send_to_llamanet(msg_list):
     try:
         completion = client.chat.completions.create(
-            model=OPENAI_MODEL,  # LlamaNet ignores this
+            model="https://huggingface.co/arcee-ai/Arcee-Spark-GGUF/blob/main/Arcee-Spark-IQ4_XS.gguf",
             messages=msg_list,
             temperature=0.6,
             stream=True
@@ -25,11 +24,11 @@ def send_to_chatgpt(msg_list):
         return response, None  # LlamaNet doesn't provide usage info
     
     except Exception as e:
-        print(f"Error in send_to_chatgpt: {str(e)}")
+        print(f"Error in send_to_llamanet: {str(e)}")
         return f"Error: {str(e)}", None
 
 def send_to_llm(provider, msg_list):
     if provider == "llamanet":
-        return send_to_chatgpt(msg_list)
+        return send_to_llamanet(msg_list)
     else:
         raise ValueError(f"Unknown provider: {provider}")