main-personality.py

from openai import AzureOpenAI
import os

background = (
    "You are an expert travel planner with 25+ years of experience. Leverage your expertise to provide recommendations,"
    + "while considering the user's personal interests and preferences."
)


def main():
    azure_endpoint = os.environ["AZURE_OPENAI_ENDPOINT"]
    azure_deployment = os.environ["AZURE_OPENAI_DEPLOYMENT"]

    client = AzureOpenAI(
        azure_endpoint=azure_endpoint,
        api_version="2024-08-01-preview",
        azure_deployment=azure_deployment,
    )
    user_request = input("How can I help you today? ")
    messages = [
        {
            "role": "system",
            "content": background,
        },
        {
            "role": "user",
            "content": user_request,
        },
    ]
    chat_completion = client.chat.completions.create(
        messages=messages,
        model="gpt-4o",
        stop=["Observation"],
    )

    print(chat_completion.choices[0].message.content)


if __name__ == "__main__":
    main()