2023-04-05 20:23:25 +00:00
|
|
|
"""Example FastAPI server for llama.cpp.
|
|
|
|
|
|
|
|
To run this example:
|
|
|
|
|
|
|
|
```bash
|
|
|
|
pip install fastapi uvicorn sse-starlette
|
|
|
|
export MODEL=../models/7B/...
|
|
|
|
```
|
|
|
|
|
2023-04-29 05:43:37 +00:00
|
|
|
Then run:
|
|
|
|
```
|
|
|
|
uvicorn llama_cpp.server.app:app --reload
|
|
|
|
```
|
2023-04-05 20:23:25 +00:00
|
|
|
|
2023-04-29 05:43:37 +00:00
|
|
|
or
|
2023-04-05 20:23:25 +00:00
|
|
|
|
2023-04-29 05:43:37 +00:00
|
|
|
```
|
|
|
|
python3 -m llama_cpp.server
|
|
|
|
```
|
2023-04-05 20:23:25 +00:00
|
|
|
|
2023-04-29 05:43:37 +00:00
|
|
|
Then visit http://localhost:8000/docs to see the interactive API docs.
|
2023-04-05 20:23:25 +00:00
|
|
|
|
2023-04-29 05:43:37 +00:00
|
|
|
"""
|
|
|
|
import os
|
|
|
|
import uvicorn
|
2023-04-05 20:23:25 +00:00
|
|
|
|
2023-05-02 02:38:46 +00:00
|
|
|
from llama_cpp.server.app import create_app
|
2023-04-05 20:23:25 +00:00
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2023-05-02 02:38:46 +00:00
|
|
|
app = create_app()
|
2023-04-05 20:23:25 +00:00
|
|
|
|
2023-04-07 01:07:35 +00:00
|
|
|
uvicorn.run(
|
|
|
|
app, host=os.getenv("HOST", "localhost"), port=int(os.getenv("PORT", 8000))
|
|
|
|
)
|