Function get_assistant_message

byod/api/app/utils.py:48–62

(
        messages: List[Dict[str, str]],
        temperature: int = 0,
        model: str = "gpt-3.5-0301",
        # model: str = "gpt-3.5-turbo",
        # model: str = "gpt-4",
)

def get_assistant_message(
        messages: List[Dict[str, str]],
        temperature: int = 0,
        model: str = "gpt-3.5-0301",
        # model: str = "gpt-3.5-turbo",
        # model: str = "gpt-4",
):
    res = openai.ChatCompletion.create(
        model=model,
        temperature=temperature,
        messages=messages
    )
    # completion = res['choices'][0]["message"]["content"]
    assistant_message = res['choices'][0]
    return assistant_message


def clean_message_content(assistant_message_content):