/
openai.py
76 lines (62 loc) · 2.02 KB
/
openai.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import openai
import openai_secret_manager
# Set up your OpenAI API key
assert "openai" in openai_secret_manager.get_services()
secrets = openai_secret_manager.get_secret("openai")
openai.api_key = secrets["api_key"]
model_engine = "davinci"
# Define a function to generate a response using GPT-3
def generate_response(prompt):
response = openai.Completion.create(
engine=model_engine,
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.5,
)
message = response.choices[0].text.strip()
return message
# Define a function to handle user input and generate a response
def chatbot():
print("Hello, I'm a chatbot. How can I help you?")
while True:
user_input = input("> ")
if user_input.lower() in ["bye", "goodbye"]:
print("Goodbye!")
break
prompt = f"User: {user_input}\nChatbot:"
response = generate_response(prompt)
print(response)
# Call the chatbot function to start the conversation
chatbot()
# import openai
# import json
# import os
# # Set up your OpenAI API key
# openai.api_key = os.getenv("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")
# # Define a function to generate a response using GPT-3
# def generate_response(prompt):
# response = openai.api.Completion.create(
# engine="text-davinci-003",
# prompt=prompt,
# max_tokens=1024,
# n=1,
# stop=None,
# temperature=0.5,
# )
# message = response.choices[0].text.strip()
# return message
# # Define a function to handle user input and generate a response
# def chatbot():
# print("Hello, I'm a chatbot. How can I help you?")
# while True:
# user_input = input("> ")
# if user_input.lower() in ["bye", "goodbye"]:
# print("Goodbye!")
# break
# prompt = f"User: {user_input}\nChatbot:"
# response = generate_response(prompt)
# print(response)
# # Call the chatbot function to start the conversation
# chatbot()