mirror of https://github.com/ashikslab/smartsh.git
21 lines
665 B
Python
21 lines
665 B
Python
|
import openai
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
#Get the OpenAI API key from environment variable
|
||
|
|
openai.api_key = os.environ.get("OPENAI_API_KEY")
|
||
|
|
# argcmd contains the entire command line arguments as a space separated string
|
||
|
|
argcmd = " ".join(sys.argv)
|
||
|
|
|
||
|
|
prompt = "Suggest a linux shell command to accomplish the following. If the entered string is already a valid command, suggest installing the required packages: " + argcmd
|
||
|
|
completion = openai.ChatCompletion.create(
|
||
|
|
model = 'gpt-3.5-turbo',
|
||
|
|
messages = [
|
||
|
|
{'role': 'user', 'content': prompt}
|
||
|
|
],
|
||
|
|
temperature = 0
|
||
|
|
)
|
||
|
|
|
||
|
|
# print the response to stdout
|
||
|
|
print(completion['choices'][0]['message']['content'])
|