llama-cpp-python/examples/high_level_api/fastapi_server.py at main · Mrgithub93/llama-cpp-python

Repository navigation

"""Example FastAPI server for llama.cpp.

To run this example:

```bash

pip install fastapi uvicorn sse-starlette

export MODEL=../models/7B/...

```

Then run:

```

uvicorn llama_cpp.server.app:app --reload

```

python3 -m llama_cpp.server

```

Then visit http://localhost:8000/docs to see the interactive API docs.

To actually see the implementation of the server, see llama_cpp/server/app.py

"""

import os

import uvicorn

from llama_cpp.server.app import create_app

if __name__ == "__main__":

app = create_app()

uvicorn.run(

app, host=os.getenv("HOST", "localhost"), port=int(os.getenv("PORT", 8000))

)