refactor: adjust structure

This commit is contained in:
Florian Hönicke
2023-03-18 20:05:52 +01:00
parent d84cc8ee04
commit 47aa456e88
10 changed files with 221 additions and 131 deletions

27
src/gpt.py Normal file
View File

@@ -0,0 +1,27 @@
import os
import openai
openai.api_key = os.environ['OPENAI_API_KEY']
def get_response(system_definition, user_query):
response = openai.ChatCompletion.create(
temperature=0,
model="gpt-4",
messages=[
{
"role": "system",
"content": system_definition
},
{
"role": "user",
"content":
user_query
},
]
)
content = response['choices'][0]['message']['content']
print(content)
return content