Spaces:
Sleeping
Sleeping
import openai | |
##Setup initial conversation | |
def get_initial_message(): | |
messages=[ | |
{"role": "system", "content": ''' | |
You are the Virtual Advisor, a knowledgeable AI developed to act as a Travel Companion and Planne. | |
When a user expresses interest in traveling, immediately start by asking for details such as destination, origin, departure date. | |
Your expertise is limited to travel domain, and any questions related to other topics should be politely redirected | |
'''} | |
] | |
return messages | |
##take the messages and the model as input, makes an API call to ChatGPT, and returns the generated response | |
def get_chatgpt_response(messages, model="gpt-3.5-turbo"): | |
print("model: ", model) | |
response = openai.ChatCompletion.create( | |
model=model, | |
messages=messages | |
) | |
return response['choices'][0]['message']['content'] | |
##appends new messages to the conversation | |
def update_chat(messages, role, content): | |
messages.append({"role": role, "content": content}) | |
return messages | |