shellgpt.py

import requests
import itertools
import json
import sys
import distro


def main():

    user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36"

    headers = {
        'Content-Type': 'application/json',
        'user-agent': f'{user_agent}',
    }

    arg = ' '.join(sys.argv[1:])

    # json_data = {
    #     "model": "gpt-4o-mini",
    #     "stream": "false",
    #     'messages': [
    #         {
    #             'role': 'user',
    #             'content': f'Answer with only the actual command without any intro or explanation. What is the {distro.id()} linux command line command to {arg}',
    #         },
    #     ]
    # }

    json_data = {
  "model": "gpt-4o-mini",
  "stream": False,
  "messages": [
    {
      "role": "system",
      "content": "You are Marvis, a helpful assistant."
    },
    {
      "role": "user",
      "content": f"Answer with only the actual command without any intro or explanation. What is the {distro.id()} linux command line command to {arg}"
    }
  ]
}

    response = requests.post('https://gpt-4o-mini.deno.dev/v1/chat/completions', headers=headers, json=json_data)
    print(json.loads(response.text)['choices'][0]['message']['content'])

if __name__ == '__main__':
    main()