llm.py

import openai
from os import environ

from prompt import get_prompt

openai.api_key = environ.get("OPENAI_API_KEY")


def get_response(city, preferences):
    response = openai.Completion.create(
        model="text-davinci-003",
        prompt=get_prompt(city, preferences),
        temperature=0.6,
        max_tokens=200
    )
    return response["choices"][0]["text"]