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