Update main.py

main.py

@@ -23,13 +23,13 @@ async def solve_dsa_problem(item: DSAPrompt):
                     If the input is not a DSA problem, politely refuse their request and reinsist to provide a DSA problem.
                     '''
     
-    res = llm.create_chat_completion(
-        messages=[
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": item.prompt}
-        ],
-        temperature = 0.7,
-        max_tokens = 2048,
-    )
+    # res = llm.create_chat_completion(
+    #     messages=[
+    #         {"role": "system", "content": system_prompt},
+    #         {"role": "user", "content": item.prompt}
+    #     ],
+    #     temperature = 0.7,
+    #     max_tokens = 2048,
+    # )
 
-    return res
+    return llm(item.prompt, temperature = 0.9, max_tokens = 2048)