mirror of
https://github.com/marcel-dempers/docker-development-youtube-series.git
synced 2025-06-04 16:56:56 +00:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import sys
|
|
import os
|
|
import json
|
|
import openai
|
|
|
|
openai.api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
#read the incoming message
|
|
message = sys.argv[1]
|
|
user_message = {
|
|
"role" : "user",
|
|
"content" : message
|
|
}
|
|
|
|
systemMessage = {
|
|
"role": "system",
|
|
"content": "You are a kubernetes exper that can assist developers with troubleshooting deployments\n\nTo help the developer you will need to know the namespaces as well as the pod name. Ask for missing information\n\nGenerate a command to help the developer surface logs or information\n"
|
|
}
|
|
|
|
# read the cached user messages if there are any
|
|
userMessages = []
|
|
if os.path.isfile("messages.json"):
|
|
with open('messages.json', newline='') as messagesFile:
|
|
data = messagesFile.read()
|
|
userMessages = json.loads(data)
|
|
|
|
# add the new message to it and update the cached messages
|
|
userMessages.append(user_message)
|
|
with open('messages.json', 'w', newline='') as messagesFile:
|
|
msgJSON = json.dumps(userMessages)
|
|
messagesFile.write(msgJSON)
|
|
print(msgJSON)
|
|
|
|
messages = []
|
|
messages.append(systemMessage)
|
|
messages.extend(userMessages)
|
|
|
|
response = openai.ChatCompletion.create(
|
|
model="gpt-3.5-turbo",
|
|
messages=messages,
|
|
temperature=1,
|
|
max_tokens=256,
|
|
top_p=1,
|
|
frequency_penalty=0,
|
|
presence_penalty=0
|
|
)
|
|
|
|
responseMessage = response.choices[0].message.content
|
|
print(responseMessage) |