diff --git a/searx/engines/chatgpt.py b/searx/engines/chatgpt.py new file mode 100644 index 000000000..af80800be --- /dev/null +++ b/searx/engines/chatgpt.py @@ -0,0 +1,28 @@ +import os +from flask import Flask, jsonify, request +import openai + +app = Flask(__name__) + +# retrieve ChatGPT API key from system variable +API_KEY = os.environ.get("CHATGPT_API_KEY") + +# initialize OpenAI API client +openai.api_key = API_KEY +engine_id = "text-davinci-002" + +@app.route('/chatgpt', methods=['GET']) +def chatgpt(): + query = request.args.get('query') + response = openai.Completion.create( + engine=engine_id, + prompt=query, + max_tokens=150, + n=1, + stop="\n" + ) + chatgpt_response = response.choices[0].text + return chatgpt_response + +if __name__ == '__main__': + app.run()