mirror of https://github.com/ashikslab/smartsh.git
Initial commit
This commit is contained in:
commit
2824ab6089
|
@ -0,0 +1,11 @@
|
|||
You need to add the following to your .bashrc
|
||||
|
||||
```
|
||||
export OPENAI_API_KEY="your_api_key"
|
||||
|
||||
export SMARTSH_PATH="path_to_dir_with_smartsh.py"
|
||||
command_not_found_handle () {
|
||||
echo "Let's get help from OpenAI API!"
|
||||
python3 $SMARTSH_PATH/smartsh.py "$@"
|
||||
}
|
||||
```
|
|
@ -0,0 +1,20 @@
|
|||
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'])
|
Loading…
Reference in New Issue