Add console script

This commit is contained in:
Arran Hobson Sayers 2023-08-31 12:32:23 +01:00
parent 51c5197c5c
commit eee553bd46
3 changed files with 21 additions and 7 deletions

View File

@ -12,6 +12,7 @@ from g4f import ChatCompletion
app = Flask(__name__) app = Flask(__name__)
CORS(app) CORS(app)
@app.route("/chat/completions", methods=["POST"]) @app.route("/chat/completions", methods=["POST"])
def chat_completions(): def chat_completions():
model = request.get_json().get("model", "gpt-3.5-turbo") model = request.get_json().get("model", "gpt-3.5-turbo")
@ -87,5 +88,9 @@ def chat_completions():
return app.response_class(streaming(), mimetype="text/event-stream") return app.response_class(streaming(), mimetype="text/event-stream")
def main():
app.run(host="0.0.0.0", port=1337, debug=True)
if __name__ == "__main__": if __name__ == "__main__":
app.run(host="0.0.0.0", port=1337, debug=True) main()

View File

@ -8,4 +8,5 @@ websockets
js2py js2py
quickjs quickjs
flask flask
flask-cors flask-cors
httpx

View File

@ -11,7 +11,10 @@ with codecs.open(os.path.join(here, "README.md"), encoding="utf-8") as fh:
with open("requirements.txt") as f: with open("requirements.txt") as f:
required = f.read().splitlines() required = f.read().splitlines()
VERSION = '0.0.2.6' with open("interference/requirements.txt") as f:
api_required = f.read().splitlines()
VERSION = "0.0.2.6"
DESCRIPTION = ( DESCRIPTION = (
"The official gpt4free repository | various collection of powerful language models" "The official gpt4free repository | various collection of powerful language models"
) )
@ -26,11 +29,16 @@ setup(
long_description_content_type="text/markdown", long_description_content_type="text/markdown",
long_description=long_description, long_description=long_description,
packages=find_packages(), packages=find_packages(),
data_files=["interference/app.py"],
install_requires=required, install_requires=required,
url='https://github.com/xtekky/gpt4free', # Link to your GitHub repository extras_require={"api": api_required},
entry_points={
"console_scripts": ["g4f=interference.app:main"],
},
url="https://github.com/xtekky/gpt4free", # Link to your GitHub repository
project_urls={ project_urls={
'Source Code': 'https://github.com/xtekky/gpt4free', # GitHub link "Source Code": "https://github.com/xtekky/gpt4free", # GitHub link
'Bug Tracker': 'https://github.com/xtekky/gpt4free/issues', # Link to issue tracker "Bug Tracker": "https://github.com/xtekky/gpt4free/issues", # Link to issue tracker
}, },
keywords=[ keywords=[
"python", "python",
@ -65,4 +73,4 @@ setup(
"Operating System :: MacOS :: MacOS X", "Operating System :: MacOS :: MacOS X",
"Operating System :: Microsoft :: Windows", "Operating System :: Microsoft :: Windows",
], ],
) )