fixing message content
Browse files
main.py
CHANGED
@@ -24,7 +24,7 @@ def get_completion(prompt, model="solar-1-mini-chat"):
|
|
24 |
messages=messages,
|
25 |
temperature=0, # this is the degree of randomness of the model's output
|
26 |
)
|
27 |
-
return response.choices[0].message
|
28 |
|
29 |
def get_completion_from_messages(messages, model="solar-1-mini-chat", temperature=0):
|
30 |
response = client.chat.completions.create(
|
@@ -32,6 +32,7 @@ def get_completion_from_messages(messages, model="solar-1-mini-chat", temperatur
|
|
32 |
messages=messages,
|
33 |
temperature=temperature, # this is the degree of randomness of the model's output
|
34 |
)
|
|
|
35 |
|
36 |
# Chat message component (renders a chat bubble)
|
37 |
def ChatMessage(msg, user):
|
|
|
24 |
messages=messages,
|
25 |
temperature=0, # this is the degree of randomness of the model's output
|
26 |
)
|
27 |
+
return response.choices[0].message.content
|
28 |
|
29 |
def get_completion_from_messages(messages, model="solar-1-mini-chat", temperature=0):
|
30 |
response = client.chat.completions.create(
|
|
|
32 |
messages=messages,
|
33 |
temperature=temperature, # this is the degree of randomness of the model's output
|
34 |
)
|
35 |
+
return response.choices[0].message.content
|
36 |
|
37 |
# Chat message component (renders a chat bubble)
|
38 |
def ChatMessage(msg, user):
|