mirror of
https://github.com/ollama/ollama.git
synced 2025-09-26 18:57:04 +02:00
add GET /models
endpoint
This commit is contained in:
6
proto.py
6
proto.py
@@ -74,6 +74,12 @@ def generate():
|
||||
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__":
|
||||
app.run(debug=True, threaded=True, port=5001)
|
||||
app.run()
|
||||
|
Reference in New Issue
Block a user