baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add GET /models endpoint

Browse source
This commit is contained in:
Jeffrey Morgan 2023-06-25 14:22:07 -04:00
parent 6ba93652c6
commit 51164f57a2
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
proto.py
View file

@ -74,6 +74,12 @@ def generate():
stream_with_context(stream_response()), mimetype="text/event-stream" stream_with_context(stream_response()), mimetype="text/event-stream"
) )
@app.route("/models", methods=["GET"])
def models():
all_files = os.listdir("./models")
bin_files = [file.replace(".bin", "") for file in all_files if file.endswith(".bin")]
return Response(json.dumps(bin_files), mimetype="application/json")
if __name__ == "__main__": if __name__ == "__main__":
app.run(debug=True, threaded=True, port=5001) app.run(debug=True, threaded=True, port=5001)
app.run() app.run()