Coverage for docs_src/events/tutorial003.py: 100%

1from contextlib import asynccontextmanager 5 ctx1abcde

2 

3from fastapi import FastAPI 5 ctx1abcde

4 

5 

6def fake_answer_to_everything_ml_model(x: float): 5 ctx1abcde

7 return x * 42 5 ctx1abcde

8 

9 

10ml_models = {} 5 ctx1abcde

11 

12 

13@asynccontextmanager 5 ctx1abcde

14async def lifespan(app: FastAPI): 5 ctx1abcde

15 # Load the ML model 

16 ml_models["answer_to_everything"] = fake_answer_to_everything_ml_model 5 ctx1abcde

17 yield 5 ctx1abcde

18 # Clean up the ML models and release the resources 

19 ml_models.clear() 5 ctx1abcde

20 

21 

22app = FastAPI(lifespan=lifespan) 5 ctx1abcde

23 

24 

25@app.get("/predict") 5 ctx1abcde

26async def predict(x: float): 5 ctx1abcde

27 result = ml_models["answer_to_everything"](x) 5 ctx1abcde

28 return {"result": result} 5 ctx1abcde