Upload app.py

app.py

@@ -116,13 +116,10 @@ def llm_query(user_input,company):
         )
     ]
 
-    # Flatten the list of messages
-    prompt = [message for sublist in prompt for message in (sublist if isinstance(sublist, list) else [sublist])]
-
     # Get response from the LLM
     try:
         # Call the chat model with the message
-        response = llm([prompt])
+        response = llm(prompt)
         
         # response = llm_client.chat.completions.create(
         #     model=model_name,