Last active
June 12, 2024 09:53
-
-
Save irem6142/f17ec5fad60fd9bbd0dcd673e6bd2abd to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import os | |
from dotenv import load_dotenv | |
import openai | |
def main(): | |
try: | |
# Get configuration settings | |
load_dotenv() | |
azure_oai_endpoint = os.getenv("AZURE_OAI_ENDPOINT") | |
azure_oai_key = os.getenv("AZURE_OAI_KEY") | |
azure_oai_deployment = os.getenv("AZURE_OAI_DEPLOYMENT") | |
# Initialize the Azure OpenAI client | |
client = openai.AzureOpenAI( | |
azure_endpoint=azure_oai_endpoint, | |
api_key=azure_oai_key, | |
api_version="2024-02-01" | |
) | |
# Create a system message | |
system_message = "I am a hiking enthusiast named Forest who helps people discover hikes in their area. If no area is specified, I will default to near Rainier National Park. I will then provide three suggestions for nearby hikes that vary in length. I will also share an interesting fact about the local nature on the hikes when making a recommendation." | |
while True: | |
# Get input text | |
input_text = input("Enter the prompt (or type 'quit' to exit): ") | |
if input_text.lower() == "quit": | |
break | |
if len(input_text) == 0: | |
print("Please enter a prompt.") | |
continue | |
print("\nSending request for summary to Azure OpenAI endpoint...\n\n") | |
# Send request to Azure OpenAI model | |
response = client.chat.completions.create( | |
model=azure_oai_deployment, | |
temperature=0.7, | |
max_tokens=400, | |
messages=[ | |
{"role": "system", "content": system_message}, | |
{"role": "user", "content": input_text} | |
] | |
) | |
generated_text = response.choices[0].message.content | |
print("Generated text:", generated_text) | |
except Exception as ex: | |
print(ex) | |
if __name__ == '__main__': | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment