Added get_completion

main.py

@@ -17,6 +17,15 @@ client = OpenAI(
 
 sp = "You are a helpful and concise assistant."
 
+def get_completion(prompt, model="solar-1-mini-chat"):
+    messages = [{"role": "user", "content": prompt}]
+    response = client.chat.completions.create(
+        model=model,
+        messages=messages,
+        temperature=0, # this is the degree of randomness of the model's output
+    )
+    return response.choices[0].message["content"]
+
 def get_completion_from_messages(messages, model="solar-1-mini-chat", temperature=0):
     response = client.chat.completions.create(
         model=model,
@@ -57,7 +66,7 @@ def send(msg:str, messages:list[str]=None):
     if not messages: messages = []
     messages.append(msg.rstrip())
     print(messages)
-    r = get_completion_from_messages(messages) # get response from chat model
+    r = get_completion(messages) # get response from chat model
     return (ChatMessage(msg, True),    # The user's message
             ChatMessage(r.rstrip(), False), # The chatbot's response
             ChatInput()) # And clear the input field via an OOB swap