keyword.py

import openai

def keyword_agent(query, key):

    system_prompt = """
    You are an AI assistant that looks at a given query about a school class, and comes up with a list of 3 keywords related to the query.
        Two keywords will be used that are similar in semantics to the given query but not already used in query.
        The third keyword should focus on an aspect related to the query that is less obvious, but might be helpful when finding similar data entries.
    If the query does not seem to relate to a school class, please output nothing or just spaces " ".
    """

    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": system_prompt},
            {"role": "user", "content": query}
        ]
    )

    return response["choices"][0]["message"]["content"]